Format with black

Author: agriyakhetarpal

make_wheels.py

@@ -12,20 +12,19 @@
 from zipfile import ZipFile, ZipInfo, ZIP_DEFLATED
 
 
-ZIG_VERSION_INFO_URL = 'https://ziglang.org/download/index.json'
+ZIG_VERSION_INFO_URL = "https://ziglang.org/download/index.json"
 ZIG_PYTHON_PLATFORMS = {
-    'x86_64-windows': 'win_amd64',
-    'x86-windows':    'win32',
-    'x86_64-macos':   'macosx_12_0_x86_64',
-    'aarch64-macos':  'macosx_12_0_arm64',
-    'i386-linux':     'manylinux_2_12_i686.manylinux2010_i686.musllinux_1_1_i686',
+    "x86_64-windows": "win_amd64",
+    "x86-windows": "win32",
+    "x86_64-macos": "macosx_12_0_x86_64",
+    "aarch64-macos": "macosx_12_0_arm64",
+    "i386-linux": "manylinux_2_12_i686.manylinux2010_i686.musllinux_1_1_i686",
     # renamed i386 to x86 since v0.11.0, i386 was last supported in v0.10.1
-    'x86-linux':      'manylinux_2_12_i686.manylinux2010_i686.musllinux_1_1_i686',
-    'x86_64-linux':   'manylinux_2_12_x86_64.manylinux2010_x86_64.musllinux_1_1_x86_64',
-    'aarch64-linux':
-        'manylinux_2_17_aarch64.manylinux2014_aarch64.musllinux_1_1_aarch64',
-    'armv7a-linux':   'manylinux_2_17_armv7l.manylinux2014_armv7l.musllinux_1_1_armv7l',
-    'powerpc64le-linux':  'manylinux_2_17_ppc64le.manylinux2014_ppc64le.musllinux_1_1_ppc64le',
+    "x86-linux": "manylinux_2_12_i686.manylinux2010_i686.musllinux_1_1_i686",
+    "x86_64-linux": "manylinux_2_12_x86_64.manylinux2010_x86_64.musllinux_1_1_x86_64",
+    "aarch64-linux": "manylinux_2_17_aarch64.manylinux2014_aarch64.musllinux_1_1_aarch64",
+    "armv7a-linux": "manylinux_2_17_armv7l.manylinux2014_armv7l.musllinux_1_1_armv7l",
+    "powerpc64le-linux": "manylinux_2_17_ppc64le.manylinux2014_ppc64le.musllinux_1_1_ppc64le",
 }
 
 
@@ -42,7 +41,7 @@ def writestr(self, zinfo_or_arcname, data, *args, **kwargs):
                 zinfo.external_attr = 0o0664 << 16
 
         zinfo.compress_type = ZIP_DEFLATED
-        zinfo.date_time = (1980,1,1,0,0,0)
+        zinfo.date_time = (1980, 1, 1, 0, 0, 0)
         zinfo.create_system = 3
         super().writestr(zinfo, data, *args, **kwargs)
 
@@ -61,34 +60,42 @@ def make_message(headers, payload=None):
 
 
 def write_wheel_file(filename, contents):
-    with ReproducibleWheelFile(filename, 'w') as wheel:
+    with ReproducibleWheelFile(filename, "w") as wheel:
         for member_info, member_source in contents.items():
             wheel.writestr(member_info, bytes(member_source))
     return filename
 
 
 def write_wheel(out_dir, *, name, version, tag, metadata, description, contents):
-    wheel_name = f'{name}-{version}-{tag}.whl'
-    dist_info  = f'{name}-{version}.dist-info'
+    wheel_name = f"{name}-{version}-{tag}.whl"
+    dist_info = f"{name}-{version}.dist-info"
     filtered_metadata = []
     for header, value in metadata:
         filtered_metadata.append((header, value))
 
