Python解壓.zip文件

''' 解壓一個.zip文件或一個目錄下的所有.zip文件到指定目錄。
    
    運行方法： 
    格式：
        python unzip.py "source_dir" "dest_dir" password
    參數說明：
        source_dir和dest_dir既可以絕對路徑也可以為相對路徑。用""將它們括起為了防止路徑中出現空格。
        source_dir和dest_dir的缺省值表示當前目錄。
        password缺省表示壓縮文件未加密。
    注意：
        1. 若目錄太長，可以將上述語句直接寫入.bat腳本，然后運行腳本。
        2. 密碼的編碼格式為“utf-8”，且不支持WinRAR中zip格式的默認加密算法--CBC模式下的AES-256。
           若想要WinRAR加密壓縮的.zip文件能用本程序順利解壓，請在加密壓縮時勾選“ZIP傳統加密”。     
'''

import sys, os, zipfile

def unzip_single(src_file, dest_dir, password):
    ''' 解壓單個文件到目標文件夾。
    '''
    if password:
        password = password.encode()
    zf = zipfile.ZipFile(src_file)
    try:
        zf.extractall(path=dest_dir, pwd=password)
    except RuntimeError as e:
        print(e)
    zf.close()

def unzip_all(source_dir, dest_dir, password):
    if not os.path.isdir(source_dir):    # 如果是單一文件
        unzip_single(source_dir, dest_dir, password)
    else:
        it = os.scandir(source_dir)
        for entry in it:
            if entry.is_file() and os.path.splitext(entry.name)[1]=='.zip' :
                unzip_single(entry.path, dest_dir, password)
    
if __name__ == "__main__":
    
    # 獲取源路徑和目標路徑
    source_dir = os.getcwd()
    dest_dir = os.getcwd()
    password = None
    if len(sys.argv) == 2: # 指定了壓縮文件所在路徑
        source_dir = sys.argv[1]
    if len(sys.argv) == 3: # 壓縮文件所在和解壓到路徑均指定
        source_dir, dest_dir = os.path.abspath(sys.argv[1].strip('"')), os.path.abspath(sys.argv[2].strip('"'))
    if len(sys.argv) == 4: # 還指定了密碼
        source_dir, dest_dir, password =  os.path.abspath(sys.argv[1].strip('"')), os.path.abspath(sys.argv[2].strip('"')), sys.argv[3]
    if len(sys.argv) > 4:
        print('過多的參數，可能是路徑中有空白字符，請用""將路徑括起。')
        exit()
    
    print("源目錄:", source_dir)
    print("解壓到:", dest_dir)
    print("解壓密碼:", password)
    
    # 判斷源路徑是否合法
    if not os.path.exists(source_dir):
        print("壓縮文件或壓縮文件所在路徑不存在！")
        exit()
    if not os.path.isdir(source_dir) and not zipfile.is_zipfile(source_dir):
        print("指定的源文件不是一個合法的.zip文件！")
        exit()
    
    # 如果解壓到的路徑不存在，則創建     
    if not os.path.exists(dest_dir):
        os.mkdir(dest_dir)
    
    unzip_all(source_dir, dest_dir, password)