mm-std 0.4.3__py3-none-any.whl → 0.4.4__py3-none-any.whl

mm_std/config.py

@@ -1,7 +1,7 @@
 import sys
 import tomllib
 from pathlib import Path
-from typing import NoReturn, Self
+from typing import Any, NoReturn, Self, TypeVar
 
 from pydantic import BaseModel, ConfigDict, ValidationError
 
@@ -9,6 +9,8 @@ from .print_ import print_json, print_plain
 from .result import Result
 from .zip import read_text_from_zip_archive
 
+T = TypeVar("T", bound="BaseConfig")
+
 
 class BaseConfig(BaseModel):
     model_config = ConfigDict(extra="forbid")
@@ -18,21 +20,18 @@ class BaseConfig(BaseModel):
         sys.exit(0)
 
     @classmethod
-    def read_toml_config_or_exit[T](cls: type[T], config_path: Path, zip_password: str = "") -> T:  # noqa: PYI019 # nosec
-        res: Result[T] = cls.read_toml_config(config_path, zip_password)  # type:ignore[attr-defined]
+    def read_toml_config_or_exit(cls, config_path: Path, zip_password: str = "") -> Self:  # nosec
+        res: Result[Self] = cls.read_toml_config(config_path, zip_password)
         if res.is_ok():
             return res.unwrap()
+        cls._print_error_and_exit(res)
 
-        if res.error == "validator_error" and res.extra:
-            print_plain("config validation errors")
-            for e in res.extra["errors"]:
-                loc = e["loc"]
-                field = ".".join(str(lo) for lo in loc) if len(loc) > 0 else ""
-                print_plain(f"{field} {e['msg']}")
-        else:
-            print_plain(f"can't parse config file: {res.error}")
-
-        sys.exit(1)
+    @classmethod
+    async def read_toml_config_or_exit_async(cls, config_path: Path, zip_password: str = "") -> Self:  # nosec
+        res: Result[Self] = await cls.read_toml_config_async(config_path, zip_password)
+        if res.is_ok():
+            return res.unwrap()
+        cls._print_error_and_exit(res)
 
     @classmethod
     def read_toml_config(cls, config_path: Path, zip_password: str = "") -> Result[Self]:  # nosec
@@ -48,3 +47,31 @@ class BaseConfig(BaseModel):
             return Result.failure(("validator_error", e), extra={"errors": e.errors()})
         except Exception as e:
             return Result.failure(e)
+
+    @classmethod
+    async def read_toml_config_async(cls, config_path: Path, zip_password: str = "") -> Result[Self]:  # nosec
+        try:
+            config_path = config_path.expanduser()
+            if config_path.name.endswith(".zip"):
+                data = tomllib.loads(read_text_from_zip_archive(config_path, password=zip_password))
+            else:
+                with config_path.open("rb") as f:
+                    data = tomllib.load(f)
+            model = await cls.model_validate_async(data)  # type:ignore[attr-defined]
+            return Result.success(model)
+        except ValidationError as e:
+            return Result.failure(("validator_error", e), extra={"errors": e.errors()})
+        except Exception as e:
+            return Result.failure(e)
+
+    @classmethod
+    def _print_error_and_exit(cls, res: Result[Any]) -> NoReturn:
+        if res.error == "validator_error" and res.extra:
+            print_plain("config validation errors")
+            for e in res.extra["errors"]:
+                loc = e["loc"]
+                field = ".".join(str(lo) for lo in loc) if len(loc) > 0 else ""
+                print_plain(f"{field} {e['msg']}")
+        else:
+            print_plain(f"can't parse config file: {res.error}")
+        sys.exit(1)

{mm_std-0.4.3.dist-info → mm_std-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mm-std
-Version: 0.4.3
+Version: 0.4.4
 Requires-Python: >=3.12
 Requires-Dist: aiohttp-socks~=0.10.1
 Requires-Dist: aiohttp~=3.11.16

{mm_std-0.4.3.dist-info → mm_std-0.4.4.dist-info}/RECORD

@@ -1,6 +1,6 @@
 mm_std/__init__.py,sha256=gSMy_KY9kffPFnrDzdUti_JiQP1hB3ZtF7GxhOCc1Vc,2814
 mm_std/command.py,sha256=ze286wjUjg0QSTgIu-2WZks53_Vclg69UaYYgPpQvCU,1283
-mm_std/config.py,sha256=QtF-SCzaHZf511miTDjmIWR2dFfVPsnw2x6B5rmhMMk,1884
+mm_std/config.py,sha256=KArBvdt1gr_YvAtdXA4bqzF5W3IyUh25TexFoJCcN-E,3094
 mm_std/crypto.py,sha256=jdk0_TCmeU0pPXMyz9xH6kQHSjjZ9GcGClBwQps5vBo,340
 mm_std/date.py,sha256=976eEkSONuNqHQBgSRu8hrtH23tJqztbmHFHLdbP2TY,1879
 mm_std/dict.py,sha256=6GkhJPXD0LiJDxPcYe6jPdEDw-MN7P7mKu6U5XxwYDk,675
@@ -28,6 +28,6 @@ mm_std/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mm_std/http/http_request.py,sha256=VnjZKrfaXQfMxrHRUqQ-Sxtr5Qf9FXBiJ-mmJTCzNkY,3709
 mm_std/http/http_request_sync.py,sha256=bqCBilbe4ZJ9vkhuBQeU5UMTJh6BtvtUwjieEodu6rw,1542
 mm_std/http/http_response.py,sha256=gz4kCCL0qr7wvw4NNiK1K2hXlrY0B0Iq4fJuYSrv3bw,3920
-mm_std-0.4.3.dist-info/METADATA,sha256=JSvnB6eXS5YrbuHr15igF-YOvHK7_YD_1bPP4was71k,446
-mm_std-0.4.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_std-0.4.3.dist-info/RECORD,,
+mm_std-0.4.4.dist-info/METADATA,sha256=zQdcBSHWHf-504aqEWMZzK84MwT6stxPdvRryHjymZ4,446
+mm_std-0.4.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mm_std-0.4.4.dist-info/RECORD,,