-    return write_wheel_file(os.path.join(out_dir, wheel_name), {
-        **contents,
-        f'{dist_info}/METADATA': make_message([
-            ('Metadata-Version', '2.4'),
-            ('Name', name),
-            ('Version', version),
-            *filtered_metadata,
-        ], description),
-        f'{dist_info}/WHEEL': make_message([
-            ('Wheel-Version', '1.0'),
-            ('Generator', 'ziglang make_wheels.py'),
-            ('Root-Is-Purelib', 'false'),
-            ('Tag', tag),
-        ]),
-    })
+    return write_wheel_file(
+        os.path.join(out_dir, wheel_name),
+        {
+            **contents,
+            f"{dist_info}/METADATA": make_message(
+                [
+                    ("Metadata-Version", "2.4"),
+                    ("Name", name),
+                    ("Version", version),
+                    *filtered_metadata,
+                ],
+                description,
+            ),
+            f"{dist_info}/WHEEL": make_message(
+                [
+                    ("Wheel-Version", "1.0"),
+                    ("Generator", "ziglang make_wheels.py"),
+                    ("Root-Is-Purelib", "false"),
+                    ("Tag", tag),
+                ]
+            ),
+        },
+    )
 
 
 def iter_archive_contents(archive):
@@ -109,37 +116,37 @@ def iter_archive_contents(archive):
 
 def write_ziglang_wheel(out_dir, *, version, platform, archive):
     contents = {}
-    contents['ziglang/__init__.py'] = b''
+    contents["ziglang/__init__.py"] = b""
 
     license_files = {}
 
     # The paths to these licenses MUST match both the actual files
     # in the Zig source tarballs and the License-File entries listed
     # below in the metadata.
     license_paths = [
-        'LICENSE',
-        'lib/libc/glibc/LICENSES',
-        'lib/libc/mingw/COPYING',
-        'lib/libc/musl/COPYRIGHT',
-        'lib/libc/wasi/LICENSE',
-        'lib/libc/wasi/LICENSE-APACHE',
-        'lib/libc/wasi/LICENSE-APACHE-LLVM',
-        'lib/libc/wasi/LICENSE-MIT',
-        'lib/libcxx/LICENSE.TXT',
-        'lib/libcxxabi/LICENSE.TXT',
-        'lib/libunwind/LICENSE.TXT'
+        "LICENSE",
+        "lib/libc/glibc/LICENSES",
+        "lib/libc/mingw/COPYING",
+        "lib/libc/musl/COPYRIGHT",
+        "lib/libc/wasi/LICENSE",
+        "lib/libc/wasi/LICENSE-APACHE",
+        "lib/libc/wasi/LICENSE-APACHE-LLVM",
+        "lib/libc/wasi/LICENSE-MIT",
+        "lib/libcxx/LICENSE.TXT",
+        "lib/libcxxabi/LICENSE.TXT",
+        "lib/libunwind/LICENSE.TXT",
     ]
 
     found_license_files = set()
 
     for entry_name, entry_mode, entry_data in iter_archive_contents(archive):
-        entry_name = '/'.join(entry_name.split('/')[1:])
+        entry_name = "/".join(entry_name.split("/")[1:])
         if not entry_name:
             continue
-        if entry_name.startswith('doc/'):
+        if entry_name.startswith("doc/"):
             continue
 
-        zip_info = ZipInfo(f'ziglang/{entry_name}')
+        zip_info = ZipInfo(f"ziglang/{entry_name}")
         zip_info.external_attr = (entry_mode & 0xFFFF) << 16
         contents[zip_info] = entry_data
 
@@ -149,15 +156,19 @@ def write_ziglang_wheel(out_dir, *, version, platform, archive):
             license_files[entry_name] = entry_data  # adding to license files
             found_license_files.add(entry_name)  # tracking found licenses
 
