ul-api-utils 7.10.5__py3-none-any.whl → 7.10.7__py3-none-any.whl

ul_api_utils/utils/instance_checks.py

@@ -0,0 +1,16 @@
+from typing import Any
+
+
+def isinstance_namedtuple(obj: Any) -> bool:
+    namedtuple_unique_attributes = ('_asdict', '_fields')
+    is_tuple = isinstance(obj, tuple)
+    has_namedtuple_attributes = hasattr(obj, namedtuple_unique_attributes[0]) and hasattr(obj, namedtuple_unique_attributes[1])
+    return has_namedtuple_attributes and is_tuple
+
+
+def is_iterable(obj: Any) -> bool:
+    try:
+        iter(obj)
+    except TypeError:
+        return False
+    return True

ul_api_utils/utils/memory_db/errors.py

@@ -0,0 +1,8 @@
+class CompositeKeyError(LookupError):
+    """Raised in memory db repository when composite key is invalid."""
+    ...
+
+
+class UnsupportedParsingType(TypeError):
+    """Raised in memory db repository when repo can't parse memory db object as provided type."""
+    ...

ul_api_utils/utils/memory_db/repository.py

@@ -0,0 +1,96 @@
+import decimal
+import redis
+import ormsgpack
+import collections
+
+from pydantic import ValidationError, parse_obj_as, BaseModel
+from typing import Any, Type, overload, Iterator, KeysView, cast
+
+from ul_api_utils.utils.instance_checks import isinstance_namedtuple
+from ul_api_utils.utils.memory_db.errors import CompositeKeyError, UnsupportedParsingType
+
+CompositeKeyT = tuple[str, Type[BaseModel]]
+AnyKeyT = str | CompositeKeyT
+AnyT = Any
+RedisClientT = redis.StrictRedis | redis.Redis
+
+
+class BaseMemoryDbRepository(collections.abc.MutableMapping[str, AnyT]):
+    def __init__(self, redis_client: RedisClientT) -> None:
+        self._db = redis_client
+        self._composite_key_max_length = 2
+
+    @property
+    def db(self) -> RedisClientT:
+        return self._db
+
+    def get(self, __key: str, *, parse_as_type: Type[BaseModel] | None = None, default: Any | None = None) -> AnyT | None:  # type: ignore
+        try:
+            if parse_as_type is None:
+                return self[__key]
+            return self[__key, parse_as_type]
+        except KeyError:
+            return default
+
+    @overload
+    def __getitem__(self, key: str) -> AnyT:
+        ...
+
+    @overload
+    def __getitem__(self, key: CompositeKeyT) -> AnyT:
+        ...
+
+    def __getitem__(self, key: AnyKeyT) -> AnyT:
+        key_complex = isinstance(key, tuple)
+
+        if not key_complex:
+            single_key = cast(str, key)
+            return ormsgpack.unpackb(self._db[single_key])
+
+        composite_key = cast(CompositeKeyT, key)
+
+        if len(composite_key) > self._composite_key_max_length:
+            raise CompositeKeyError(f"Can't retrieve an item with {key=}. Composite key should have only two arguments.")
+
+        composite_key_name, _parse_as_type = composite_key
+        parsing_type_supported = issubclass(_parse_as_type, BaseModel) and _parse_as_type is not BaseModel
+        if not parsing_type_supported:
+            raise UnsupportedParsingType(f"Unsupported parsing type {_parse_as_type}.")
+        value = ormsgpack.unpackb(self._db[composite_key_name])
+        try:
+            if isinstance(value, list):
+                return parse_obj_as(list[_parse_as_type], value)  # type: ignore
+            return parse_obj_as(_parse_as_type, value)
+        except ValidationError:
+            raise UnsupportedParsingType(f"Could not parse the value of key '{composite_key_name}' with type {_parse_as_type}") from None
+
+    def __setitem__(self, key: str, value: AnyT) -> None:
+        self._db[key] = ormsgpack.packb(value, option=ormsgpack.OPT_SERIALIZE_PYDANTIC, default=self._default_serializer)
+
+    def __delitem__(self, key: str) -> None:
+        del self._db[key]
+
+    def __iter__(self) -> Iterator[str]:
+        return iter(self.keys())
+
+    def __len__(self) -> int:
+        return len(self._db.keys())
+
+    def keys(self) -> KeysView[str]:
+        available_keys = [key.decode() for key in self._db.keys()]
+        return cast(KeysView[str], available_keys)
+
+    def clear(self) -> None:
+        self._db.flushdb()
+
+    @staticmethod
+    def _default_serializer(obj: Any) -> Any:
+        if isinstance_namedtuple(obj):
+            return obj._asdict()
+        if isinstance(obj, decimal.Decimal):
+            return str(obj)
+        if isinstance(obj, set):
+            return list(obj)
+        if isinstance(obj, frozenset):
+            return list(obj)
+        raise TypeError

