Extensions Docs Server: Samples zip files

chrome/common/extensions/docs/server2/zip_data_source.py

+# 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 logging
+import os
+from io import BytesIO
+from zipfile import ZipFile
+
+from subversion_data_source import SubversionDataSource
+
+class ZipDataSource(SubversionDataSource):

[not at google - send to devlin, 2012/07/11 00:35:09]

Why does this need to expose a data source interface, do templates need to be
able to embed zip files? I doubt it. Just a normal class would be fine.

[cduvall, 2012/07/11 20:56:30]

Done.

+  """This class creates a zip file given a samples directory.
+  """
+  def __init__(self, cache_builder, base_path, ignore_path):
+    super(ZipDataSource, self).__init__(cache_builder, base_path, ignore_path)
+    self._zip_cache = cache_builder.build(self._MakeZipFile)
+
+  def _MakeZipFile(self, directory):
+    file_list = self._ListFilesInDirectory(directory, True)
+    common_prefix = os.path.commonprefix(file_list)
+    prefix = common_prefix.rsplit('/', 2)[0]
+    zip_bytes = BytesIO()
+    zip_file = ZipFile(zip_bytes, mode='w')
+    for filename in file_list:
+      try:
+        zip_file.writestr(
+            filename.replace(prefix, ''),
+            self._file_cache.get(self._base_path + '/' + filename))
+      except Exception as e:
+        logging.info(e)
+    zip_file.close()
+    return zip_bytes.getvalue()
+
+  def __getitem__(self, key):
+    return self.get(key)
+
+  def get(self, key):
+    base, ext = os.path.splitext(key)
+    try:
+      return self._zip_cache.get(self._base_path + '/' + base)
+    except:
+      return None