python 利用腳本命令壓縮加密文件並刪除源文件

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()

如果您有更好的解決方案,希望不吝賜教!!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章