用python压缩文件
概述
上次利用 python 中的 os 模块的**<os.walk()>**成功遍历的目录树
昨天又学到
程序目的
每执行一次,新建一个压缩文件,压缩文件名如下
- <备份目录名>_N.zip
- 下一次执行脚本文件,数字 N+1 备份的zip文件,放入back_up 文件夹中
代码实现思路
① #TODO: 导入 zipfile, os 模块
#!python3
import os
import zipfile
② #TODO: 检测 back_up 目录下,是否存在同名压缩文件,存在则生成n+1的新文件名
number = 1
while True:
zipFilename = os.path.basename(folder) + '_' + str(number) + '.zip'
if not os.path.exists(zipFilename):
break
number = number + 1
③ #TODO: 以 ‘w’ 新建一个压缩文件,用 ② 的 zipFilename 命名
with zipfile.ZipFile(zipFilename, 'w') as backZip:
④ #TODO: 接下来,遍历要压缩的目录,把每一个目录,和其下的文件都写入压缩文件
#os.walk()遍历每一个目录下的文件
for folderName, subfolders, filenames in os.walk(folder):
print('当前目录下文件正在压缩 %s ...' % (folderName))
#先添加文件夹,到备份zip压缩包中去
backupZip.write(folderName)
#再添加当前目录下的文件到压缩包中去
for filename in filenames:
backupZip.write(os.path.join(folderName, filename))
⑤ #TODO: 提示完成
print("[!] 备份完成.")
整理代码
#!python3
#backupToZip.py - 备份任意目录下所有文件,到 back_up 文件夹中
import sys, os, time
import zipfile
def backupToZip(folder):
#备份输入的目录下的所有内容
folder = os.path.abspath(folder) #输入的路径转化为绝对路径
number = 1
while True:
zipFilename = os.path.basename(folder) + '_' + str(number) + '.zip'
if not os.path.exists(zipFilename):
break
number = number + 1
print('建立备份文件 %s ...' % (zipFilename))
with zipfile.ZipFile(zipFilename, 'w') as backupZip:
for folderName, subfolders, filenames in os.walk(folder):
print('当前目录下文件正在压缩 %s ...' % (folderName))
backupZip.write(folderName)
for filename in filenames:
backupZip.write(os.path.join(folderName, filename))
print('[!] 备份完成.')
def main():
os.mkdir(r'.\back_up')
os.chdir(r'.\back_up')
print('backupTozip脚本运行目录:' + str(os.getcwd()))
# 选择要备份的文件夹,以及所在的路径
target = r'D:\py\Auto\delicious'
backupToZip(target)
if __name__ == '__main__':
main()
总结
- 和文件有关的操作,要记得检查路径、文件夹或文件是否存在、是否有**权限(r,w,x)**等 下面这段,通常用来判断 文件不存在、没有权限等情况,执行 if 内的代码,要么报错,要么放心的去做其他的事情
if not os的<文件夹、文件>检查方法:
pass
- 检查文件,是否以 “xxx” 名字开始,并且以某种后缀结尾 下面这段,检查文件是否为 xxx.zip 文件
if filename.startswith("xxx") and filename.endswith('.zip'):
pass