python/scripts/vendor.py

   1 #!/usr/bin/env python3
   2 """
   3 vendor - QEMU python vendoring utility
   4
   5 usage: vendor [-h]
   6
   7 QEMU python vendoring utility
   8
   9 options:
  10   -h, --help  show this help message and exit
  11 """
  12
  13 # Copyright (C) 2023 Red Hat, Inc.
  14 #
  15 # Authors:
  16 #  John Snow <jsnow@redhat.com>
  17 #
  18 # This work is licensed under the terms of the GNU GPL, version 2 or
  19 # later. See the COPYING file in the top-level directory.
  20
  21 import argparse
  22 import os
  23 from pathlib import Path
  24 import subprocess
  25 import sys
  26 import tempfile
  27
  28
  29 def main() -> int:
  30     """Run the vendoring utility. See module-level docstring."""
  31     loud = False
  32     if os.environ.get("DEBUG") or os.environ.get("V"):
  33         loud = True
  34
  35     # No options or anything for now, but I guess
  36     # you'll figure that out when you run --help.
  37     parser = argparse.ArgumentParser(
  38         prog="vendor",
  39         description="QEMU python vendoring utility",
  40     )
  41     parser.parse_args()
  42
  43     packages = {
  44         "meson==0.63.3":
  45         "d677b809c4895dcbaac9bf6c43703fcb3609a4b24c6057c78f828590049cf43a",
  46
  47         "tomli==2.0.1":
  48         "939de3e7a6161af0c887ef91b7d41a53e7c5a1ca976325f429cb46ea9bc30ecc",
  49     }
  50
  51     vendor_dir = Path(__file__, "..", "..", "wheels").resolve()
  52
  53     with tempfile.NamedTemporaryFile(mode="w", encoding="utf-8") as file:
  54         for dep_spec, checksum in packages.items():
  55             print(f"{dep_spec} --hash=sha256:{checksum}", file=file)
  56         file.flush()
  57
  58         cli_args = [
  59             "pip",
  60             "download",
  61             "--dest",
  62             str(vendor_dir),
  63             "--require-hashes",
  64             "-r",
  65             file.name,
  66         ]
  67         if loud:
  68             cli_args.append("-v")
  69
  70         print(" ".join(cli_args))
  71         subprocess.run(cli_args, check=True)
  72
  73     return 0
  74
  75
  76 if __name__ == "__main__":
  77     sys.exit(main())