dev.py

# A script for running pyperformance out of the repo in dev-mode.

import os.path
import sys


REPO_ROOT = os.path.dirname(os.path.abspath(__file__))
VENVS = os.path.join(REPO_ROOT, '.venvs')


def resolve_venv_root(kind='dev', venvsdir=VENVS):
    import sysconfig
    if sysconfig.is_python_build():
        sys.exit('please install your built Python first (or pass it using --python)')
    # XXX Handle other implementations too?
    base = os.path.join(venvsdir, kind or 'dev')
    major, minor = sys.version_info[:2]
    pyloc = ((os.path.abspath(sys.executable)
              ).partition(os.path.sep)[2].lstrip(os.path.sep)
             ).replace(os.path.sep, '-')
    return f'{base}-{major}.{minor}-{pyloc}'


def ensure_venv_ready(venvroot=None, kind='dev', venvsdir=VENVS):
    if sys.prefix != sys.base_prefix:
        assert os.path.exists(os.path.join(sys.prefix, 'pyvenv.cfg'))
        venvroot = sys.prefix
        python = sys.executable
        readyfile = os.path.join(sys.prefix, 'READY')
        isready = os.path.exists(readyfile)
    else:
        import venv
        if not venvroot:
            venvroot = resolve_venv_root(kind, venvsdir)
        # Make sure the venv exists.
        readyfile = os.path.join(venvroot, 'READY')
        isready = os.path.exists(readyfile)
        if not isready:
            relroot = os.path.relpath(venvroot)
            if not os.path.exists(venvroot):
                print(f'creating venv at {relroot}...')
            else:
                print(f'venv {relroot} not ready, re-creating...')
            venv.create(venvroot, with_pip=True, clear=True)
        else:
            assert os.path.exists(os.path.join(venvroot, 'pyvenv.cfg'))
        # Return the venv's Python executable.
        binname = 'Scripts' if os.name == 'nt' else 'bin'
        exename = os.path.basename(sys.executable)
        python = os.path.join(venvroot, binname, exename)

    # Now make sure the venv has pyperformance installed.
    if not isready:
        import subprocess
        relroot = os.path.relpath(venvroot)
        print(f'venv {relroot} not ready, installing dependencies...')
        proc = subprocess.run(
            [python, '-m', 'pip', 'install',
             '--upgrade',
             '--editable', REPO_ROOT],
        )
        if proc.returncode != 0:
            sys.exit('ERROR: install failed')
        with open(readyfile, 'w'):
            pass
        print('...venv {relroot} ready!')

    return venvroot, python


def main(venvroot=None):
    _, python = ensure_venv_ready(venvroot)
    if python != sys.executable:
        # Now re-run using the venv.
        os.execv(python, [python, *sys.argv])
        # <unreachable>

    # Now run pyperformance.
    import pyperformance.cli
    pyperformance.cli.main()


if __name__ == '__main__':
    main()