spacepaste

Paste details

reply | raw

  1.  
  2. #!/usr/bin/env python3
  4. import argparse
  5. import os
  6. import pathlib
  7. import shutil
  8. import subprocess
  9. import sys
  10. import types
  13. class _Registry:
  14. def __init__(self):
  15. self.registered = {}
  17. def __call__(self, function):
  18. self.registered[function.__name__] = function
  19. return function
  22. _register = _Registry()
  25. @_register
  26. def clean(cfg):
  27. """Remove extraneous files."""
  28. paths = [cfg.venv_path, ".coverage"] + list(pathlib.Path().glob(".coverage.*"))
  30. for path in paths:
  31. try:
  32. shutil.rmtree(path)
  33. except FileNotFoundError:
  34. pass
  37. @_register
  38. def init(cfg):
  39. """Set up a virtualenv, install requirements.txt, dev-requirements.txt, and current dir."""
  41. subprocess.run(
  42. ["virtualenv", "--python", sys.executable, cfg.venv_path], check=True
  43. )
  44. lock(cfg)
  45. if pathlib.Path("dev-requirements.txt").exists():
  46. subprocess.run(
  47. [
  48. cfg.venv_path / "bin/pip",
  49. "install",
  50. "--requirement",
  51. "dev-requirements.txt",
  52. ],
  53. check=True,
  54. )
  55. subprocess.run(
  56. [cfg.venv_path / "bin/pip", "install", "--requirement", "requirements.txt"],
  57. check=True,
  58. )
  59. subprocess.run(
  60. [cfg.venv_path / "bin/pip", "install", "--editable", "."], check=True
  61. )
  64. @_register
  65. def lock(cfg):
  66. """Use pip-compile to generate package hashes from setup.py and write them into requirements.txt."""
  67. subprocess.run([cfg.venv_path / "bin/pip", "install", "pip-tools"], check=True)
  68. subprocess.run(
  69. [
  70. cfg.venv_path / "bin/pip-compile",
  71. "--generate-hashes",
  72. "--output-file",
  73. "requirements.txt",
  74. "setup.py",
  75. ],
  76. check=True,
  77. )
  78. subprocess.run(
  79. [
  80. cfg.venv_path / "bin/pip-compile",
  81. "--generate-hashes",
  82. "--output-file",
  83. "dev-requirements.txt",
  84. "setup.py",
  85. "dev-requirements.in",
  86. ],
  87. check=True,
  88. )
  91. @_register
  92. def build(cfg):
  93. """Build source and binary distributions."""
  94. subprocess.run(
  95. [cfg.venv_path / "bin/python", "setup.py", "sdist", "bdist_wheel"], check=True
  96. )
  99. @_register
  100. def upload(cfg):
  101. """Upload the distributions to PyPI."""
  102. subprocess.run([cfg.venv_path / "bin/python", "-m", "pip", "install", "twine"])
  103. dists = [str(path) for path in pathlib.Path("dist").iterdir()]
  104. subprocess.run([cfg.venv_path / "bin/twine", "upload", *dists], check=True)
  107. def _get_default_venv_path():
  108. """Get the default path of the venv."""
  109. venv_path = os.environ.get("VENV_PATH")
  110. if venv_path:
  111. return pathlib.Path(venv_path)
  113. workon_home = os.environ.get("WORKON_HOME")
  114. if workon_home is not None:
  115. project_name = pathlib.Path(os.getcwd()).name
  116. return pathlib.Path(workon_home) / project_name
  118. return pathlib.Path().resolve() / "venv"
  121. def cli():
  122. parser = argparse.ArgumentParser()
  124. parser.add_argument(
  125. "--venv",
  126. default=_get_default_venv_path(),
  127. type=pathlib.Path,
  128. help="Path of the venv directory. Defaults, in order: $VENV_PATH, $WORKON_HOME/[current directory name], venv",
  129. )
  131. subparsers = parser.add_subparsers(dest="command_name")
  132. for name, function in _register.registered.items():
  133. subparsers.add_parser(name, help=function.__doc__)
  135. args = parser.parse_args()
  136. if args.command_name is None:
  137. parser.print_help()
  138. sys.exit(2)
  140. function = _register.registered[args.command_name]
  141. cfg = types.SimpleNamespace()
  142. cfg.venv_path = args.venv
  143. function(cfg)
  146. if __name__ == "__main__":
  147. cli()
  148.  
Powered by spacepaste. Paste removal.