setup_dev.py

"""
This script is used to setup development environment of pdm. It follows the naming
convention of setuptools but it is not basically a setuptools setup script.

After running this script, an editable version of pdm will be installed into
`__packages__`.
"""
import os
import shutil
import subprocess
import sys
from pathlib import Path

BASE_DIR = Path(__file__).parent
if sys.version_info < (3, 7):
    sys.exit("PDM requires Python 3.7 or higher.")


def main():
    venv_path = BASE_DIR / "env"
    scripts_dir = "Scripts" if os.name == "nt" else "bin"
    venv_python = venv_path / scripts_dir / "python"

    print(f"Creating a venv using {sys.executable} at {venv_path}...")
    subprocess.check_call([sys.executable, "-m", "venv", venv_path.as_posix()])

    print("Installing base requirements...")
    subprocess.check_call([venv_python.as_posix(), "-m", "pip", "install", "pdm"])

    print("Setup project for development...")
    subprocess.check_call([venv_python.as_posix(), "-m", "pdm", "install", "-d"])

    pdm_path = (
        BASE_DIR
        / "__pypackages__"
        / ".".join(sys.version_info[:2])
        / scripts_dir
        / "pdm"
    )

    print(f"\nDeleting venv {venv_path}...")
    shutil.rmtree(venv_path, ignore_errors=True)

    print(
        f"An editable version of pdm is installed at {pdm_path}, "
        "you can create an alias for it for convenience."
    )


if __name__ == "__main__":
    main()