提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
说明
该代码实现一键获取你想从sql server里的数据并按照日期命名并导出到桌面
代码
代码如下(示例):
import os
from datetime import datetime
import pandas as pd
import csv
import time
import pymssql
#读取当前日期
times = datetime.strftime(datetime.today(),'%Y%m%d')
def JoinDB():
conn=pymssql.connect('数据库地址','登录名','密码','使用的库')
##创建游标
cur = conn.cursor()
##执行sql语句
count=cur.execute('查询语句')
# 搜取所有结果
results = cur.fetchall()
# 获取表的数据结构字段
fields = cur.description
return list(results), list(fields)
S=JoinDB()
results=S[0]
fields=S[1]
with open('C:/Users/Administrator/Desktop/data_'+times+'.csv','w',newline='') as f:
f.close()
def writer_file(results,fields):
##查看文件大小
file_size = os.path.getsize('C:/Users/Administrator/Desktop/data_'+times+'.csv')
if file_size == 0:
##表头
name=[]
results_list=[]
for i in range(len(fields)):
name.append(fields[i][0])
for i in range(len(results)):
results_list.append(results[i])
##建立DataFrame对象
file_test = pd.DataFrame(columns=name, data=results_list)
##数据写入,不要索引
file_test.to_csv('C:/Users/Administrator/Desktop/data_'+times+'.csv', encoding='utf-8-sig', index=False)
else:
with open('C:/Users/Administrator/Desktop/data_'+times+'.csv', 'a+', encoding='utf-8-sig',newline='') as file_test:
##追加到文件后面
writer = csv.writer(file_test)
##写文件
writer.writerows(results)
if __name__ == '__main__':
writer_file(results,fields)