-
Notifications
You must be signed in to change notification settings - Fork 0
/
zipClass.py
87 lines (75 loc) · 2.64 KB
/
zipClass.py
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#http://blog.donews.com/limodou/archive/2005/11/13/625145.aspx
import zipfile
import os.path
import os
gLocalEncode = 'gbk'
def encode2Local(s):
if type(s) == unicode:
return s.encode(gLocalEncode)
else:
return s
def decode2Local(s):
return s.decode(gLocalEncode)
class ZFile(object):
def __init__(self, filename, mode='r', basedir=''):
self.filename = filename
self.mode = mode
if self.mode in ('w', 'a'):
self.zfile = zipfile.ZipFile(filename, self.mode, compression=zipfile.ZIP_DEFLATED)
else:
self.zfile = zipfile.ZipFile(filename, self.mode)
self.basedir = basedir
if not self.basedir:
self.basedir = os.path.dirname(filename)
def addfile(self, path, arcname=None):
'''
if (type(path) != unicode):
raise 'input must be unicode'
if (not (arcname is None)):
if (type(arcname) != unicode):
raise 'input must be unicode'
'''
path = path.replace('\\', '/')
if not arcname:
if path.startswith(self.basedir):
arcname = path[len(self.basedir):]
else:
arcname = ''
#########################
# The first param should be unicode, so zipfile can retrieve the info from
# file system by this path, check zipfile line 541
#self.zfile.write(encode2Local(path), encode2Local(arcname))
self.zfile.write(encode2Local(path), encode2Local(arcname))
#return the info of the newly added file
return self.zfile.filelist[-1]
def addfiles(self, paths):
for path in paths:
if isinstance(path, tuple):
self.addfile(*path)
else:
self.addfile(path)
def close(self):
self.zfile.close()
def extract_to(self, path):
for p in self.zfile.namelist():
self.extract(p, path)
def extract(self, filename, path):
if not filename.endswith('/'):
f = os.path.join(path, filename)
dir = os.path.dirname(f)
if not os.path.exists(dir):
os.makedirs(dir)
#print 'extracting:', f
zipFilename = encode2Local(filename)
file(f, 'wb').write(self.zfile.read(zipFilename))
def list(self):
for i in self.zfile.namelist():
yield decode2Local(i)
def create(zfile, files):
z = ZFile(zfile, 'w')
z.addfiles(files)
z.close()
def extract(zfile, path):
z = ZFile(zfile)
z.extract_to(path)
z.close()