前面写了篇文章,介绍了怎么使用定时任务发送周报,有些朋友想要知道后续操作,既然朋友有需求,那么必须满足呀,这篇文章就介绍怎么使用定时任务加上python定时发送文件给同事。
一、用python发送邮件
先上代码
class MailSender:
...
def check_latest_file(p, today):
counter = 0
for file in p.glob('*.csv'):
if today in file.name:
counter += 1
return counter
def main(date=''):
user = os.environ.get('EMAIL_NAME')
pwd = os.environ.get('EMAIL_PWD')
to_addr = 'email@email.com'
smtpSvr = 'smtp.exmail.qq.com' #这里用的是腾讯的邮箱所以是这个,其它邮箱请按相关邮箱设置
content = '请查看附件'
if date == '':
today = datetime.now().strftime('%Y-%m-%d')
else:
today = date
if os.name == 'nt':
p = Path(r"E:\销售情况")
else:
p = Path("/home/steven/sales_collect")
subject = today + '销售情况'
m = MailSender(smtpSvr, 25)
m.login(user, pwd)
for file in p.glob('*.csv'):
if today in file.name:
m.add_attachment(file)
m.send(subject, content, to_addr)
m.close()
if __name__ == '__main__':
if len(sys.argv) > 1:
main(sys.argv[1])
else:
main()
内容其实不复杂,主要用到了smtplib作为发送邮件的库,email库用来构建邮件, 定义了一个MailSender类,方法实现了邮件的创建和附件的添加,还有最后的发送。
user = os.environ.get('EMAIL_NAME')
pwd = os.environ.get('EMAIL_PWD')
这里为什么要这样写,因为我的代码是托管在github上的,但是测试时又要用到这些比较敏感的内容,比如密码,所以比较安全的方式就是把敏感信息放到系统环境变量中,这样push的时候就不用担心又有什么不该公开的信息放上去了。不过也就是因为这里的设置,后面在ubuntu中出现一个坑,在win里面很正常,但是在ubuntu中却是不太好弄,这个后面再讲。
if os.name == 'nt':
p = Path(r"E:\销售情况")
else:
p = Path("/home/steven/sales_collect")
这一段代码是为了实现,linux和window的兼容,由于我是在win下写的代码和测试的,但是使用的时候是要放在服务器的,所以这里做了个判断。如果是win系统就把文件存到E盘下面的文件夹,如果是linux就存到/home/steven/sales_collect
下面。
if date == '':
today = datetime.now().strftime('%Y-%m-%d')
else:
today = date
这里是为了能按需求取要发送的文件,如果没有传入日期就按最新的日期来取文件,也就是今天,如果有日期,就按照传入的日期取文件,注意我的文件都是按照日期加文件名的方式命名的,所以很方便的按照日期选取。
二、 sh脚本配置
python脚本写好后经过测试,能够正常运行。是时候写sh脚本了。
#!/usr/bin/env bash
cd /home/steven/sales_collect
/usr/bin/python3.6 send_mail.py
写好bash脚本,更改文件权限chmod 700 send_mail.sh
后可以直接运行。可是后面加入定时任务后,却总是取不到系统环境变量。这个就是前面所说的在ubuntu中的坑。几经周折,经Q群朋友 @不再留戀 提醒,做了以下修改:
#!/bin/bash
cd /home/steven/sales_collect/ && /usr/bin/python3.6 send_mail.py
经过测试,终于没有问题了。修改了几个地方,#!
后面我之前的内容是参考python写的,有可能是错的。还有 cd
进入与后面的执行操作最好用 &&
连接一起使用。
三、 定时任务设置
crontab -e #进入编辑模式
# 添加内容
00 20 * * * /home/steven/send_email.sh # 最开始的设置
00 20 * * * /bin/bash /home/steven/send_email.sh # 经朋友提醒后修改
这里可能是和上面的sh脚本的#!
问题一起的,按照最开始的设置,python脚本始终没有取到系统环境变量,加上/bin/bash
后,python脚本正常取到系统环境变量,测试邮箱收到邮件。
之前为了测试的时候,设置的每过2分钟发送一次。至此任务终于完成,后面有时间再研究下,到底是
#!
写错的原因还是定时任务时应该加上/bin/bash
后记
其实,经过验证,最后成功。在编程里一点也不奇怪,总是不断调整思路,最后解决问题。做为一个coder,可能最大的能力就是解决问题的能力了。最后,大家如果需要源码,请关注我的公众号回复关键词 定时邮件
获取源码。