![](/img/trans.png)
[英]SQLAlchemy - How to access column names from ResultProxy and write to CSV headers
[英]Use pymssql to write to CSV and use column names as headers
我有以下腳本用於i)連接到數據庫,ii)運行查詢,iii)將查詢結果保存為csv,以及iv)通過電子郵件發送輸出。
一切正常,除了我似乎無法將列名作為文件頭作為標題。 我已經瀏覽了SO,但是找不到適合我的解決方案。 我的腳本如下。 任何幫助都會很棒:
import pymssql
import csv
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email import encoders
fromaddr='XXXX@gmail.com'
toaddr='XXXX@gmail.com'
msg = MIMEMultipart()
msg['From'] = fromaddr
msg['To'] = toaddr
msg['Subject'] = "This is a test"
body = "This is still a test"
conn = pymssql.connect(server='XXXXXXXXXX.net',
port=XXXX,
user='XXXX',
password='XXXX',
database='XXXX')
cursor = conn.cursor()
query = 'Select * From Table'
cursor.execute(query)
with open("XXXXXXXX.csv","w") as outfile:
writer = csv.writer(outfile, quoting=csv.QUOTE_NONNUMERIC)
for row in cursor:
writer.writerow(row)
msg.attach(MIMEText(body,'plain'))
filename = "XXXXXX.csv"
attachment = open("XXXXXXXXXXX.csv","rb")
from email.mime.base import MIMEBase
part = MIMEBase('application','octet-stream')
part.set_payload((attachment).read())
encoders.encode_base64(part)
part.add_header('Content-Disposition',"attachment; filename= %s" % filename)
msg.attach(part)
text = msg.as_string()
server = smtplib.SMTP('smtp.office365.com',587)
server.starttls()
server.login(fromaddr,"XXXXXX")
text = msg.as_string()
server.sendmail(fromaddr,toaddr,text)
在cursor
對象上使用.description
並遍歷它,以獲取列名,例如:
colNameList = []
for i in range(len(cursor.description)):
desc = cursor.description[i]
colNameList.append(desc[0])
colNames = ','.join(colNameList)
print colNames
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.