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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
mm_std/__init__.py CHANGED
@@ -21,8 +21,8 @@ from .dict import replace_empty_dict_values as replace_empty_dict_values
21
21
  from .env import get_dotenv as get_dotenv
22
22
  from .http.http_request import http_request as http_request
23
23
  from .http.http_request_sync import http_request_sync as http_request_sync
24
- from .http.response import HttpError as HttpError
25
- from .http.response import HttpResponse as HttpResponse
24
+ from .http.http_response import HttpError as HttpError
25
+ from .http.http_response import HttpResponse as HttpResponse
26
26
  from .json_ import CustomJSONEncoder as CustomJSONEncoder
27
27
  from .json_ import json_dumps as json_dumps
28
28
  from .log import configure_logging as configure_logging
mm_std/config.py CHANGED
@@ -1,7 +1,7 @@
1
1
  import sys
2
2
  import tomllib
3
3
  from pathlib import Path
4
- from typing import NoReturn, Self
4
+ from typing import Any, NoReturn, Self, TypeVar
5
5
 
6
6
  from pydantic import BaseModel, ConfigDict, ValidationError
7
7
 
@@ -9,6 +9,8 @@ from .print_ import print_json, print_plain
9
9
  from .result import Result
10
10
  from .zip import read_text_from_zip_archive
11
11
 
12
+ T = TypeVar("T", bound="BaseConfig")
13
+
12
14
 
13
15
  class BaseConfig(BaseModel):
14
16
  model_config = ConfigDict(extra="forbid")
@@ -18,21 +20,18 @@ class BaseConfig(BaseModel):
18
20
  sys.exit(0)
19
21
 
20
22
  @classmethod
21
- def read_toml_config_or_exit[T](cls: type[T], config_path: Path, zip_password: str = "") -> T: # noqa: PYI019 # nosec
22
- res: Result[T] = cls.read_toml_config(config_path, zip_password) # type:ignore[attr-defined]
23
+ def read_toml_config_or_exit(cls, config_path: Path, zip_password: str = "") -> Self: # nosec
24
+ res: Result[Self] = cls.read_toml_config(config_path, zip_password)
23
25
  if res.is_ok():
24
26
  return res.unwrap()
27
+ cls._print_error_and_exit(res)
25
28
 
26
- if res.error == "validator_error" and res.extra:
27
- print_plain("config validation errors")
28
- for e in res.extra["errors"]:
29
- loc = e["loc"]
30
- field = ".".join(str(lo) for lo in loc) if len(loc) > 0 else ""
31
- print_plain(f"{field} {e['msg']}")
32
- else:
33
- print_plain(f"can't parse config file: {res.error}")
34
-
35
- sys.exit(1)
29
+ @classmethod
30
+ async def read_toml_config_or_exit_async(cls, config_path: Path, zip_password: str = "") -> Self: # nosec
31
+ res: Result[Self] = await cls.read_toml_config_async(config_path, zip_password)
32
+ if res.is_ok():
33
+ return res.unwrap()
34
+ cls._print_error_and_exit(res)
36
35
 
37
36
  @classmethod
38
37
  def read_toml_config(cls, config_path: Path, zip_password: str = "") -> Result[Self]: # nosec
@@ -48,3 +47,31 @@ class BaseConfig(BaseModel):
48
47
  return Result.failure(("validator_error", e), extra={"errors": e.errors()})
49
48
  except Exception as e:
50
49
  return Result.failure(e)
