工作中每天需要收集部门内的fr文件,发送给外部部门的同事帮忙上传,这么发了有大半年,昨天亮光一闪,为什么不做成自动化呢,于是用python实现了整个流程,今天体验了一下真是美滋滋。
代码如下
首先导入需要的包
1
2
3
4
|
import win32com.client as win32 import datetime import os import zipfile |
定义三个函数,都是网上抄别的同学作业来的(侵删)
邮箱用的是outlook
1
2
3
4
5
6
7
8
9
10
11
|
#压缩文件夹函数 def zip_ya(startdir,file_news): file_news = startdir + '.rar' # 压缩后文件夹的名字 z = zipfile.zipfile(file_news, 'w' ,zipfile.zip_deflated) #参数一:文件夹名 for dirpath, dirnames, filenames in os.walk(startdir): fpath = dirpath.replace(startdir,'') #这一句很重要,不replace的话,就从根目录开始复制 fpath = fpath and fpath + os.sep or '' #这句话理解我也点郁闷,实现当前文件夹以及包含的所有文件的压缩 for filename in filenames: z.write(os.path.join(dirpath, filename),fpath + filename) print ( '压缩成功' ) z.close() |
1
2
3
4
5
6
7
8
|
#创建文件夹函数 def mkdir(path): folder = os.path.exists(path) if not folder: os.makedirs(path) print "创建文件夹成功" else : print "文件夹已存在" |
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
#发送邮件函数 def sendmail(path): sub = '上传fr文件申请' body = '@xx,\r请帮忙上传fr文件,谢谢!' outlook = win32.dispatch( 'outlook.application' ) receiver = [ 'xxx@xx.com' ] ccreceiver = [ 'xxx@xx.com;xxx@xx.com;xxx@xx.com;xxx@xx.com' ] mail = outlook.createitem( 0 ) mail.to = receiver[ 0 ] mail.cc = ccreceiver[ 0 ] mail.subject = sub.decode( 'utf-8' ) mail.body = body.decode( 'utf-8' ) mail.attachments.add(path) mail.send() |
文件夹名称为日期,每天脚本运行时,会新建一个明天的文件夹,并把昨天的压缩文件删除,所以先定义几个日期参数。
这里碰到一个坑,文件路径含中文时,用这个函数os.path.exists()测试都是false,即没有被识别到,用unicode(todaypath,'utf-8')转为unicode后问题解决。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
#获取今天明天昨天的日期 today = datetime.date.today().strftime( "%y%m%d" ) tomorrow = (datetime.date.today() + datetime.timedelta(days = 1 )).strftime( "%y%m%d" ) yesterday = (datetime.date.today() + datetime.timedelta(days = - 1 )).strftime( "%y%m%d" ) #定义文件路径 path = '//tcent.cn/dfs/26.xx事业部/10.xx市场营销中心/04.xxx部/02.xxx组/fr文件上传/' todaypath = path + today todayfile = path + today + '.rar' tomorrowpath = path + tomorrow utodaypath = unicode (todaypath, 'utf-8' ) utodayfile = unicode (todayfile, 'utf-8' ) utomorrowpath = unicode (tomorrowpath, 'utf-8' ) #定义昨天的压缩文件 yesterdayfile = path + yesterday + '.rar' uyesterdayfile = unicode (yesterdayfile, 'utf-8' ) #计算今天文件夹下的文件个数 filenum = 0 for filename in os.listdir(utodaypath): filenum + = 1 #创建明天的文件夹 mkdir(utomorrowpath) #删除昨天的压缩文件 if os.path.exists(uyesterdayfile): # 如果文件存在 os.remove(uyesterdayfile) else : print ( 'no such file:%s' % uyesterdayfile) |
在思考如何让脚本每天自动运行时,决定采用windows定时任务配置(因为没看懂python定时器..)但是windows只能设置为每天运行,实际上周末、节假日是不需要发送邮件的,而节假日补班时需要运行任务,可以在代码端进行控制。
if条件那段就是先判断是否是空文件夹,如果没有文件就不用发了,如果有文件,再判断今天的日期,决定要不要发邮件。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
#获取今天是周几 weekoftoday = datetime.date.today().weekday() #节假日列表 holiday = [ '20180924' , '20181001' , '20181002' , '20181003' , '20181004' , '20181005' ] #补班列表 workday = [ '20180924' , '20180925' ] #是否是周末 isweekend = (weekoftoday = = 5 or weekoftoday = = 6 ) #是否是小长假 isholiday = today in holiday #是否不要补班 isworkday = today not in workday #文件夹是否为空 isnullfile = (filenum = = 0 ) #判断是否要压缩文件并发送邮件 #周末、工作日放假的节假日、文件夹为空时不执行 #补班的周末例外 if isnullfile: pass else : if ((isweekend or isholiday) and isworkday ): pass else : #压缩今天的文件夹 zip_ya(utodaypath,today) #发送邮件 sendmail(utodayfile) |
最后把这个python存成bat文件,去windows定时任务里配置即可。
1
2
3
|
@echo off cd d:\myprograms\sendmail start python sendmail.py |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/madaokuma/article/details/81543973