split platform

lib/python3.5/site-packages/wheel/archive.py

@@ -0,0 +1,61 @@
+"""
+Archive tools for wheel.
+"""
+
+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)
+
+    # 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):
+        zip.write(path, path)
+        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)
+
+    deferred.sort()
+    for score, path in deferred:
+        writefile(path)
+
+    zip.close()
+
+    return zip_filename