python自身有一個比較好的包 tarfile以及zipfile都可以壓縮文件,但是當我們需要加密壓縮文件的時候,這兩個包無法提供,根據官方資料 zipfile的setpassword 是無法設置密碼的
ZipFile.setpassword(pwd):
Set pwd as default password to extract encrypted files
可以看到只能用在提取加密的文件的時候(解壓)纔能有效,詳情見這裏
另闢蹊徑
利用腳本調用命令並壓縮添加文件的方式(支持linux與window),主要方式是壓縮用命令方式,解壓用zipFile並設置密碼解壓
import subprocess
import zipfile as zf
import platform as pf
import os
class ZipObj():
def __init__(self,filepathname,passwd):
self.filepathname = filepathname
self.passwd = passwd
def enCrypt (self,deleteSource=False):
"""
壓縮加密,並刪除原數據
window系統調用rar程序
linux等其他系統調用內置命令 zip -P123 tar source
默認不刪除原文件
"""
target = self.filepathname+".zip"
source = self.filepathname+".txt"
if pf.system()=="Windows":
cmd = ['rar','a','-p%s'%(self.passwd),target,source]
p = subprocess.Popen(cmd,executable=r'C:\Program Files\WinRAR\WinRAR.exe')
p.wait()
else:
cmd = ['zip','-P %s'%(self.passwd),target,source]
p = subprocess.Popen(cmd)
p.wait()
# os.system(" ".join(cmd))
if deleteSource:
os.remove(source)
def deCrypt (self):
"""
使用之前先創造ZipObj類
解壓文件
"""
zfile = zf.ZipFile(self.filepathname+".zip")
zfile.extractall(r"zipdata",pwd=self.passwd.encode('utf-8'))
if __name__ == "__main__":
zipo = ZipObj("文件名","文件密碼")
# zipo.enCrypt (deleteSource=False) ##
zipo.deCrypt()
如果您有更好的解決方案,希望不吝賜教!!