使用Python生成自动报表(Excel)以邮件发送

数据分析师肯定每天都被各种各样的数据数据报表搞得焦头烂额,老板的,运营的、产品的等等。而且大部分报表都是重复性的工作,这篇文章就是帮助大家如何用Python来实现报表的自动发送,解放你的劳动力,可以让你有时间去做更有意思的事情。

首先来介绍下实现自动报表要使用到的Python库:

·pymysql 一个可以连接MySQL实例并且实现增删改查功能的库

·datetime Python标准库中自带的关于时间的库

·openpyxl 一个可以读写07版以后的Excel文档(.xlsx格式也支持)的库

·smtplib SMTP即简单邮件传输协议,Python简单封装成了一个库

·email 一个用来处理邮件消息的库

为什么使用openpyxl库来处理Excel呢?因为它支持每个sheet的行数为100W+,也是支持xlsx格式的文件。如果你接受xls文件,并且每个sheet的行数小于6W,也是可以使用xlwt库,它对大文件的读取速度要大于openpyxl。

接下来我们就进入实战部分,来正式实现这个过程。我把整个实现过程分成几个函数的方式来实现,这样看着会比较有结构感。

一、首先导入所有要用到的库

 
  1. # encoding=utf-8

  2. import pymysql as pms

  3. import openpyxl

  4. import datetime

  5. from email.mime.text import MIMEText

  6. from email.mime.multipart import MIMEMultipart

  7. from email.header import Header

  8. import smtplib

二、 编写一个传入sql就返回数据的函数get_datas(sql)

 
  1. def get_datas(sql):

  2. # 一个传入sql导出数据的函数

  3. # 跟数据库建立连接

  4. conn = pms.connect(host='实例地址', user='用户',

  5. passwd='密码', database='库名', port=3306, charset="utf8")

  6. # 使用 cursor() 方法创建一个游标对象 cursor

  7. cur = conn.cursor()

  8. # 使用 execute() 方法执行 SQL

  9. cur.execute(sql)

  10. # 获取所需要的数据

  11. datas = cur.fetchall()

  12. #关闭连接

  13. cur.close()

  14. #返回所需的数据

  15. return datas

 

