ominfra 0.0.0.dev384__py3-none-any.whl → 0.0.0.dev386__py3-none-any.whl

ominfra/scripts/manage.py

@@ -155,76 +155,6 @@ SubprocessChannelOption = ta.Literal['pipe', 'stdout', 'devnull']  # ta.TypeAlia
 SystemPackageOrStr = ta.Union['SystemPackage', str]  # ta.TypeAlias
 
 
-########################################
-# ../../../omdev/home/paths.py
-"""
-TODO:
- - XDG cache root
-"""
-
-
-##
-
-
-HOME_DIR_ENV_VAR = 'OMLISH_HOME'
-DEFAULT_HOME_DIR = '~/.omlish'
-
-
-def get_home_dir() -> str:
-    return os.path.expanduser(os.getenv(HOME_DIR_ENV_VAR, DEFAULT_HOME_DIR))
-
-
-#
-
-
-class HomePaths:
-    def __init__(self, home_dir: ta.Optional[str] = None) -> None:
-        super().__init__()
-
-        if home_dir is None:
-            home_dir = get_home_dir()
-        self._home_dir = home_dir
-
-    @property
-    def home_dir(self) -> str:
-        return self._home_dir
-
-    @property
-    def config_dir(self) -> str:
-        return os.path.join(self._home_dir, 'config')
-
-    @property
-    def log_dir(self) -> str:
-        return os.path.join(self._home_dir, 'log')
-
-    @property
-    def run_dir(self) -> str:
-        return os.path.join(self._home_dir, 'run')
-
-    @property
-    def shadow_dir(self) -> str:
-        return os.path.join(self._home_dir, 'shadow')
-
-    @property
-    def state_dir(self) -> str:
-        return os.path.join(self._home_dir, 'state')
-
-
-def get_home_paths() -> HomePaths:
-    return HomePaths()
-
-
-##
-
-
-CACHE_DIR_ENV_VAR = 'OMLISH_CACHE'
-DEFAULT_CACHE_DIR = '~/.cache/omlish'
-
-
-def get_cache_dir() -> str:
-    return os.path.expanduser(os.getenv(DEFAULT_CACHE_DIR, DEFAULT_CACHE_DIR))
-
-
 ########################################
 # ../../../omdev/packaging/versions.py
 # Copyright (c) Donald Stufft and individual contributors.
@@ -3867,6 +3797,158 @@ def set_process_deathsig(sig: int) -> bool:
         return False
 
 
