daylin/viv
1
0
Fork 0
mirror of https://github.com/daylinmorgan/viv.git synced 2024-11-12 12:13:15 -06:00
Code Issues Projects Releases Packages Wiki Activity

refactor: more consolidation

Browse source
This commit is contained in:
Daylin Morgan 2023-08-22 21:41:59 -05:00
parent 78e533b0a2
commit 0136568137
Signed by: daylin
GPG key ID: C1E52E7DD81DF79F
1 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/viv/viv.py
View file

@ -52,7 +52,7 @@ from typing import (
Union,
)
__version__ = "23.8b2-11-gf522352-dev"
__version__ = "23.8b2-13-g1f0eed9-dev"
class Spinner:
@ -1006,7 +1006,7 @@ class ViVenv:
)
def ensure(self) -> None:
self.existse()
self.exists()
if not self.loaded or Env().viv_force:
self.create()
self.install_pkgs()
@ -1029,7 +1029,7 @@ class ViVenv:
*(
p
for p in sys.path
if not any(map(p.endswith, ("dist-packages", "site-packages")))
if not p.endswith(("dist-packages", "site-packages"))
),
]
site.addsitedir(path_to_add)
@ -1048,21 +1048,24 @@ class ViVenv:
self.size = f"{size:.1f}{unit}B"
# TODO: reconsider this function which is starting to do heavy lifting
@contextmanager
def use(self, keep: bool = True) -> Generator[None, None, None]:
run_mode = Env().viv_run_mode
_path = self.path
def common():
self.ensure()
self.touch()
try:
if self.loaded:
self.ensure()
yield
elif keep or run_mode == "persist":
self.ensure()
if self.loaded or keep or run_mode == "persist":
common()
yield
elif run_mode == "ephemeral":
with tempfile.TemporaryDirectory(prefix="viv-") as tmpdir:
self.set_path(Path(tmpdir))
self.ensure()
common()
yield
elif run_mode == "semi-ephemeral":
ephemeral_cache = _path_ok(
@ -1070,7 +1073,7 @@ class ViVenv:
)
os.environ.update(dict(VIV_CACHE=str(ephemeral_cache)))
self.set_path(ephemeral_cache / "venvs" / self.name)
self.ensure()
common()
yield
finally:
self.set_path(_path)