Extensions Docs Server: Samples zip files

chrome/common/extensions/docs/server2/example_zipper.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
+import re
+from zipfile import ZipFile
+
+class ExampleZipper(object):
+  """This class creates a zip file given a samples directory.
+  """
+  def __init__(self, cache_builder, base_path, match_path):
+    self._base_path = base_path
+    self._zip_cache = cache_builder.build(self._MakeZipFile)
+    self._svn_cache = cache_builder.build(lambda x: x)
+    self._match_path = match_path
+
+  def _MakeZipFile(self, files):
+    zip_bytes = BytesIO()
+    zip_file = ZipFile(zip_bytes, mode='w')
+    zip_path = os.path.commonprefix(files).rsplit('/', 1)[-2]
+    prefix = zip_path.rsplit('/', 1)[-2]
+    if zip_path + '/manifest.json' not in files:
+      return None
+    for filename in files:
+      try:
+        zip_file.writestr(
+            filename.replace(prefix, ''),
+            self._svn_cache.get(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):
+    if not re.match(self._match_path + '/.*\.zip$', key):
+      return None
+    base, ext = os.path.splitext(key)
+    try:
+      return self._zip_cache.get(self._base_path + '/' + base, True)
+    except:
+      return None