脚本如下:
#!/user/bin/python #-*-coding: utf-8-*- # 1.登录数据库查询数据 2.将查询的数据生成excel文件 3.xls文件通过邮件发送 # 导入模块:pymysql模块用于登录数据库等相关操作(python3),python2中使用mysqldb模块,需自行安装:pip3 install pymysql # xlwt用于生成excel文件,需自行安装:pip3 istall xlwt # smtplib与email为python自带模块,用于邮件的发送 import datetime import pymysql import xlwt import smtplib from email.mime.multipart import MIMEMultipart from email.mime.text import MIMEText from email.mime.application import MIMEApplication # 定义数据查询日期 # datetime.timedelta 时间加减 # datetime.date.today() 2019-10-25 # tmf_day 本月第一天 2019-10-01 # lme_day 上月最后一天 2019-09-30 # lmf_day 上月第一天 2019-09-01 # datetime.datetime.now() 2019-10-25 11:03:52 # today_zero 今天的零时 2019-10-25 00:00:00 # tmf_zero 本月的第一个零时 2019-10-01 00:00:00 # lme_minute 上月的最后一分钟 2019-09-30 23:59:59 today = datetime.date.today() tmf_day = today.replace(day=1) lme_day = tmf_day – datetime.timedelta(days=1) lmf_day = lme_day.replace(day=1) now = datetime.datetime.now() today_zero = now – datetime.timedelta(hours=now.hour, minutes=now.minute, seconds=now.second) tmf_zero = today_zero.replace(day=1) lme_minute = tmf_zero – datetime.timedelta(hours=0, minutes=0, seconds=1) # 1.登录数据库 # myconn 定义连接数据库语句 # pymysql.cursors.DictCursor会输出字典格式,pymysql.cursors.Cursor默认情况下cursor方法返回的是BaseCursor类型对象,BaseCursor类型对象在>执行查询后每条记录的结果以列表(list)表示。 # query_sql 定义查询语句 # mycursor.fetchmany(x)接收x个数据结果,mycursor.fetchall()接收所有数据结果。 # mycursor.close()查询结束 # myconn.close()断开连接 # 传递单个参数:field = ‘-‘ # sql_talk=”UPDATE cnp.Test set a=” where b=’%s'” # cursor.execute(sql_talk % field) # 传递多个参数:使用{0}占位符 # field = ‘-‘ # a = ‘code’ # sql_talk=”UPDATE cnp.Test set {0}=” where business_registration_code='{1}'”.format(a,field) # cursor.execute(sql_talk) # 或者 # cur.execute(“select score from class_info where sex=’%s’ and age=%d” % (‘XX’,’XX’)) %s表示字符,%d表示整型 myconn = pymysql.connect(host=’192.168.42.128′,user=’XXXX’,passwd=’xxxx’,database=’test’,charset=”utf8″) mycursor = myconn.cursor(cursor=pymysql.cursors.Cursor) query_sql = “”” SELECT conference_number,conference_title,conference_name,ccs_login_name,start_time,amount FROM `cost_08` where end_time between ‘%s’ and ‘%s’ “”” mycursor.execute(query_sql % (lmf_day,lme_minute)) query_result=mycursor.fetchall() mycursor.close() myconn.close() # 2.生成excel文件 # excel的位置坐标为0开始,第一行第一列为(0,0),第一行第二列(0,1),第二行第一列(1,0)… # list1 第一行,用于生成需要查询的字段 # wbk.add_sheet创建worksheet,相当于excel中的的sheet1 # xlwt.XFStyle()初始化样式 # datastyle.num_format_str 设置excel单元格格式 list1 = (‘第一列’,’第二列’,’第三列’,’第四列’,’第五列’,’第六列’) wbk = xlwt.Workbook(encoding=’utf-8′) worksheet = wbk.add_sheet(‘My worksheet’) datastyle = xlwt.XFStyle() datastyle.num_format_str = ‘yyyy-mm-dd hh:mm:ss’ # len(list1):list1的长度,用于表示list1的字段数 # list 查询结果的数据列数,即需要查询的字段数 # worksheet.write:将元素写入对应的表格中 # worksheet.write(0,list,list1[list]):将list1的相应字段填入第一行,及需要查询的字段。 # row 查询结果的条数数,即共查询到多少条记录。(0,0)已填入数据,所以从(1,0)开始,row+1行结束。 # list循环到第四列时,设置excel单元格格式,其余列未设置格式,循环不变。 # worksheet.write(row+1,list,query_result[row][list])等同于:worksheet.write(row+1,0,query_result[row][0]) # 如查询结果为(‘1,2′,’2,3′,’3,4’)则: # (1,0)结果为1 (1,1)结果为2 12 # (2,0)结果为2 (2,1)结果为3 23 # (3,0)结果为3 (3,1)结果为4 34 # wbk.save 保存文件,后跟文件名称
for list in range (0,len(list1)): worksheet.write(0,list,list1[list]) for row in range(0,len(query_result)): if list == 4:
worksheet.write(row+1,4,query_result[row][4],datastyle)
else:
worksheet.write(row+1,list,query_result[row][list])
wbk.save(‘xx.xls’) # 3.定义发送邮件函数 # def send_email(): 定义函数 # msg = MIMEMultipart()用于构建带附件的实例 # msg[“Subject”] 邮件主题 # msg[“From”] = user 发件人 # msg[“To”] = to 收件人 # part1 = MIMEText(“你好,\n\n 顶育上月使用数据见附件。”) 构建纯文本文件 # msg.attach(part1) attach表示固定,用于构建邮件内容 # part2 = MIMEApplication(open(r’C:\Users\Administrator.USER-20190917HT\Desktop\顶育.xls’,’rb’).read()) 构建附件1 # open(r’C:\Users\Administrator.USER-20190917HT\Desktop\顶育.xls’,’rb’).read() 以二进制只读方式打开xx文件 # 等同于 f = open(‘test.txt’) # f.read() # rb二进制只读模式 ‘r’读模式、’w’写模式、’a’追加模式、’b’二进制模式、’+’读/写模式。 # open(r’C:\) r’是防止字符转义的 如果路径中出现’\t’的话 不加r的话\t就会被转义 而加了’r’之后’\t’就能保留原有的样子 # part2.add_header(‘Content-Disposition’,’attachment’,filename=’顶育.xls’) 头部消息,filename可以任意写,写什么名字,邮件中显示什么名字 # smtplib.SMTP(“smtp.139.com”,timeout=30) 发件人邮箱中的SMTP服务器,默认端口是25 # s.login(user,pwd)登录邮箱 # send_email()调用函数 def send_email():
user = ‘xxxxxx@139.com’
pwd = ‘xxxxxxx’
to = ‘xxxxxx@qq.com’
msg = MIMEMultipart()
msg[“Subject”] = ‘xxxx上月使用情况统计’
msg[“From”] = user
msg[“To”] = to
part1 = MIMEText(“你好,\n\n xxxx上月使用数据见附件。”)
msg.attach(part1)
part2 = MIMEApplication(open(r’C:\Users\Administrator.USER-20190917HT\Desktop\xx.xls’,’rb’).read())
part2.add_header(‘Content-Disposition’,’attachment’,filename=’xx.xls’)
msg.attach(part2)
s = smtplib.SMTP(“smtp.139.com”,timeout=30) s.login(user,pwd) s.sendmail(user,to,msg.as_string()) s.close() send_email()
|