Python杂俎 —— 自动压缩指定格式文件&自动删除

 |||一、问题场景：

之前写过脚本，在远程主机里连接Oracle，每天自动查询数据，导出csv文件，按日期命名排序。但是每天我需要自己手动压缩然后再复制出来。结果就是一个元旦假期回来，后台堆满了每天的csv文件-=。=#

所以想在原本的脚本上新增一个每日自动压缩当天的数据，删除源文件的功能。

 |||二、需求分析：

新增的功能需要满足1.识别指定格式的文件csv——》2.压缩csv文件，按对应日期命名——》3.压缩完成后删除源文件。

 |||三、效果演示：

 |||四、源代码：

import os
import zipfile
import datetime
import time

#定义压缩函数
def csv_zip():
    path = "C:/Users/Administrator/Desktop/ziptest"
    files = os.listdir(path)
    csv_files = []
    for f in files:
        if f.endswith('.csv'):
            csv_files.append(f)
            name_date = f[-8:-4]
    with zipfile.ZipFile('data'+ name_date + '.zip', 'w') as myzip:
        for csv in csv_files:
            myzip.write(csv)
            os.remove(csv)       
    print(name_date+' Zip Complete')
    
def main(h=12, m=30,s=10):
    while True:
        while True:
            now = datetime.datetime.now()
            # 到达设定时间，结束内循环 以小时判断为例
            if now.hour==h:
                break
            # 不到时间就等10秒之后再次检测
            time.sleep(10)
        csv_zip()
        # 设置压缩函数的运行周期
        #time.sleep(86400)
        time.sleep(15)
        
main()