Home编程语言pythonPython 3.2 实现zip压缩与解压缩功能

Python 3.2 实现zip压缩与解压缩功能

2015 年 1 月 29 日

<

div id=”content” contentScore=”568″>刚接触Python，用Python3试着写了一个Zip文件的压缩和解压缩脚本，不足之处，欢迎拍砖 
功能有限，要按照脚本压缩文件或者文件夹，必须进入文件所在目录，以文件所在目录为工作路径，并且默认压缩后的文件名同于原文件所在的目录名、解压后的文件夹名同于解压前的文件名 
压缩功能分为压缩所在目录的所有文件跟部分文件（部分文件需要手动添加文件名）；解压功能分为解压到当前目录跟解压到指定目录，以下为代码： 

  1. #!/usr/bin/env python3   
  2. import os,zipfile   
  3. def Zip(target_dir):   
  4.     target_file=os.path.basename(os.getcwd())+‘.zip’  
  5.     zip_opt=input(“Will you zip all the files in this dir?(Choose ‘n’ you should add files by hand)y/n: “)   
  6.     while True:   
  7.         if zip_opt==‘y’:       #compress all the files in this dir   
  8.             filenames=os.listdir(os.getcwd())    #get the file-list of this dir   
  9.             zipfiles=zipfile.ZipFile(os.path.join(target_dir,target_file),‘w’,compression=zipfile.ZIP_DEFLATED)   
  10.             for files in filenames:   
  11.                 zipfiles.write(files)   
  12.             zipfiles.close()   
  13.             print(“Zip finished!”)   
  14.             break  
  15.         elif zip_opt==‘n’:     #compress part of files of this dir   
  16.             filenames=list(input(“Please input the files’ name you wanna zip:”))   
  17.             zipfiles=zipfile.ZipFile(os.path.join(target_dir,target_file),‘w’,compression=zipfile.ZIP_DEFLATED)   
  18.             for files in filenames:   
  19.                 zipfiles.write(files)   
  20.             zipfiles.close()   
  21.             print(“Zip finished!”)   
  22.             break  
  23.         else:   
  24.             print(“Please in put the character ‘y’ or ‘n'”)   
  25.             zip_opt=input(“Will you zip all the files in this dir?(Choose ‘n’ you should add files by hand)y/n: “)   
  26. def Unzip(target_dir):   
  27.     target_name=input(“Please input the file you wanna unzip:”)   
  28.     zipfiles=zipfile.ZipFile(target_name,‘r’)   
  29.     zipfiles.extractall(os.path.join(target_dir,os.path.splitext(target_name)[0]))   
  30.     zipfiles.close()   
  31.     print(“Unzip finished!”)   
  32. def main():   
  33.     opt=input(“What are you gonna do?Zip choose ‘y’,unzip choose ‘n’.y/n: “)   
  34.     while True:   
  35.         if opt==‘y’:  #compress files   
  36.             zip_dir=input(“Please input the absdir you wanna put the zip file in:”)   
  37.             Zip(zip_dir)   
  38.             break  
  39.         elif opt==‘n’:  #unzip files   
  40.             unzip_dir=input(“Please input the absdir you wanna put the zip file in(Nothing should be done if you wann unzip files in the current dir):”)   
  41.             if unzip_dir==”:   
  42.                 Unzip(os.getcwd())   
  43.             else:   
  44.                 Unzip(unzip_dir)   
  45.             break  
  46.         else:   
  47.             print(“Please input the character ‘y’ or ‘n'”)   
  48.             opt=input(“What are you gonna do?Zip choose ‘y’,unzip choose ‘n’.y/n: “)   
  49. <

Related Posts

About The Author

bjmayor

程序员，码农，php,python,ios,android,go，产品经理，创业。