三、 编写一个传入sql就返回数据的字段名称的函数get_datas(sql),因为一个函数只能返回一个值,这边就用2个函数来分别返回数据和字段名称(也就是excel里的表头)

 
  1. def get_fields(sql):

  2. # 一个传入sql导出字段的函数

  3. conn = pms.connect(host='数据库实例地址', user='账号',

  4. passwd='密码', database=库名', port=3306, charset="utf8")

  5. cur = conn.cursor()

  6. cur.execute(sql)

  7. # 获取所需要的字段名称

  8. fields = cur.description

  9. cur.close()

  10. return fields

  11.  

四、 编写一个传入数据、字段名称、存储地址返回一个excel 的函数et_excel(data, field, file)

 
  1. def get_excel(data, field, file):

  2. # 将数据和字段名写入excel的函数

  3. # 更多Python视频、源码、资料加群683380553免费获取

  4. #新建一个工作薄对象

  5. new = openpyxl.Workbook()

  6. #激活一个新的sheet

  7. sheet = new.active

  8. #给sheet命名

  9. sheet.title = '数据展示'

  10. #将字段名称循环写入excel第一行,因为字段格式列表里包含列表,每个列表的第一元素才是字段名称

  11. for col in range(len(field)):

  12. #row代表行数,column代表列数,value代表单元格输入的值,行数和列数都是从1开始,这点于python不同要注意

  13. _ = sheet.cell(row=1, column=col+1, value=u'%s' % field[col][0])

  14. #将数据循环写入excel的每个单元格中

  15. for row in range(len(data)):

  16. for col in range(len(field)):

  17. #因为第一行写了字段名称,所以要从第二行开始写入

  18. _ = sheet.cell(row=row+2, column=col + 1, value=u'%s' % data[row][col])

  19. #将生成的excel保存,这步是必不可少的

  20. newworkbook = new.save(file)

  21. #返回生成的excel

  22. return newworkbook

五、 编写一个自动获取昨天日期字符串格式的函数getYesterday()

 
  1. def getYesterday():

  2. # 获取昨天日期的字符串格式的函数

  3. #获取今天的日期

  4. today = datetime.date.today()

  5. #获取一天的日期格式数据

  6. oneday = datetime.timedelta(days=1)

  7. #昨天等于今天减去一天

  8. yesterday = today - oneday

  9. #获取昨天日期的格式化字符串

  10. yesterdaystr = yesterday.strftime('%Y-%m-%d')

  11. #返回昨天的字符串

  12. return yesterdaystr

六、编写一个生成邮件的函数create_email(email_from, email_to, email_Subject, email_text, annex_path, annex_name)

 
  1. def create_email(email_from, email_to, email_Subject, email_text, annex_path, annex_name):

  2. # 输入发件人昵称、收件人昵称、主题,正文,附件地址,附件名称生成一封邮件Python学习交流群:548377875

  3. #生成一个空的带附件的邮件实例

  4. message = MIMEMultipart()

  5. #将正文以text的形式插入邮件中

  6. message.attach(MIMEText(email_text, 'plain', 'utf-8'))

  7. #生成发件人名称(这个跟发送的邮件没有关系)

  8. message['From'] = Header(email_from, 'utf-8')

  9. #生成收件人名称(这个跟接收的邮件也没有关系)

  10. message['To'] = Header(email_to, 'utf-8')

  11. #生成邮件主题

  12. message['Subject'] = Header(email_Subject, 'utf-8')

  13. #读取附件的内容

  14. att1 = MIMEText(open(annex_path, 'rb').read(), 'base64', 'utf-8')

  15. att1["Content-Type"] = 'application/octet-stream'

  16. #生成附件的名称

  17. att1["Content-Disposition"] = 'attachment; filename=' + annex_name

  18. #将附件内容插入邮件中

  19. message.attach(att1)

  20. #返回邮件

  21. return message

七、 生成一个发送邮件的函数send_email(sender, password, receiver, msg)

 
  1. def send_email(sender, password, receiver, msg):

  2. # 一个输入邮箱、密码、收件人、邮件内容发送邮件的函数

  3. try:

  4. #找到你的发送邮箱的服务器地址,已加密的形式发送

  5. server = smtplib.SMTP_SSL("smtp.mxhichina.com", 465) # 发件人邮箱中的SMTP服务器

  6. server.ehlo()

  7. #登录你的账号

  8. server.login(sender, password) # 括号中对应的是发件人邮箱账号、邮箱密码

  9. #发送邮件

  10. server.sendmail(sender, receiver, msg.as_string()) # 括号中对应的是发件人邮箱账号、收件人邮箱账号(是一个列表)、邮件内容

  11. print("邮件发送成功")

  12. server.quit() # 关闭连接

  13. except Exception:

  14. print(traceback.print_exc())

  15. print("邮件发送失败")

八、建立一个main函数,把所有的自定义内容输入进去,最后执行main函数

 
  1. def main():

  2. print(datetime.datetime.now())

  3. my_sql = sql = "SELECT a.id '用户ID',\

  4. a.gmtCreate '用户注册时间',\

  5. af.lastLoginTime '最后登录时间',\

  6. af.totalBuyCount '历史付款子单数',\

  7. af.paidmountUSD '历史付款金额',\

  8. af.lastPayTime '用户最后支付时间'\

  9. FROM table a\

  10. LEFT JOIN tableb af ON a.id= af.accountId ;"

  11. # 生成数据

  12. my_data = get_datas(my_sql)

  13. # 生成字段名称

  14. my_field = get_fields(my_sql)

  15. # 得到昨天的日期

  16. yesterdaystr = getYesterday()

  17. # 文件名称

  18. my_file_name = 'user attribute' + yesterdaystr + '.xlsx'

  19. # 文件路径

  20. file_path = 'D:/work/report/' + my_file_name

  21. # 生成excel

  22. get_excel(my_data, my_field, file_path)

  23.  
  24. my_email_from = 'BI部门自动报表机器人'

  25. my_email_to = '运营部'

  26. # 邮件标题

  27. my_email_Subject = 'user' + yesterdaystr

  28. # 邮件正文

  29. my_email_text = "Dear all,\n\t附件为每周数据,请查收!\n\nBI团队 "

  30. #附件地址

  31. my_annex_path = file_path

  32. #附件名称

  33. my_annex_name = my_file_name

  34. # 生成邮件

  35. my_msg = create_email(my_email_from, my_email_to, my_email_Subject,

  36. my_email_text, my_annex_path, my_annex_name)

  37. my_sender = '阿里云邮箱'

  38. my_password = '我的密码'

  39. my_receiver = [10001@qq.com']#接收人邮箱列表

  40. # 发送邮件

  41. send_email(my_sender, my_password, my_receiver, my_msg)

  42. print(datetime.datetime.now())

  43.  
  44. if __name__ == "__main__":

  45. main();

  46.  

到这里呢发自动邮件的内容就全部结束啦,我每一行都标了注释,如果对于某个库还有些不理解,大家可以自行google或者百度一下。

更多Python视频、源码、资料加群683380553免费获取

转自:https://zhuanlan.zhihu.com/p/41670796

相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页