江苏龙网

  1. 主页 > 生活百科 > >

Python查询DB数据发送指定邮箱

Python

业务需求：
1.读取SQL中的统计数据
2.将统计数据导出CSV
3.把CSV发送指定邮箱
引入需要处理的包：csv（或xlwt），smtplib，MySQL
【Python查询DB数据发送指定邮箱】 
Python/ target=_blank class=infotextkey>Python版本: 3.9
在输出表头时使用2个数组，以顺序建立对应关系输出，还有其他的巧妙的方法吗？
话不多说，直接上代码
DB建表就不贴了，一个user表（name,phone,sex）
# -*- coding: utf-8 -*-# 导入CSV安装包import csv# 日期import datetime# 邮件import smtplibfrom email.header import Headerfrom email.mime.multipart import MIMEMultipartfrom email.mime.text import MIMEText# mysqlimport pymysql.cursors# Excel处理import xlwt# 连接数据库connect = pymysql.Connect(host='127.0.0.1',port=3306,user='test_db',passwd='dbmima',db='test',charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)# 第三方 SMTP 服务# 设置服务器mail_host = "smtp.exmail.qq.com"# 用户名mail_user = "76079901@qq.com"# 口令mail_pass = "mima"# 查询数据sql = '''select * from user'''# 定义表头array = [ 'name', 'phone', 'sex']title = ['用户名', '手机号', '性别']def process():export_csv()send_email()def export_excel():# 获取游标cursor = connect.cursor()book = xlwt.Workbook()sheet = book.add_sheet("sheet1")cursor.execute(sql)result = cursor.fetchall()# 输出表头for z, t in enumerate(title):sheet.write(0, z, t)# 输出数据for x, row in enumerate(result):for y, field in enumerate(array):# print(x+1, y, row[field])sheet.write(x + 1, y, get_value(str(row[field])))bookName = 'UR_Record_%s.xls' % datetime.datetime.now().strftime('%Y%m%d')# 保存Excelbook.save(bookName)# 关闭连接cursor.close()connect.close()def export_csv():cursor = connect.cursor()cursor.execute(sql)result = cursor.fetchall()print("数据量: " + str(len(result)))book_name = 'Record_%s.csv' % datetime.datetime.now().strftime('%Y%m%d')# 1. 创建文件对象 解决表情符乱码，设置utf-8-sig，且数据库链接改为utf8mb4f = open(book_name, 'w', encoding='utf-8-sig', newline='')# 2. 基于文件对象构建 csv写入对象csv_writer = csv.writer(f)# 3. 构建列表头csv_writer.writerow(title)# 4. 写入csv文件内容for x, row in enumerate(result):r = ['init'] * len(array)for y, field in enumerate(array):r[y] = row[field]# 保存单条记录，使用writerow(result)即可# print(r)# csv_writer.writerow([row['mobile'], row['join_at'], row['source']])csv_writer.writerow(r)print("导出完成")# 关闭连接f.close()def get_value(val):if val == 'None':return ' 'else:return valdef send_email():sender = mail_user# 接收邮件，可设置为你的QQ邮箱或者其他邮箱receivers = ['76079901@qq.com']# 三个参数：第一个为文本内容，第二个 plain 设置文本格式，第三个 utf-8 设置编码message = MIMEMultipart()# 发送者message['From'] = Header("FROM", 'utf-8')# 接收者message['To'] = Header("To", 'utf-8')# 主题message['Subject'] = Header('【Header Content】', 'utf-8')# 邮件正文内容message.attach(MIMEText('''正文''', 'plain', 'utf-8'))# 附件curDay = datetime.datetime.now().strftime('%Y%m%d')att1 = MIMEText(open('Record_%s.csv' % curDay, 'rb').read(), 'base64', 'utf-8')att1["Content-Type"] = 'Application/octet-stream'# filename = 文件名att1["Content-Disposition"] = 'attachment;filename="Record_%s.csv"' % curDaymessage.attach(att1)try:smtp_obj = smtplib.SMTP_SSL(mail_host, 465)smtp_obj.login(mail_user, mail_pass)smtp_obj.sendmail(sender, receivers, message.as_string())smtp_obj.quit()print("send success!!!")except smtplib.SMTPException as e:print("Error: send Fail!!! ", e)if __name__ == '__main__':process()



    推荐阅读


    上一篇：轻量级低侵入式在线构建、自动部署、日常运维、项目监控软件

    下一篇：linux系统中Nginx+FFmPeg+vlc实现网页视频播放配置过程