omdev 0.0.0.dev209__py3-none-any.whl → 0.0.0.dev211__py3-none-any.whl

omdev/ci/shell.py

@@ -0,0 +1,42 @@
+# ruff: noqa: UP006 UP007
+# @omlish-lite
+import dataclasses as dc
+import os
+import typing as ta
+
+
+T = ta.TypeVar('T')
+
+
+##
+
+
+@dc.dataclass(frozen=True)
+class ShellCmd:
+    s: str
+
+    env: ta.Optional[ta.Mapping[str, str]] = None
+
+    def build_run_kwargs(
+            self,
+            *,
+            env: ta.Optional[ta.Mapping[str, str]] = None,
+            **kwargs: ta.Any,
+    ) -> ta.Dict[str, ta.Any]:
+        if env is None:
+            env = os.environ
+        if self.env:
+            if (ek := set(env) & set(self.env)):
+                raise KeyError(*ek)
+            env = {**env, **self.env}
+
+        return dict(
+            env=env,
+            **kwargs,
+        )
+
+    def run(self, fn: ta.Callable[..., T], **kwargs) -> T:
+        return fn(
+            'sh', '-c', self.s,
+            **self.build_run_kwargs(**kwargs),
+        )

omdev/ci/utils.py

@@ -0,0 +1,81 @@
+# ruff: noqa: UP006 UP007
+# @omlish-lite
+import hashlib
+import logging
+import os.path
+import tempfile
+import time
+import typing as ta
+
+from omlish.lite.logs import log
+
+
+##
+
+
+def make_temp_file() -> str:
+    file_fd, file = tempfile.mkstemp()
+    os.close(file_fd)
+    return file
+
+
+##
+
+
+def read_yaml_file(yaml_file: str) -> ta.Any:
+    yaml = __import__('yaml')
+
+    with open(yaml_file) as f:
+        return yaml.safe_load(f)
+
+
+##
+
+
+def sha256_str(s: str) -> str:
+    return hashlib.sha256(s.encode('utf-8')).hexdigest()
+
+
+##
+
+
+class LogTimingContext:
+    DEFAULT_LOG: ta.ClassVar[logging.Logger] = log
+
+    def __init__(
+            self,
+            description: str,
+            *,
+            log: ta.Optional[logging.Logger] = None,  # noqa
+            level: int = logging.DEBUG,
+    ) -> None:
+        super().__init__()
+
+        self._description = description
+        self._log = log if log is not None else self.DEFAULT_LOG
+        self._level = level
+
+    def set_description(self, description: str) -> 'LogTimingContext':
+        self._description = description
+        return self
+
+    _begin_time: float
+    _end_time: float
+
+    def __enter__(self) -> 'LogTimingContext':
+        self._begin_time = time.time()
+
+        self._log.log(self._level, f'Begin {self._description}')  # noqa
+
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self._end_time = time.time()
+
+        self._log.log(
+            self._level,
+            f'End {self._description} - {self._end_time - self._begin_time:0.2f} s elapsed',
+        )
+
+
+log_timing_context = LogTimingContext

omdev/interp/cli.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python3
 # @omlish-amalg ../scripts/interp.py
 # ruff: noqa: UP007
 """

omdev/interp/providers/system.py

@@ -104,7 +104,8 @@ class SystemInterpProvider(InterpProvider):
     async def get_exe_version(self, exe: str) -> ta.Optional[InterpVersion]:
         if not self._options.inspect:
             s = os.path.basename(exe)
-            s = s.removeprefix('python')
+            if s.startswith('python'):  # noqa
+                s = s[len('python'):]
             if '.' in s:
                 try:
                     return InterpVersion.parse(s)

omdev/pycharm/cli.py

@@ -88,7 +88,7 @@ class Cli(ap.Cli):
 
     @ap.cmd(
         ap.arg('dir', nargs='?'),
-        ap.arg('--clion', action='store_true'),
+        ap.arg('-c', '--clion', action='store_true'),
     )
     def open(self) -> None:
         dir = os.path.abspath(self.args.dir or '.')  # noqa

omdev/pyproject/cli.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python3
 # @omlish-amalg ../scripts/pyproject.py
 # ruff: noqa: UP006 UP007
 """

omdev/revisions.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python3
 """
 TODO:
  - omlish-lite, move to pyproject/