{ul_api_utils-7.10.5.dist-info → ul_api_utils-7.10.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ul-api-utils
-Version: 7.10.5
+Version: 7.10.7
 Summary: Python api utils
 Author: Unic-lab
 Author-email: 

{ul_api_utils-7.10.5.dist-info → ul_api_utils-7.10.7.dist-info}/RECORD

@@ -108,6 +108,7 @@ ul_api_utils/utils/decode_base64.py,sha256=rqiD5Whzm2MPx-bk_4r4G4Pe4UUFyU_u-UAjZ
 ul_api_utils/utils/deprecated.py,sha256=xR3ELgoDj7vJEY4CAYeEhdbtSJTfkukbjxcULtpMw58,866
 ul_api_utils/utils/flags.py,sha256=AYN5nKWp4-uu6PSlPptL7ZiLqr3Pu-x5dffF6SBsqfg,957
 ul_api_utils/utils/imports.py,sha256=i8PhoD0c_jnWTeXt_VxW_FihynwXSL_dHRT7jQiFyXE,376
+ul_api_utils/utils/instance_checks.py,sha256=9punTfY5uabuJhmSGLfTgBqRderoFysCXBSI8rvbPco,467
 ul_api_utils/utils/json_encoder.py,sha256=knpEW1c9XNdhgQg5gSeobEO2gPCasQB-0c8ig2Rc8Ho,4140
 ul_api_utils/utils/load_modules.py,sha256=_CPmQuB6o_33FE6zFl_GyO5xS5gmjfNffB6k-cglKAA,685
 ul_api_utils/utils/token_check.py,sha256=-Quuh8gOs9fNE1shYhdiMpQedafsLN7MB2ilSxG_F8E,489
@@ -135,15 +136,19 @@ ul_api_utils/utils/flask_swagger_generator/utils/security_type.py,sha256=AcWOQZB
 ul_api_utils/utils/jinja/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ul_api_utils/utils/jinja/t_url_for.py,sha256=PG9W4UbkWv2pLXNMQiCt22vp4sDi-Uz5w2u6DJ1fLkE,495
 ul_api_utils/utils/jinja/to_pretty_json.py,sha256=wcc_EJ6yM4lipE0Vr6cgYOB-rBk7_j_Wa7bijjI_bCs,302
+ul_api_utils/utils/memory_db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ul_api_utils/utils/memory_db/errors.py,sha256=Bw1O1Y_WTCeCRNgb9iwrGT1fst6iHORhrNstwxiaUu8,267
+ul_api_utils/utils/memory_db/repository.py,sha256=c_4BJuakhAtf0X3nlIKeWI4bsb1Wf2CK0xjKdY4_o-I,3395
+ul_api_utils/utils/memory_db/__tests__/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ul_api_utils/validators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ul_api_utils/validators/custom_fields.py,sha256=iqDnAvO-Bs13ZtLtccAG8SdJ8jWDlVcW6_ckChrTdXQ,4023
 ul_api_utils/validators/validate_empty_object.py,sha256=3Ck_iwyJE_M5e7l6s1i88aqb73zIt06uaLrMG2PAb0A,299
 ul_api_utils/validators/validate_uuid.py,sha256=EfvlRirv2EW0Z6w3s8E8rUa9GaI8qXZkBWhnPs8NFrA,257
 ul_api_utils/validators/__tests__/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ul_api_utils/validators/__tests__/test_custom_fields.py,sha256=QLZ7DFta01Z7DOK9Z5Iq4uf_CmvDkVReis-GAl_QN48,1447
-ul_api_utils-7.10.5.dist-info/LICENSE,sha256=6Qo8OdcqI8aGrswJKJYhST-bYqxVQBQ3ujKdTSdq-80,1062
-ul_api_utils-7.10.5.dist-info/METADATA,sha256=0E2lz0_Qwz4U4-NAbRulxYXAyCtJVYxfls1z4K1hlZc,14715
-ul_api_utils-7.10.5.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-ul_api_utils-7.10.5.dist-info/entry_points.txt,sha256=8tL3ySHWTyJMuV1hx1fHfN8zumDVOCOm63w3StphkXg,53
-ul_api_utils-7.10.5.dist-info/top_level.txt,sha256=1XsW8iOSFaH4LOzDcnNyxHpHrbKU3fSn-aIAxe04jmw,21
-ul_api_utils-7.10.5.dist-info/RECORD,,
+ul_api_utils-7.10.7.dist-info/LICENSE,sha256=6Qo8OdcqI8aGrswJKJYhST-bYqxVQBQ3ujKdTSdq-80,1062
+ul_api_utils-7.10.7.dist-info/METADATA,sha256=2hbqUk-lZ-f5Z1ObSQ_Ax31kio0SvrLlqDzZaecK-C8,14715
+ul_api_utils-7.10.7.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+ul_api_utils-7.10.7.dist-info/entry_points.txt,sha256=8tL3ySHWTyJMuV1hx1fHfN8zumDVOCOm63w3StphkXg,53
+ul_api_utils-7.10.7.dist-info/top_level.txt,sha256=1XsW8iOSFaH4LOzDcnNyxHpHrbKU3fSn-aIAxe04jmw,21
+ul_api_utils-7.10.7.dist-info/RECORD,,