Usage:
1. Create a blank / empty Zip file:
Zip(r'c:\temp\foo.zip')
2. Create a zip file for files and folders
Zip(r'c:\temp\foo.zip', [r'*.txt', r'Folder', r'Folder2\*.txt'])
Assume following directory structure.
c:\temp
c:\temp\Apple\Pie.txt
c:\temp\Orange\Slice.txt
c:\temp\Banana\Pudding.txt
Set the current directory to c:\temp & execute following command will create a Zip file with following path names:
# set current working directory to c:\temp
Zip(r'c:\temp\foo.zip', [r'c:\temp\Apple', r'c:\temp\Orange', r'c:\temp\Banana'])
Apple\Pie.txt
Orange\Slice.txt
Banana\Pudding.txt
Code:
import collections
import glob
import os
import re
import shutil
import sys
import threading
import time
import zipfile
# Helper function to retrieve files from a list of paths. Paths can contain wildcards
def FileListFromPathList(PathList=[]):
if not isinstance(PathList, list):
PathList = [PathList]
for path in PathList:
if not path.strip():
continue
path = os.path.abspath(path)
for found in glob.glob(path):
if os.path.isfile(found):
yield found
else:
for base, dirs, files in os.walk(found):
for file in files:
yield os.path.join(base, file)
# Function to do the zipping
def Zip(ZipFilePath, PathList=[], Mode='w', ZipFolderPrefix='', FromCurrentDirectory=True):
ZipFilePath = Expand(ZipFilePath)
cwd = os.getcwd()
rootlen = len(cwd) + 1
if ZipFolderPrefix:
ZipFolderPrefix = ZipFolderPrefix + r'\\'
def MakeZipFolderName(FilePath):
if FromCurrentDirectory and cwd in FilePath:
return ZipFolderPrefix + FilePath[rootlen:]
else:
return ZipFolderPrefix + os.path.basename(FilePath)
zipFile = zipfile.ZipFile(ZipFilePath, Mode)
print r'Zipping to %s' % (ZipFilePath)
for path in FileListFromPathList(PathList):
zipPath = MakeZipFolderName(path)
print r' Zipping %s' % (zipPath)
zipFile.write(path, zipPath)
zipFile.close()
