diff --git a/spcd/__init__.py b/spcd/__init__.py index 4a81720..3532b60 100644 --- a/spcd/__init__.py +++ b/spcd/__init__.py @@ -3,8 +3,10 @@ __version__ = "0.0.1" import sys +from os import environ, pathsep from pathlib import Path +import env from rwx import fs from rwx.log import stream as log from rwx.ps import run @@ -61,6 +63,9 @@ def list_environment_variables() -> None: def main(main: str) -> None: + paths = environ["PATH"].split(pathsep) + if env.SPCD_PYTHON_VENV_BINARIES not in paths: + environ["PATH"] = pathsep.join([env.SPCD_PYTHON_VENV_BINARIES, *paths]) path, *arguments = sys.argv name = Path(path).name if name == "__main__.py": diff --git a/spcd/__main__.py b/spcd/__main__.py index 6c62422..1c82dba 100755 --- a/spcd/__main__.py +++ b/spcd/__main__.py @@ -3,7 +3,6 @@ """Entry point.""" import sys -from os import environ, pathsep import env @@ -12,7 +11,4 @@ if env.SPCD_PYTHON_VENV_PACKAGES not in sys.path: import spcd if __name__ == "__main__": - paths = environ["PATH"].split(pathsep) - if env.SPCD_PYTHON_VENV_BINARIES not in paths: - environ["PATH"] = pathsep.join([env.SPCD_PYTHON_VENV_BINARIES, *paths]) spcd.main(__file__)