097900648b
Bug: skia:7077 Change-Id: I5119a7d6c7cd2317129d40ae3e0997dfc9978c25 Reviewed-on: https://skia-review.googlesource.com/52664 Reviewed-by: Ravi Mistry <rmistry@google.com> Commit-Queue: Eric Boren <borenet@google.com>
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
#!/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)
|