-        if entry_name.startswith('zig'):
-            contents['ziglang/__main__.py'] = f'''\
+        if entry_name.startswith("zig"):
+            contents["ziglang/__main__.py"] = (
+                f"""\
 import os, sys
 argv = [os.path.join(os.path.dirname(__file__), "{entry_name}"), *sys.argv[1:]]
 if os.name == 'posix':
     os.execv(argv[0], argv)
 else:
     import subprocess; sys.exit(subprocess.call(argv))
-'''.encode('ascii')
+""".encode(
+                    "ascii"
+                )
+            )
 
     # Check if any license files are missing and warn. This
     # is useful to detect changes in the Zig archive structure
@@ -167,46 +178,47 @@ def write_ziglang_wheel(out_dir, *, version, platform, archive):
         print(f"Warning: The following license files were not found in the Zig archive: {', '.join(sorted(missing_licenses))}")
         print("This may indicate a change in Zig's license file structure or an error in the list of license files and/or paths.")
 
-    with open('README.pypi.md') as f:
+    with open("README.pypi.md") as f:
         description = f.read()
 
-    dist_info = f'ziglang-{version}.dist-info'
+    dist_info = f"ziglang-{version}.dist-info"
     for license_path, license_data in license_files.items():
-        contents[f'{dist_info}/licenses/{license_path}'] = license_data
+        contents[f"{dist_info}/licenses/{license_path}"] = license_data
 
-    return write_wheel(out_dir,
-        name='ziglang',
+    return write_wheel(
+        out_dir,
+        name="ziglang",
         version=version,
-        tag=f'py3-none-{platform}',
+        tag=f"py3-none-{platform}",
         metadata=[
-            ('Summary', 'Zig is a general-purpose programming language and toolchain for maintaining robust, optimal, and reusable software.'),
-            ('Description-Content-Type', "'text/markdown'; charset=UTF-8; variant=GFM"),
+            ("Summary", "Zig is a general-purpose programming language and toolchain for maintaining robust, optimal, and reusable software."),
+            ("Description-Content-Type", "'text/markdown'; charset=UTF-8; variant=GFM"),
             # The license expression and the file paths MUST remain in sync
             # with the paths in the official Zig tarballs and with the ones
             # defined above in the contents.
-            ('License-Expression', 'MIT'),
-            ('License-File', 'LICENSE'),
-            ('License-File', 'lib/libc/glibc/LICENSES'),
-            ('License-File', 'lib/libc/mingw/COPYING'),
-            ('License-File', 'lib/libc/musl/COPYRIGHT'),
-            ('License-File', 'lib/libc/wasi/LICENSE'),
-            ('License-File', 'lib/libc/wasi/LICENSE-APACHE'),
-            ('License-File', 'lib/libc/wasi/LICENSE-APACHE-LLVM'),
-            ('License-File', 'lib/libc/wasi/LICENSE-MIT'),
-            ('License-File', 'lib/libcxx/LICENSE.TXT'),
-            ('License-File', 'lib/libcxxabi/LICENSE.TXT'),
-            ('License-File', 'lib/libunwind/LICENSE.TXT'),
-            ('Classifier', 'Development Status :: 4 - Beta'),
-            ('Classifier', 'Intended Audience :: Developers'),
-            ('Classifier', 'Topic :: Software Development :: Compilers'),
-            ('Classifier', 'Topic :: Software Development :: Code Generators'),
-            ('Classifier', 'Topic :: Software Development :: Build Tools'),
-            ('Classifier', 'Programming Language :: Other'),
-            ('Classifier', 'Programming Language :: Other Scripting Engines'),
-            ('Project-URL', 'Homepage, https://ziglang.org'),
-            ('Project-URL', 'Source Code, https://github.com/ziglang/zig-pypi'),
-            ('Project-URL', 'Bug Tracker, https://github.com/ziglang/zig-pypi/issues'),
-            ('Requires-Python', '~=3.5'),
+            ("License-Expression", "MIT"),
+            ("License-File", "LICENSE"),
+            ("License-File", "lib/libc/glibc/LICENSES"),
+            ("License-File", "lib/libc/mingw/COPYING"),
+            ("License-File", "lib/libc/musl/COPYRIGHT"),
+            ("License-File", "lib/libc/wasi/LICENSE"),
+            ("License-File", "lib/libc/wasi/LICENSE-APACHE"),
+            ("License-File", "lib/libc/wasi/LICENSE-APACHE-LLVM"),
+            ("License-File", "lib/libc/wasi/LICENSE-MIT"),
+            ("License-File", "lib/libcxx/LICENSE.TXT"),
+            ("License-File", "lib/libcxxabi/LICENSE.TXT"),
+            ("License-File", "lib/libunwind/LICENSE.TXT"),
+            ("Classifier", "Development Status :: 4 - Beta"),
+            ("Classifier", "Intended Audience :: Developers"),
+            ("Classifier", "Topic :: Software Development :: Compilers"),
+            ("Classifier", "Topic :: Software Development :: Code Generators"),
+            ("Classifier", "Topic :: Software Development :: Build Tools"),
+            ("Classifier", "Programming Language :: Other"),
+            ("Classifier", "Programming Language :: Other Scripting Engines"),
+            ("Project-URL", "Homepage, https://ziglang.org"),
+            ("Project-URL", "Source Code, https://github.com/ziglang/zig-pypi"),
+            ("Project-URL", "Bug Tracker, https://github.com/ziglang/zig-pypi/issues"),
+            ("Requires-Python", "~=3.5"),
         ],
         description=description,
         contents=contents,
@@ -218,69 +230,60 @@ def fetch_zig_version_info():
         return json.loads(request.read())
 
 
-def fetch_and_write_ziglang_wheels(
-    outdir='dist/', zig_version='master', wheel_version_suffix='', platforms=tuple()
-):
+def fetch_and_write_ziglang_wheels(outdir="dist/", zig_version="master", wheel_version_suffix="", platforms=tuple()):
     Path(outdir).mkdir(exist_ok=True)
     if not platforms:
         platforms = list(ZIG_PYTHON_PLATFORMS)
     zig_versions_info = fetch_zig_version_info()
 
-    if zig_version == 'latest':
-        zig_version = [version for version in zig_versions_info if version != 'master'][0]
+    if zig_version == "latest":
+        zig_version = [version for version in zig_versions_info if version != "master"][0]
 
     try:
         zig_version_info = zig_versions_info[zig_version]
     except KeyError:
         print(f"Invalid version, valid values: {list(zig_versions_info)}")
         raise
 
-    effective_zig_version = zig_version_info.get('version', zig_version)
+    effective_zig_version = zig_version_info.get("version", zig_version)
 
     for zig_platform in platforms:
         python_platform = ZIG_PYTHON_PLATFORMS[zig_platform]
         if zig_platform not in zig_version_info:
-            print(f"{zig_platform} not present for "
-                  f"version {zig_version} / {effective_zig_version}")
+            print(f"{zig_platform} not present for " f"version {zig_version} / {effective_zig_version}")
             continue
         zig_download = zig_version_info[zig_platform]
-        zig_url = zig_download['tarball']
-        expected_hash = zig_download['shasum']
+        zig_url = zig_download["tarball"]
+        expected_hash = zig_download["shasum"]
 
         with urllib.request.urlopen(zig_url) as request:
             zig_archive = request.read()
             zig_archive_hash = hashlib.sha256(zig_archive).hexdigest()
             if zig_archive_hash != expected_hash:
                 print(zig_download, "SHA256 hash mismatch!")
                 raise AssertionError
-            print(f'{hashlib.sha256(zig_archive).hexdigest()} {zig_url}')
+            print(f"{hashlib.sha256(zig_archive).hexdigest()} {zig_url}")
 
-        wheel_version = effective_zig_version.split('+')[0].replace('-', '.')
-        wheel_path = write_ziglang_wheel(outdir,
-            version=wheel_version + wheel_version_suffix,
-            platform=python_platform,
-            archive=zig_archive)
-        with open(wheel_path, 'rb') as wheel:
-            print(f'  {hashlib.sha256(wheel.read()).hexdigest()} {wheel_path}')
+        wheel_version = effective_zig_version.split("+")[0].replace("-", ".")
+        wheel_path = write_ziglang_wheel(outdir, version=wheel_version + wheel_version_suffix, platform=python_platform, archive=zig_archive)
+        with open(wheel_path, "rb") as wheel:
+            print(f"  {hashlib.sha256(wheel.read()).hexdigest()} {wheel_path}")
 
 
 def get_argparser():
     parser = argparse.ArgumentParser(prog=__file__, description="Repackage official Zig downloads as Python wheels")
-    parser.add_argument('--version', default='latest',
-                        help="version to package, use `latest` for latest release, `master` for nightly build")
-    parser.add_argument('--suffix', default='', help="wheel version suffix")
-    parser.add_argument('--outdir', default='dist/', help="target directory")
-    parser.add_argument('--platform', action='append', choices=list(ZIG_PYTHON_PLATFORMS.keys()), default=[],
-                        help="platform to build for, can be repeated")
+    parser.add_argument("--version", default="latest", help="version to package, use `latest` for latest release, `master` for nightly build")
+    parser.add_argument("--suffix", default="", help="wheel version suffix")
+    parser.add_argument("--outdir", default="dist/", help="target directory")
+    parser.add_argument("--platform", action="append", choices=list(ZIG_PYTHON_PLATFORMS.keys()), default=[], help="platform to build for, can be repeated")
     return parser
 
 
 def main():
     args = get_argparser().parse_args()
     logging.getLogger("wheel").setLevel(logging.WARNING)
-    fetch_and_write_ziglang_wheels(outdir=args.outdir, zig_version=args.version,
-                                   wheel_version_suffix=args.suffix, platforms=args.platform)
+    fetch_and_write_ziglang_wheels(outdir=args.outdir, zig_version=args.version, wheel_version_suffix=args.suffix, platforms=args.platform)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()