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
|
#!/usr/bin/env python
#
# Copyright 2016 Google Inc.
#
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Utilities for zipping and unzipping files."""
import fnmatch
import ntpath
import os
import posixpath
import zipfile
def filtered(names, blacklist):
"""Filter the list of file or directory names."""
rv = names[:]
for pattern in blacklist:
rv = [n for n in rv if not fnmatch.fnmatch(n, pattern)]
return rv
def zip(target_dir, zip_file, blacklist=None): # pylint: disable=W0622
"""Zip the given directory, write to the given zip file."""
if not os.path.isdir(target_dir):
raise IOError('%s does not exist!' % target_dir)
blacklist = blacklist or []
with zipfile.ZipFile(zip_file, 'w', zipfile.ZIP_DEFLATED, True) as z:
for r, d, f in os.walk(target_dir, topdown=True):
d[:] = filtered(d, blacklist)
for filename in filtered(f, blacklist):
filepath = os.path.join(r, filename)
zi = zipfile.ZipInfo(filepath)
zi.filename = os.path.relpath(filepath, target_dir)
if os.name == 'nt':
# Dumb path separator replacement for Windows.
zi.filename = zi.filename.replace(ntpath.sep, posixpath.sep)
perms = os.stat(filepath).st_mode
zi.external_attr = perms << 16L
zi.compress_type = zipfile.ZIP_DEFLATED
with open(filepath, 'rb') as f:
content = f.read()
z.writestr(zi, content)
for dirname in d:
dirpath = os.path.join(r, dirname)
z.write(dirpath, os.path.relpath(dirpath, target_dir))
def unzip(zip_file, target_dir):
"""Unzip the given zip file into the target dir."""
if not os.path.isdir(target_dir):
os.makedirs(target_dir)
with zipfile.ZipFile(zip_file, 'r', zipfile.ZIP_DEFLATED, True) as z:
for zi in z.infolist():
dst_subpath = zi.filename
if os.name == 'nt':
# Dumb path separator replacement for Windows.
dst_subpath = dst_subpath.replace(posixpath.sep, ntpath.sep)
dst_path = os.path.join(target_dir, dst_subpath)
if dst_path.endswith(os.path.sep):
os.mkdir(dst_path)
else:
with open(dst_path, 'wb') as f:
f.write(z.read(zi))
perms = zi.external_attr >> 16L
os.chmod(dst_path, perms)
|
