GeezerVison is showing github.com — pyperformance/dev.py at main · Python-Repository-Hub/pyperformance

Expand file tree

dev.py

Copy path

More file actions

Latest commit

History

84 lines (70 loc) · 2.76 KB

dev.py

Copy path

File metadata and controls

84 lines (70 loc) · 2.76 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

# 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()