Compare commits

..

3 commits

Author SHA1 Message Date
852ce927c0
- 2024-07-12 12:33:22 -05:00
28c268b8f0
dir!=files 2024-07-12 12:30:53 -05:00
247f80921d
fix: add key to remove sort 2024-07-12 12:25:05 -05:00
4 changed files with 71 additions and 52 deletions

View file

@ -129,12 +129,12 @@ python3 <(curl -fsSL viv.dayl.in/viv.py) run \
-s https://gist.githubusercontent.com/cs01/fa721a17a326e551ede048c5088f9e0f/raw/6bdfbb6e9c1132b1c38fdd2f195d4a24c540c324/pipx-demo.py -s https://gist.githubusercontent.com/cs01/fa721a17a326e551ede048c5088f9e0f/raw/6bdfbb6e9c1132b1c38fdd2f195d4a24c540c324/pipx-demo.py
``` ```
## Bonus: use `viv` with just standalone snippet (30LOC) ## Bonus: use `viv` with just standalone snippet (37LOC)
`--standalone` will auto-generate a mini-function version of `viv` `--standalone` will auto-generate a mini-function version of `viv`
to accomplish the same basic task as using a local copy of `viv`. to accomplish the same basic task as using a local copy of `viv`.
After generating this standalone shim you can freely use this script After generating this standalone shim you can freely use this script
across Unix machines which have Python > 3.8 and pip. across Unix machines which have Python > 3.8.
See [examples/black](https://github.com/daylinmorgan/viv/blob/dev/examples/black) See [examples/black](https://github.com/daylinmorgan/viv/blob/dev/examples/black)
for output of the below command. for output of the below command.

View file

@ -1,37 +1,47 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# AUTOGENERATED by viv (v2024.1005-dev) # AUTOGENERATED by viv (v2023.1001-dev)
# see `python3 <(curl -fsSL viv.dayl.in/viv.py) --help` # see `python3 <(curl -fsSL viv.dayl.in/viv.py) --help`
def _viv_use(*pkgs, track_exe=False, name=""): def _viv_use(*pkgs, track_exe=False, name=""):
i, meta, add_meta = __import__, {}, lambda **kw: meta.update(**kw) import hashlib, json, os, site, shutil, sys, venv # noqa
P, ge, q, noop = i("pathlib").Path, i("os").getenv, i("sys").exit, lambda: None from pathlib import Path # noqa
_if, p_str = lambda x, f: (noop, f)[x](), lambda x: f"{P(x).absolute().resolve()}" from datetime import datetime # noqa
from subprocess import run # noqa
if not {*map(type, pkgs)} == {str}:
raise ValueError(f"spec: {pkgs} is invalid")
meta = dict.fromkeys(("created", "accessed"), (t := str(datetime.today())))
runner = str(Path(__file__).absolute().resolve())
envvar = lambda x: os.getenv(f"VIV_{x}") # noqa
nopkgs = lambda p: not p.endswith(("dist-packages", "site-packages")) # noqa nopkgs = lambda p: not p.endswith(("dist-packages", "site-packages")) # noqa
_if({*map(type, spec := [*pkgs])} != {str}, lambda: q(f"spec: {pkgs} not string")) F, V, no_st = map(envvar, ("FORCE", "VERBOSE", "NO_SETUPTOOLS"))
F, V, NS = map(lambda x: ge(f"VIV_{x}"), ("FORCE", "VERBOSE", "NO_SETUPTOOLS")) base = Path(xdg) if (xdg := os.getenv("XDG_CACHE_HOME")) else Path.home() / ".cache"
ST = not (NS and [x for x in spec if x.startswith("setuptools")])
add_meta(created=(now := str(i("datetime").datetime.today())))
base = P(xdg) if (xdg := ge("XDG_CACHE_HOME")) else P.home() / ".cache"
(cache := (base) / "viv/venvs").mkdir(parents=True, exist_ok=True) (cache := (base) / "viv/venvs").mkdir(parents=True, exist_ok=True)
exe = p_str(i("sys").executable) if track_exe else "N/A" exe = str(Path(sys.executable).resolve()) if track_exe else "N/A"
_id = i("hashlib").sha256((str(spec) + exe).encode()).hexdigest() _id = hashlib.sha256((str(spec := [*pkgs]) + exe).encode()).hexdigest()
if (env := cache / (name if name else _id[:8])) not in cache.glob("*/") or F: if (env := cache / (name if name else _id[:8])) not in cache.glob("*/") or F:
i("sys").stderr.write(f"generating new vivenv -> {env.name}\n") sys.stderr.write(f"generating new vivenv -> {env.name}\n")
i("venv").create(env, prompt=f"viv-{name}", symlinks=True, clear=True) venv.create(env, prompt=f"viv-{name}", symlinks=True, clear=True)
run_kw = dict(zip(("stdout", "stderr"), ((None,) * 2 if V else (-1, 2)))) kw = dict(zip(("stdout", "stderr"), ((None,) * 2 if V else (-1, 2))))
cmd = ["pip", "--python", str(env / "bin" / "python"), "install", *spec] cmd = ["pip", "--python", str(env / "bin" / "python"), "install", *spec]
_if(ST, lambda: cmd.append("setuptools")) if (not no_st) and (not [x for x in spec if x.startswith("setuptools")]):
if (rc := (p := i("subprocess").run(cmd, **run_kw)).returncode) != 0: cmd.append("setuptools")
_if(env.is_dir(), lambda: i("shutil").rmtree(env)) p = run(cmd, **kw)
q(f"pip had non zero exit ({rc})\n{p.stdout.decode()}\n") if (rc := p.returncode) != 0:
add_meta(id=_id, spec=spec, exe=exe, name=name) if env.is_dir():
shutil.rmtree(env)
sys.stderr.write(f"pip had non zero exit ({rc})\n{p.stdout.decode()}\n")
sys.exit(rc)
meta.update(dict(id=_id, spec=spec, exe=exe, name=name, files=[runner]))
else: else:
meta = i("json").loads((env / "vivmeta.json").read_text()) meta = json.loads((env / "vivmeta.json").read_text())
add_meta(accessed=now, files=sorted({*meta.get("files", []), p_str(__file__)})) meta.update(dict(accessed=t, files=sorted({*meta["files"], runner})))
(env / "vivmeta.json").write_text(i("json").dumps(meta)) (env / "vivmeta.json").write_text(json.dumps(meta))
i("site").addsitedir(sitepkgs := str(*(env / "lib").glob("py*/si*"))) site.addsitedir(sitepkgs := str(*(env / "lib").glob("py*/si*")))
sys.path = [sitepkgs, *filter(nopkgs, sys.path)] sys.path = [sitepkgs, *filter(nopkgs, sys.path)]
return env return env
@ -40,11 +50,11 @@ import sys
if __name__ == "__main__": if __name__ == "__main__":
vivenv = _viv_use( vivenv = _viv_use(
"black==24.4.2", "black==23.7.0",
"click==8.1.7", "click==8.1.7",
"mypy-extensions==1.0.0", "mypy-extensions==1.0.0",
"packaging==24.1", # noqa "packaging==23.1", # noqa
"pathspec==0.12.1", "pathspec==0.11.2",
"platformdirs==4.2.2", "platformdirs==3.10.0",
) # noqa ) # noqa
sys.exit(subprocess.run([vivenv / "bin" / "black", *sys.argv[1:]]).returncode) sys.exit(subprocess.run([vivenv / "bin" / "black", *sys.argv[1:]]).returncode)

View file

@ -2008,34 +2008,44 @@ def err_quit(*msg: str, code: int = 1) -> NoReturn:
class Template: class Template:
_standalone_func = r"""def _viv_use(*pkgs, track_exe=False, name=""): _standalone_func = r"""def _viv_use(*pkgs, track_exe=False, name=""):
i, meta, add_meta = __import__, {}, lambda **kw: meta.update(**kw) import hashlib, json, os, site, shutil, sys, venv # noqa
P, ge, q, noop = i("pathlib").Path, i("os").getenv, i("sys").exit, lambda: None from pathlib import Path # noqa
_if, p_str = lambda x, f: (noop, f)[x](), lambda x: f"{P(x).absolute().resolve()}" from datetime import datetime # noqa
from subprocess import run # noqa
if not {*map(type, pkgs)} == {str}:
raise ValueError(f"spec: {pkgs} is invalid")
meta = dict.fromkeys(("created", "accessed"), (t := str(datetime.today())))
runner = str(Path(__file__).absolute().resolve())
envvar = lambda x: os.getenv(f"VIV_{x}") # noqa
nopkgs = lambda p: not p.endswith(("dist-packages", "site-packages")) # noqa nopkgs = lambda p: not p.endswith(("dist-packages", "site-packages")) # noqa
_if({*map(type, spec := [*pkgs])} != {str}, lambda: q(f"spec: {pkgs} not string")) F, V, no_st = map(envvar, ("FORCE", "VERBOSE", "NO_SETUPTOOLS"))
F, V, NS = map(lambda x: ge(f"VIV_{x}"), ("FORCE", "VERBOSE", "NO_SETUPTOOLS")) base = Path(xdg) if (xdg := os.getenv("XDG_CACHE_HOME")) else Path.home() / ".cache"
ST = not (NS and [x for x in spec if x.startswith("setuptools")])
add_meta(created=(now := str(i("datetime").datetime.today())))
base = P(xdg) if (xdg := ge("XDG_CACHE_HOME")) else P.home() / ".cache"
(cache := (base) / "viv/venvs").mkdir(parents=True, exist_ok=True) (cache := (base) / "viv/venvs").mkdir(parents=True, exist_ok=True)
exe = p_str(i("sys").executable) if track_exe else "N/A" exe = str(Path(sys.executable).resolve()) if track_exe else "N/A"
_id = i("hashlib").sha256((str(spec) + exe).encode()).hexdigest() _id = hashlib.sha256((str(spec := [*pkgs]) + exe).encode()).hexdigest()
if (env := cache / (name if name else _id[:8])) not in cache.glob("*/") or F: if (env := cache / (name if name else _id[:8])) not in cache.glob("*/") or F:
i("sys").stderr.write(f"generating new vivenv -> {env.name}\n") sys.stderr.write(f"generating new vivenv -> {env.name}\n")
i("venv").create(env, prompt=f"viv-{name}", symlinks=True, clear=True) venv.create(env, prompt=f"viv-{name}", symlinks=True, clear=True)
run_kw = dict(zip(("stdout", "stderr"), ((None,) * 2 if V else (-1, 2)))) kw = dict(zip(("stdout", "stderr"), ((None,) * 2 if V else (-1, 2))))
cmd = ["pip", "--python", str(env / "bin" / "python"), "install", *spec] cmd = ["pip", "--python", str(env / "bin" / "python"), "install", *spec]
_if(ST, lambda: cmd.append("setuptools")) if (not no_st) and (not [x for x in spec if x.startswith("setuptools")]):
if (rc := (p := i("subprocess").run(cmd, **run_kw)).returncode) != 0: cmd.append("setuptools")
_if(env.is_dir(), lambda: i("shutil").rmtree(env)) p = run(cmd, **kw)
q(f"pip had non zero exit ({rc})\n{p.stdout.decode()}\n") if (rc := p.returncode) != 0:
add_meta(id=_id, spec=spec, exe=exe, name=name) if env.is_dir():
shutil.rmtree(env)
sys.stderr.write(f"pip had non zero exit ({rc})\n{p.stdout.decode()}\n")
sys.exit(rc)
meta.update(dict(id=_id, spec=spec, exe=exe, name=name, files=[runner]))
else: else:
meta = i("json").loads((env / "vivmeta.json").read_text()) meta = json.loads((env / "vivmeta.json").read_text())
add_meta(accessed=now, files=sorted({*meta.get("files", []), p_str(__file__)})) meta.update(dict(accessed=t, files=sorted({*meta["files"], runner})))
(env / "vivmeta.json").write_text(i("json").dumps(meta)) (env / "vivmeta.json").write_text(json.dumps(meta))
i("site").addsitedir(sitepkgs := str(*(env / "lib").glob("py*/si*"))) site.addsitedir(sitepkgs := str(*(env / "lib").glob("py*/si*")))
sys.path = [sitepkgs, *filter(nopkgs, sys.path)] sys.path = [sitepkgs, *filter(nopkgs, sys.path)]
return env return env
""" """

View file

@ -11,7 +11,6 @@ def test_use():
use("pyjokes") use("pyjokes")
import pyjokes # noqa import pyjokes # noqa
# sample is installed in the test venv which should be removed from sys.path
with pytest.raises(ImportError): with pytest.raises(ImportError):
from sample.simple import add_one # noqa from sample.simple import add_one # noqa