50
+
51
+ @classmethod
52
+ async def read_toml_config_async(cls, config_path: Path, zip_password: str = "") -> Result[Self]: # nosec
53
+ try:
54
+ config_path = config_path.expanduser()
55
+ if config_path.name.endswith(".zip"):
56
+ data = tomllib.loads(read_text_from_zip_archive(config_path, password=zip_password))
57
+ else:
58
+ with config_path.open("rb") as f:
59
+ data = tomllib.load(f)
60
+ model = await cls.model_validate_async(data) # type:ignore[attr-defined]
61
+ return Result.success(model)
62
+ except ValidationError as e:
63
+ return Result.failure(("validator_error", e), extra={"errors": e.errors()})
64
+ except Exception as e:
65
+ return Result.failure(e)
66
+
67
+ @classmethod
68
+ def _print_error_and_exit(cls, res: Result[Any]) -> NoReturn:
69
+ if res.error == "validator_error" and res.extra:
70
+ print_plain("config validation errors")
71
+ for e in res.extra["errors"]:
72
+ loc = e["loc"]
73
+ field = ".".join(str(lo) for lo in loc) if len(loc) > 0 else ""
74
+ print_plain(f"{field} {e['msg']}")
75
+ else:
76
+ print_plain(f"can't parse config file: {res.error}")
77
+ sys.exit(1)
@@ -3,7 +3,7 @@ from aiohttp.typedefs import LooseCookies, Query
3
3
  from aiohttp_socks import ProxyConnector
4
4
  from multidict import CIMultiDictProxy
5
5
 
6
- from mm_std.http.response import HttpError, HttpResponse
6
+ from mm_std.http.http_response import HttpError, HttpResponse
7
7
 
8
8
 
