Code import

venv/lib/python2.7/site-packages/wheel/archive.py

@@ -0,0 +1,80 @@
+"""
+Archive tools for wheel.
+"""
+
+import os
+import time
+import logging
+import os.path
+import zipfile
+
+log = logging.getLogger("wheel")
+
+
+def archive_wheelfile(base_name, base_dir):
+    '''Archive all files under `base_dir` in a whl file and name it like
+    `base_name`.
+    '''
+    olddir = os.path.abspath(os.curdir)
+    base_name = os.path.abspath(base_name)
+    try:
+        os.chdir(base_dir)
+        return make_wheelfile_inner(base_name)
+    finally:
+        os.chdir(olddir)
+
+
+def make_wheelfile_inner(base_name, base_dir='.'):
+    """Create a whl file from all the files under 'base_dir'.
+
+    Places .dist-info at the end of the archive."""
+
+    zip_filename = base_name + ".whl"
+
+    log.info("creating '%s' and adding '%s' to it", zip_filename, base_dir)
+
+    # Some applications need reproducible .whl files, but they can't do this
+    # without forcing the timestamp of the individual ZipInfo objects.  See
+    # issue #143.
+    timestamp = os.environ.get('SOURCE_DATE_EPOCH')
+    if timestamp is None:
+        date_time = None
+    else:
+        date_time = time.gmtime(int(timestamp))[0:6]
+
+    # XXX support bz2, xz when available
+    zip = zipfile.ZipFile(open(zip_filename, "wb+"), "w",
+                          compression=zipfile.ZIP_DEFLATED)
+
+    score = {'WHEEL': 1, 'METADATA': 2, 'RECORD': 3}
+    deferred = []
+
+    def writefile(path, date_time):
+        st = os.stat(path)
+        if date_time is None:
+            mtime = time.gmtime(st.st_mtime)
+            date_time = mtime[0:6]
+        zinfo = zipfile.ZipInfo(path, date_time)
+        zinfo.external_attr = st.st_mode << 16
+        zinfo.compress_type = zipfile.ZIP_DEFLATED
+        with open(path, 'rb') as fp:
+            zip.writestr(zinfo, fp.read())
+        log.info("adding '%s'" % path)
+
+    for dirpath, dirnames, filenames in os.walk(base_dir):
+        for name in filenames:
+            path = os.path.normpath(os.path.join(dirpath, name))
+
+            if os.path.isfile(path):
+                if dirpath.endswith('.dist-info'):
+                    deferred.append((score.get(name, 0), path))
+                else:
+                    writefile(path, date_time)
+
+    deferred.sort()
+    for score, path in deferred:
+        writefile(path, date_time)
+
+    zip.close()
+
+    return zip_filename