ul-api-utils 7.10.5__tar.gz → 7.10.6__tar.gz

{ul-api-utils-7.10.5 → ul-api-utils-7.10.6}/PKG-INFO

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

{ul-api-utils-7.10.5 → ul-api-utils-7.10.6}/setup.py

@@ -9,7 +9,7 @@ with open(path.join(HERE, 'README.md'), encoding='utf-8') as f:
 
 setup(
     name='ul-api-utils',
-    version='7.10.5',
+    version='7.10.6',
     description='Python api utils',
     author='Unic-lab',
     long_description=long_description,

ul-api-utils-7.10.6/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-7.10.6/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-7.10.6/ul_api_utils/utils/memory_db/repository.py

@@ -0,0 +1,98 @@
+import decimal
+import redis
+import ormsgpack
+import collections
+
+from pydantic import BaseModel, ValidationError
+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:
+            model_fields = _parse_as_type.__fields_set__
+            if isinstance(value, list):
+                return [_parse_as_type.construct(_fields_set=model_fields, **obj) for obj in value]
+            return _parse_as_type.construct(_fields_set=model_fields, **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 → ul-api-utils-7.10.6}/ul_api_utils.egg-info/PKG-INFO

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

{ul-api-utils-7.10.5 → ul-api-utils-7.10.6}/ul_api_utils.egg-info/SOURCES.txt

@@ -117,6 +117,7 @@ ul_api_utils/utils/decode_base64.py
 ul_api_utils/utils/deprecated.py
 ul_api_utils/utils/flags.py
 ul_api_utils/utils/imports.py
+ul_api_utils/utils/instance_checks.py
 ul_api_utils/utils/json_encoder.py
 ul_api_utils/utils/load_modules.py
 ul_api_utils/utils/token_check.py
@@ -144,6 +145,10 @@ ul_api_utils/utils/flask_swagger_generator/utils/security_type.py
 ul_api_utils/utils/jinja/__init__.py
 ul_api_utils/utils/jinja/t_url_for.py
 ul_api_utils/utils/jinja/to_pretty_json.py
+ul_api_utils/utils/memory_db/__init__.py
+ul_api_utils/utils/memory_db/errors.py
+ul_api_utils/utils/memory_db/repository.py
+ul_api_utils/utils/memory_db/__tests__/__init__.py
 ul_api_utils/validators/__init__.py
 ul_api_utils/validators/custom_fields.py
 ul_api_utils/validators/validate_empty_object.py