1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#!/usr/bin/env python
# Modified from:
# http://stackoverflow.com/questions/1855095/how-to-create-a-zip-archive-of-a-directory-in-python?answertab=active#tab-top
#
# and:
# http://pymotw.com/2/zipfile/
import os
import zipfile
import argparse
try:
import zlib
compression = zipfile.ZIP_DEFLATED
except:
compression = zipfile.ZIP_STORED
def zipdir(path, zip):
nameBase = os.path.basename(path)
for root, dirs, files in os.walk(path):
for file in files:
baseNameRoot = os.path.basename(root)
if baseNameRoot == nameBase:
archName = file
else:
archName = os.path.basename(root) + "//" + file
print archName
zip.write(
os.path.join(root, file), archName, compress_type=compression)
def parseCommandLine():
# Create parser
parser = argparse.ArgumentParser(
description="zip all files in directory tree")
# Add arguments
parser.add_argument('dirIn', action="store", help="input directory")
parser.add_argument('fileOut', action="store", help="output file")
# Parse arguments
args = parser.parse_args()
return(args)
def main():
# Get input from command line
args = parseCommandLine()
dirIn = args.dirIn
fileOut = os.path.abspath(args.fileOut)
zip = zipfile.ZipFile(fileOut, 'w')
zipdir(dirIn, zip)
zip.close()
if __name__ == "__main__":
main()
|
