71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
from timeit import default_timer
|
|
|
|
|
|
def run(args=None, options=None, env=None):
|
|
env = os.environ if env is None else env
|
|
start = default_timer()
|
|
from virtualenv.run import cli_run # noqa: PLC0415
|
|
from virtualenv.util.error import ProcessCallFailedError # noqa: PLC0415
|
|
|
|
if args is None:
|
|
args = sys.argv[1:]
|
|
try:
|
|
session = cli_run(args, options, env)
|
|
logging.warning(LogSession(session, start))
|
|
except ProcessCallFailedError as exception:
|
|
print(f"subprocess call failed for {exception.cmd} with code {exception.code}") # noqa: T201
|
|
print(exception.out, file=sys.stdout, end="") # noqa: T201
|
|
print(exception.err, file=sys.stderr, end="") # noqa: T201
|
|
raise SystemExit(exception.code) # noqa: B904
|
|
|
|
|
|
class LogSession:
|
|
def __init__(self, session, start) -> None:
|
|
self.session = session
|
|
self.start = start
|
|
|
|
def __str__(self) -> str:
|
|
spec = self.session.creator.interpreter.spec
|
|
elapsed = (default_timer() - self.start) * 1000
|
|
lines = [
|
|
f"created virtual environment {spec} in {elapsed:.0f}ms",
|
|
f" creator {self.session.creator!s}",
|
|
]
|
|
if self.session.seeder.enabled:
|
|
lines.append(f" seeder {self.session.seeder!s}")
|
|
path = self.session.creator.purelib.iterdir()
|
|
packages = sorted("==".join(i.stem.split("-")) for i in path if i.suffix == ".dist-info")
|
|
lines.append(f" added seed packages: {', '.join(packages)}")
|
|
|
|
if self.session.activators:
|
|
lines.append(f" activators {','.join(i.__class__.__name__ for i in self.session.activators)}")
|
|
return "\n".join(lines)
|
|
|
|
|
|
def run_with_catch(args=None, env=None):
|
|
from virtualenv.config.cli.parser import VirtualEnvOptions # noqa: PLC0415
|
|
|
|
env = os.environ if env is None else env
|
|
options = VirtualEnvOptions()
|
|
try:
|
|
run(args, options, env)
|
|
except (KeyboardInterrupt, SystemExit, Exception) as exception:
|
|
try:
|
|
if getattr(options, "with_traceback", False):
|
|
raise
|
|
if not (isinstance(exception, SystemExit) and exception.code == 0):
|
|
logging.error("%s: %s", type(exception).__name__, exception) # noqa: TRY400
|
|
code = exception.code if isinstance(exception, SystemExit) else 1
|
|
sys.exit(code)
|
|
finally:
|
|
logging.shutdown() # force flush of log messages before the trace is printed
|
|
|
|
|
|
if __name__ == "__main__": # pragma: no cov
|
|
run_with_catch() # pragma: no cov
|