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
|
# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import os
from io import BytesIO
import re
from zipfile import ZipFile
import compiled_file_system as compiled_fs
class ExampleZipper(object):
'''This class creates a zip file given a samples directory.
'''
def __init__(self, compiled_fs_factory, base_path):
self._base_path = base_path.rstrip('/')
self._file_cache = compiled_fs_factory.CreateIdentity(ExampleZipper)
self._zip_cache = compiled_fs_factory.Create(self._MakeZipFile,
ExampleZipper)
def _MakeZipFile(self, base_dir, files):
if 'manifest.json' not in files:
return None
zip_bytes = BytesIO()
zip_file = ZipFile(zip_bytes, mode='w')
try:
for file_name in files:
file_path = '%s%s' % (base_dir, file_name)
file_contents = self._file_cache.GetFromFile(file_path, binary=True)
if isinstance(file_contents, unicode):
# Data is sometimes already cached as unicode.
file_contents = file_contents.encode('utf8')
# We want e.g. basic.zip to expand to basic/manifest.json etc, not
# chrome/common/extensions/.../basic/manifest.json, so only use the
# end of the path component when writing into the zip file.
redundant_prefix = '%s/' % base_dir.rstrip('/').rsplit('/', 1)[0]
zip_file.writestr(file_path[len(redundant_prefix):], file_contents)
finally:
zip_file.close()
return zip_bytes.getvalue()
def Create(self, path):
''' Creates a new zip file from the recursive contents of |path|
as returned by |_zip_cache|.
Paths within the zip file are given relative to and including |path|.
'''
return self._zip_cache.GetFromFileListing(
'%s/%s' % (self._base_path, path.strip('/')))
|