+########################################
+# ../../../omlish/os/environ.py
+
+
+##
+
+
+class EnvVar:
+    def __init__(self, name: str) -> None:
+        super().__init__()
+
+        if not isinstance(name, str):
+            raise TypeError(name)
+        if not name:
+            raise NameError(name)
+        self._name = name
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    def __init_subclass__(cls, **kwargs):
+        raise TypeError
+
+    def __repr__(self) -> str:
+        return f'{self.__class__.__name__}({self._name!r})'
+
+    def __hash__(self) -> int:
+        return hash((self.__class__, self._name))
+
+    def __eq__(self, other):
+        if not isinstance(other, EnvVar):
+            return NotImplemented
+        return self._name == other._name
+
+    def __bool__(self) -> bool:
+        raise TypeError
+
+    def __str__(self) -> str:
+        raise TypeError
+
+    #
+
+    @classmethod
+    def _get_mut_environ(
+            cls,
+            environ: ta.Optional[ta.MutableMapping[str, str]] = None,
+    ) -> ta.MutableMapping[str, str]:
+        if environ is None:
+            return os.environ
+        return environ
+
+    @classmethod
+    def _get_environ(
+            cls,
+            environ: ta.Optional[ta.Mapping[str, str]] = None,
+    ) -> ta.Mapping[str, str]:
+        if environ is None:
+            return cls._get_mut_environ()
+        return environ
+
+    #
+
+    def is_set(self, environ: ta.Optional[ta.Mapping[str, str]] = None) -> bool:
+        return self._name in self._get_environ(environ)
+
+    _NO_DEFAULT: ta.ClassVar[ta.Any] = object()
+
+    @ta.overload
+    def get(
+            self,
+            *,
+            environ: ta.Optional[ta.Mapping[str, str]] = None,
+    ) -> str:
+        ...
+
+    @ta.overload
+    def get(
+            self,
+            default: str,
+            /,
+            *,
+            environ: ta.Optional[ta.Mapping[str, str]] = None,
+    ) -> str:
+        ...
+
+    @ta.overload
+    def get(
+            self,
+            default: ta.Optional[str],
+            /,
+            *,
+            environ: ta.Optional[ta.Mapping[str, str]] = None,
+    ) -> ta.Optional[str]:
+        ...
+
+    @ta.overload
+    def get(
+            self,
+            default: ta.Callable[[], str],
+            /,
+            *,
+            environ: ta.Optional[ta.Mapping[str, str]] = None,
+    ) -> str:
+        ...
+
+    @ta.overload
+    def get(
+            self,
+            default: ta.Callable[[], ta.Optional[str]],
+            /,
+            *,
+            environ: ta.Optional[ta.Mapping[str, str]] = None,
+    ) -> ta.Optional[str]:
+        ...
+
+    def get(self, default=_NO_DEFAULT, /, *, environ=None):
+        environ = self._get_environ(environ)
+        try:
+            return environ[self._name]
+        except KeyError:
+            if default is self._NO_DEFAULT:
+                raise
+            elif callable(default):
+                return default()
+            else:
+                return default
+
+    #
+
+    def set(self, value: ta.Optional[str], *, environ: ta.Optional[ta.MutableMapping[str, str]] = None) -> None:
+        environ = self._get_mut_environ(environ)
+        if value is not None:
+            environ[self._name] = value
+        else:
+            del environ[self._name]
+
+    @contextlib.contextmanager
+    def set_context(
+            self,
+            value: ta.Optional[str],
+            *,
+            environ: ta.Optional[ta.MutableMapping[str, str]] = None,
+    ) -> ta.Iterator[ta.Optional[str]]:
+        prev = self.get(None, environ=environ)
+        self.set(value, environ=environ)
+        try:
+            yield prev
+        finally:
+            self.set(prev, environ=environ)
+
+
 ########################################
 # ../../../omlish/os/linux.py
 """
@@ -4435,6 +4517,76 @@ def shlex_maybe_quote(s: str) -> str:
         return s
 
 
+########################################
+# ../../../omdev/home/paths.py
+"""
+TODO:
+ - XDG cache root
+"""
+
+
+##
+
+
+HOME_DIR_ENV_VAR = EnvVar('OMLISH_HOME')
+DEFAULT_HOME_DIR = '~/.omlish'
+
+
+def get_home_dir() -> str:
+    return os.path.expanduser(HOME_DIR_ENV_VAR.get(DEFAULT_HOME_DIR))
+
+
+#
+
+
+class HomePaths:
+    def __init__(self, home_dir: ta.Optional[str] = None) -> None:
+        super().__init__()
+
+        if home_dir is None:
+            home_dir = get_home_dir()
+        self._home_dir = home_dir
+
+    @property
+    def home_dir(self) -> str:
+        return self._home_dir
+
+    @property
+    def config_dir(self) -> str:
+        return os.path.join(self._home_dir, 'config')
+
+    @property
+    def log_dir(self) -> str:
+        return os.path.join(self._home_dir, 'log')
+
+    @property
+    def run_dir(self) -> str:
+        return os.path.join(self._home_dir, 'run')
+
+    @property
+    def shadow_dir(self) -> str:
+        return os.path.join(self._home_dir, 'shadow')
+
+    @property
+    def state_dir(self) -> str:
+        return os.path.join(self._home_dir, 'state')
+
+
+def get_home_paths() -> HomePaths:
+    return HomePaths()
+
+
+##
+
+
+CACHE_DIR_ENV_VAR = 'OMLISH_CACHE'
+DEFAULT_CACHE_DIR = '~/.cache/omlish'
+
+
+def get_cache_dir() -> str:
+    return os.path.expanduser(os.getenv(DEFAULT_CACHE_DIR, DEFAULT_CACHE_DIR))
+
+
 ########################################
 # ../../../omdev/packaging/specifiers.py
 # Copyright (c) Donald Stufft and individual contributors.

{ominfra-0.0.0.dev384.dist-info → ominfra-0.0.0.dev386.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ominfra
-Version: 0.0.0.dev384
+Version: 0.0.0.dev386
 Summary: ominfra
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -14,8 +14,8 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omdev==0.0.0.dev384
-Requires-Dist: omlish==0.0.0.dev384
+Requires-Dist: omdev==0.0.0.dev386
+Requires-Dist: omlish==0.0.0.dev386
 Provides-Extra: all
 Requires-Dist: paramiko~=4.0; extra == "all"
 Requires-Dist: asyncssh~=2.21; extra == "all"

{ominfra-0.0.0.dev384.dist-info → ominfra-0.0.0.dev386.dist-info}/RECORD

@@ -113,7 +113,7 @@ ominfra/manage/targets/inject.py,sha256=3M4wBkxtvymq_yhiotHlTN8iydELMjVCndyp9Bq-
 ominfra/manage/targets/targets.py,sha256=WmasYmL6xfAI7F0XvDhScFdBVxkqPkmo_gRgABmmkGA,1891
 ominfra/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ominfra/scripts/journald2aws.py,sha256=WOyM_3Em7F_DKB7PaGYD3hUD1L3dnvpZXNJjmNUb4xc,173849
-ominfra/scripts/manage.py,sha256=GNzN34VDtwqQuB3I1KG4wEvWvlnSYW30XOlVO0E5N7s,392468
+ominfra/scripts/manage.py,sha256=RIFrfK6yE1WdWajuJUpdA4acIiICozS_x_k5XaK2M4w,396142
 ominfra/scripts/supervisor.py,sha256=1x2J9JmRSa6uQsIHRYK3fXp_nOR-NS7akuV55__Pf-U,306331
 ominfra/supervisor/LICENSE.txt,sha256=ZrHY15PVR98y26Yg6iQfa-SXnUaYTDhrUsPVcEO5OKM,1874
 ominfra/supervisor/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
@@ -156,9 +156,9 @@ ominfra/tailscale/api.py,sha256=XASv9C_CWI-u-yX5jVzhJrkJhlwQRkYQWQQG1uJwAd8,1375
 ominfra/tailscale/cli.py,sha256=zRV7-tKB7kBah1oTVZlol-vwx1FBlnfzYAPGkeU5jX4,3543
 ominfra/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ominfra/tools/listresources.py,sha256=auGP1LlbBJSFKUWNvQo_UzA8IsBNZBTMwEkFFRJ4FX4,6185
-ominfra-0.0.0.dev384.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-ominfra-0.0.0.dev384.dist-info/METADATA,sha256=yU_RXCooNz88zTZmKkmNK8ErpbgKQuLwTmpG8YJzJjc,2377
-ominfra-0.0.0.dev384.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ominfra-0.0.0.dev384.dist-info/entry_points.txt,sha256=kgecQ2MgGrM9qK744BoKS3tMesaC3yjLnl9pa5CRczg,37
-ominfra-0.0.0.dev384.dist-info/top_level.txt,sha256=E-b2OHkk_AOBLXHYZQ2EOFKl-_6uOGd8EjeG-Zy6h_w,8
-ominfra-0.0.0.dev384.dist-info/RECORD,,
+ominfra-0.0.0.dev386.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+ominfra-0.0.0.dev386.dist-info/METADATA,sha256=MX-Ldv3ZPxZNvzvq1KuzcXLdcFNoJc-ANKW4lg7s7tA,2377
+ominfra-0.0.0.dev386.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ominfra-0.0.0.dev386.dist-info/entry_points.txt,sha256=kgecQ2MgGrM9qK744BoKS3tMesaC3yjLnl9pa5CRczg,37
+ominfra-0.0.0.dev386.dist-info/top_level.txt,sha256=E-b2OHkk_AOBLXHYZQ2EOFKl-_6uOGd8EjeG-Zy6h_w,8
+ominfra-0.0.0.dev386.dist-info/RECORD,,