最近在做游戏打点数据的一些统计处理,写了个Python脚本完成每日定时自动查询生成Excel报表并群发邮件的小功能。
拆解几个需求点:
一.连接数据库并查询
以下是一个查询总注册人数的示例代码,host port db user password等填入自己数据库的参数。
mysql就不多说了,根据统计需求自己撸吧
import pymysql
host = 'xxxx'
port = xxx
db = 'xxx'
user = 'xxx'
password = 'xxx'
# ---- 用pymysql 操作数据库
def get_connection():
conn = pymysql.connect(host=host, port=port, db=db, user=user, password=password)
return conn
conn = get_connection()
# 使用 cursor() 方法创建一个 dict 格式的游标对象 cursor
cursor = conn.cursor(pymysql.cursors.DictCursor)
#总注册
cursor.execute("select count(distinct userId) as total from Statistics where ops = 'register' ")
data = cursor.fetchone()
allTotalRegisterNum = data['total']
print("-- 总注册人数: %s " % (data['total']))
二.查询的数据生成Excel报表
以下为统计关卡数据并生成Excel的小例子,startPersonNum,finishPersonNum,ratio都是之前查库处理好的数据,生成的Excel名称为 地区+数据统计+日期,region为统计的地区,yesterday为昨天日期
import xlwt
workbook = xlwt.Workbook(encoding = 'ascii')
worksheet = workbook.add_sheet('关卡数据')
#关卡统计数据
print("-- 开始关卡数据统计 -- ")
worksheet.col(1).width = 3000
worksheet.col(2).width = 3000
worksheet.col(3).width = 3000
worksheet.write(0, 1, "开始关卡人数")
worksheet.write(0, 2, "结束关卡人数")
worksheet.write(0, 3, "单关完成率")
worksheet.write(i, 1, startPersonNum)
worksheet.write(i, 2, finishPersonNum)
worksheet.write(i, 3, ratio)
workbook.save("%s_数据统计_%s.xls" % (region, str(yesterday))) # 保存文件
三.发送邮件
sender 发送发,receivers 接收方数组,? ? ? ? ?mail_host 邮件服务器,mail_user="xxx" ? ?#用户名 ?mail_pass="xxx" ? #口令?
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.header import Header
from email.mime.base import MIMEBase
from email import encoders
sender = 'xxx.com'
receivers = ['xxx.com','xxx2.com'] # 接收邮件,可设置为你的QQ邮箱或者其他邮箱
def SendEmail():
# 今天日期
today = datetime.date.today()
# 昨天时间
yesterday = today - datetime.timedelta(days=1)
for receiver in receivers:
#创建一个带附件的实例
message = MIMEMultipart()
message['From'] = sender
message["To"] = receiver
subject = '%s数据统计'%(str(yesterday))
message['Subject'] = '%s数据统计'%(str(yesterday))
#邮件正文内容
message.attach(MIMEText('%s数据统计'%(str(yesterday)), 'plain', 'utf-8'))
for channel in channels:
if(os.path.exists("%s_数据统计_%s.xls"%(region, str(yesterday)))):
att1 = MIMEBase('application', "octet-stream")
att1.set_payload(open("%s_数据统计_%s.xls"%(channel, str(yesterday)),'rb').read())
encoders.encode_base64(att1)
att1.add_header('Content-Disposition', 'attachment; filename="%s-%s.xls"'%(channel, str(yesterday)))
message.attach(att1)
try:
# 第三方 SMTP 服务
mail_host="smtp.yeah.net" #设置服务器
mail_user="xxx" #用户名
mail_pass="xxx" #口令
smtpObj = smtplib.SMTP()
smtpObj.connect(mail_host,25) # 25 为 SMTP 端口号
smtpObj.login(mail_user, mail_pass)
smtpObj.sendmail(sender, [receiver], message.as_string())
print("邮件发送成功")
except smtplib.SMTPException as e:
print("Error: 无法发送邮件%s"% e)
四.定时功能
以下为每日1点启动发送邮件函数的例子
from apscheduler.schedulers.blocking import BlockingScheduler
def job():
SendEmail()
def start():
# BlockingScheduler
scheduler = BlockingScheduler()
scheduler.add_job(job, 'cron', day_of_week='0-6', hour=1, minute=00)
scheduler.start()
if __name__ == '__main__':
start()
五.几个坑
1.数据库连接超时。
代码如果运行在国内本机,连接海外服务器数据库的话是很容易超时的,可以将代码布置到服务器上或者挂VP
2.发送邮件代码中的Password
不是邮箱的密码,是邮箱的授权码,是邮箱的授权码,是邮箱的授权码。重要的事情说三遍。授权码去自己的发送方邮箱里设置获取
3.邮件附件的命名
命名不能有乱七八糟的字符,否则会发现贴到邮件中是bin文件
完整示例代码git地址: git@github.com:Dejavu0709/Statistics.git
发送邮件常见问题参考链接:Python发送邮件smtplib.SMTP各报错问题的解决方法_DearMorning的博客-CSDN博客
|