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

omdev/ci/github/cli.py

@@ -0,0 +1,39 @@
+# ruff: noqa: UP006 UP007
+# @omlish-lite
+"""
+See:
+ - https://docs.github.com/en/rest/actions/cache?apiVersion=2022-11-28
+"""
+import dataclasses as dc
+
+from omlish.argparse.cli import ArgparseCli
+from omlish.argparse.cli import argparse_arg
+from omlish.argparse.cli import argparse_cmd
+from omlish.lite.json import json_dumps_pretty
+
+from .cache import GithubV1CacheShellClient
+
+
+class GithubCli(ArgparseCli):
+    @argparse_cmd(
+        argparse_arg('key'),
+    )
+    def get_cache_entry(self) -> None:
+        shell_client = GithubV1CacheShellClient()
+        entry = shell_client.run_get_entry(self.args.key)
+        if entry is None:
+            return
+        print(json_dumps_pretty(dc.asdict(entry)))  # noqa
+
+    @argparse_cmd(
+        argparse_arg('repository-id'),
+    )
+    def list_cache_entries(self) -> None:
+        raise NotImplementedError
+
+
+if __name__ == '__main__':
+    def _main() -> None:
+        GithubCli().cli_run_and_exit()
+
+    _main()

omdev/ci/requirements.py

@@ -65,6 +65,7 @@ def download_requirements(
         subprocesses.check_call(
             'docker',
             'run',
+            '--rm',
             '-i',
             '-v', f'{os.path.abspath(requirements_dir)}:/requirements',
             '-v', f'{requirements_txt_dir}:/requirements_txt',

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

@@ -1,10 +1,14 @@
 # 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
+
 
 ##
 
@@ -30,3 +34,48 @@ def read_yaml_file(yaml_file: str) -> ta.Any:
 
 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