9
9
  async def http_request(
@@ -2,7 +2,7 @@ from typing import Any
2
2
 
3
3
  import requests
4
4
 
5
- from mm_std.http.response import HttpError, HttpResponse
5
+ from mm_std.http.http_response import HttpError, HttpResponse
6
6
 
7
7
 
8
8
  def http_request_sync(
@@ -57,10 +57,10 @@ class HttpResponse:
57
57
  def is_error(self) -> bool:
58
58
  return self.error is not None or (self.status_code is not None and self.status_code >= 400)
59
59
 
60
- def to_result_err[T](self, error: str | Exception | tuple[str, Exception] | None = None) -> Result[T]:
60
+ def to_result_failure[T](self, error: str | Exception | tuple[str, Exception] | None = None) -> Result[T]:
61
61
  return Result.failure(error or self.error or "error", extra=self.to_dict())
62
62
 
63
- def to_result_ok[T](self, result: T) -> Result[T]:
63
+ def to_result_success[T](self, result: T) -> Result[T]:
64
64
  return Result.success(result, extra=self.to_dict())
65
65
 
66
66
  def to_dict(self) -> dict[str, Any]:
mm_std/result.py CHANGED
@@ -79,6 +79,14 @@ class Result[T]:
79
79
  return self.exception
80
80
  raise RuntimeError("No exception provided")
81
81
 
82
+ def ok_or_error(self) -> T | str:
83
+ """
84
+ Returns the success value if available, otherwise returns the error message.
85
+ """
86
+ if self.is_ok():
87
+ return self.unwrap()
88
+ return self.unwrap_error()
89
+
82
90
  def to_dict(self) -> dict[str, object]:
83
91
  """
84
92
  Returns a dictionary representation of the result.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mm-std
3
- Version: 0.4.2
3
+ Version: 0.4.4
4
4
  Requires-Python: >=3.12
5
5
  Requires-Dist: aiohttp-socks~=0.10.1
6
6
  Requires-Dist: aiohttp~=3.11.16
@@ -1,6 +1,6 @@
1
- mm_std/__init__.py,sha256=Kl-7TT_hd-xmwDJJ0E5AO2Qjpj5H8H7A93DgDwv_r-0,2804
1
+ mm_std/__init__.py,sha256=gSMy_KY9kffPFnrDzdUti_JiQP1hB3ZtF7GxhOCc1Vc,2814
2
2
  mm_std/command.py,sha256=ze286wjUjg0QSTgIu-2WZks53_Vclg69UaYYgPpQvCU,1283
3
- mm_std/config.py,sha256=QtF-SCzaHZf511miTDjmIWR2dFfVPsnw2x6B5rmhMMk,1884
3
+ mm_std/config.py,sha256=KArBvdt1gr_YvAtdXA4bqzF5W3IyUh25TexFoJCcN-E,3094
4
4
  mm_std/crypto.py,sha256=jdk0_TCmeU0pPXMyz9xH6kQHSjjZ9GcGClBwQps5vBo,340
5
5
  mm_std/date.py,sha256=976eEkSONuNqHQBgSRu8hrtH23tJqztbmHFHLdbP2TY,1879
6
6
  mm_std/dict.py,sha256=6GkhJPXD0LiJDxPcYe6jPdEDw-MN7P7mKu6U5XxwYDk,675
@@ -12,7 +12,7 @@ mm_std/net.py,sha256=qdRCBIDneip6FaPNe5mx31UtYVmzqam_AoUF7ydEyjA,590
12
12
  mm_std/print_.py,sha256=zB7sVbSSF8RffMxvnOdbKCXjCKtKzKV3R68pBri4NkQ,1638
13
13
  mm_std/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
14
14
  mm_std/random_.py,sha256=OuUX4VJeSd13NZBya4qrGpR2TfN7_87tfebOY6DBUnI,1113
15
- mm_std/result.py,sha256=VrEIJvFaVFus-b5s75I8PATvfoIRQw084zuOhbnbRSA,7655
15
+ mm_std/result.py,sha256=Q4mmX2VP2TNpVFf1J_YWZY-rKwmqJwI1RlMjLsDF7hY,7896
16
16
  mm_std/str.py,sha256=BEjJ1p5O4-uSYK0h-enasSSDdwzkBbiwdQ4_dsrlEE8,3257
17
17
  mm_std/toml.py,sha256=CNznWKR0bpOxS6e3VB5LGS-Oa9lW-wterkcPUFtPcls,610
18
18
  mm_std/types_.py,sha256=9FGd2q47a8M9QQgsWJR1Kq34jLxBAkYSoJuwih4PPqg,257
@@ -25,9 +25,9 @@ mm_std/concurrency/sync_decorators.py,sha256=syCQBOmN7qPO55yzgJB2rbkh10CVww376hm
25
25
  mm_std/concurrency/sync_scheduler.py,sha256=j4tBL_cBI1spr0cZplTA7N2CoYsznuORMeRN8rpR6gY,2407
26
26
  mm_std/concurrency/sync_task_runner.py,sha256=s5JPlLYLGQGHIxy4oDS-PN7O9gcy-yPZFoNm8RQwzcw,1780
27
27
  mm_std/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
- mm_std/http/http_request.py,sha256=h74_ZjACwdbeINjzhuEwNUpvnaHZvyGl7TExjlBwMGg,3704
29
- mm_std/http/http_request_sync.py,sha256=aawVZfopzMI0alS3lkJQmVOVxH51rmtvsOK_inUlJOs,1537
30
- mm_std/http/response.py,sha256=Y6voWD3y_iK-tYBuXPborjcD7Y6-piupD6kcGXmAqmY,3911
31
- mm_std-0.4.2.dist-info/METADATA,sha256=j2djNKVv_DiXYat3p3hbpBaKzaSj5m-CExfViW5qVHU,446
32
- mm_std-0.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
33
- mm_std-0.4.2.dist-info/RECORD,,
28
+ mm_std/http/http_request.py,sha256=VnjZKrfaXQfMxrHRUqQ-Sxtr5Qf9FXBiJ-mmJTCzNkY,3709
29
+ mm_std/http/http_request_sync.py,sha256=bqCBilbe4ZJ9vkhuBQeU5UMTJh6BtvtUwjieEodu6rw,1542
30
+ mm_std/http/http_response.py,sha256=gz4kCCL0qr7wvw4NNiK1K2hXlrY0B0Iq4fJuYSrv3bw,3920
31
+ mm_std-0.4.4.dist-info/METADATA,sha256=zQdcBSHWHf-504aqEWMZzK84MwT6stxPdvRryHjymZ4,446
32
+ mm_std-0.4.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
33
+ mm_std-0.4.4.dist-info/RECORD,,
File without changes