这样发周报,升职加薪不是梦...

前面写了篇文章,介绍了怎么使用定时任务发送周报,有些朋友想要知道后续操作,既然朋友有需求,那么必须满足呀,这篇文章就介绍怎么使用定时任务加上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 protected]'
    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,可能最大的能力就是解决问题的能力了。最后,大家如果需要源码,请关注我的公众号回复关键词 定时邮件 获取源码。

你可能感兴趣的:(这样发周报,升职加薪不是梦...)