persistence-kit 3.5.0__tar.gz → 3.6.0__tar.gz

{persistence_kit-3.5.0 → persistence_kit-3.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: persistence-kit
-Version: 3.5.0
+Version: 3.6.0
 Summary: Reusable persistence and repository toolkit
 License: MIT
 License-File: LICENSE

{persistence_kit-3.5.0 → persistence_kit-3.6.0}/persistence_kit/__init__.py

@@ -218,6 +218,26 @@ _OPTIONAL_EXPORTS = {
     ),
     "decode": ("persistence_kit.restclient.mapping", "decode", "restclient"),
     "encode": ("persistence_kit.restclient.mapping", "encode", "restclient"),
+    "register_rest_service": (
+        "persistence_kit.restclient.provider",
+        "register_rest_service",
+        "restclient",
+    ),
+    "set_rest_registry_initializer": (
+        "persistence_kit.restclient.provider",
+        "set_rest_registry_initializer",
+        "restclient",
+    ),
+    "provide_rest_client": (
+        "persistence_kit.restclient.provider",
+        "provide_rest_client",
+        "restclient",
+    ),
+    "get_rest_client": (
+        "persistence_kit.restclient.provider",
+        "get_rest_client",
+        "restclient",
+    ),
     "NoAuth": ("persistence_kit.restclient.auth", "NoAuth", "restclient"),
     "ApiKeyAuth": ("persistence_kit.restclient.auth", "ApiKeyAuth", "restclient"),
     "BearerAuth": ("persistence_kit.restclient.auth", "BearerAuth", "restclient"),

{persistence_kit-3.5.0 → persistence_kit-3.6.0}/persistence_kit/restclient/__init__.py

@@ -44,6 +44,14 @@ from persistence_kit.restclient.populate import (
     populate,
 )
 from persistence_kit.restclient.payload import serialize_xml
+from persistence_kit.restclient.provider import (
+    default_rest_registry,
+    get_rest_client,
+    provide_rest_client,
+    register_rest_service,
+    reset_rest_registry,
+    set_rest_registry_initializer,
+)
 from persistence_kit.restclient.registry import RegisteredService, RestClientRegistry
 from persistence_kit.restclient.resolver import (
     DirectoryEndpointResolver,
@@ -85,6 +93,12 @@ __all__ = [
     "RestClientRegistry",
     "RegisteredService",
     "build_rest_client",
+    "register_rest_service",
+    "set_rest_registry_initializer",
+    "provide_rest_client",
+    "get_rest_client",
+    "default_rest_registry",
+    "reset_rest_registry",
     "RestClientError",
     "RestConfigError",
     "RestAuthError",

persistence_kit-3.6.0/persistence_kit/restclient/provider.py

@@ -0,0 +1,78 @@
+from __future__ import annotations
+
+from functools import lru_cache
+from typing import Callable
+
+from persistence_kit.resilience import CircuitBreaker
+from persistence_kit.restclient.config import ServiceConfig
+from persistence_kit.restclient.contracts import (
+    Authenticator,
+    EndpointResolver,
+    RestClient,
+)
+from persistence_kit.restclient.registry import RestClientRegistry
+from persistence_kit.restclient.retry import RetryPolicy
+
+_default_registry_instance: RestClientRegistry | None = None
+_registry_initializer: Callable[[], None] | None = None
+
+
+def default_rest_registry() -> RestClientRegistry:
+    global _default_registry_instance
+    if _default_registry_instance is None:
+        from persistence_kit.settings import PersistenceKitSettings
+
+        _default_registry_instance = RestClientRegistry(settings=PersistenceKitSettings())
+    return _default_registry_instance
+
+
+def set_rest_registry_initializer(fn: Callable[[], None] | None) -> None:
+    global _registry_initializer
+    _registry_initializer = fn
+
+
+@lru_cache(maxsize=1)
+def _init_registry() -> None:
+    if _registry_initializer is not None:
+        _registry_initializer()
+
+
+def register_rest_service(
+    name: str,
+    *,
+    base_url: str = "",
+    authenticator: Authenticator | None = None,
+    resolver: EndpointResolver | None = None,
+    config: ServiceConfig | None = None,
+    retry_policy: RetryPolicy | None = None,
+    circuit_breaker: CircuitBreaker | None = None,
+    replace: bool = False,
+) -> None:
+    default_rest_registry().register(
+        name,
+        base_url=base_url,
+        authenticator=authenticator,
+        resolver=resolver,
+        config=config,
+        retry_policy=retry_policy,
+        circuit_breaker=circuit_breaker,
+        replace=replace,
+    )
+
+
+def get_rest_client(name: str) -> RestClient:
+    _init_registry()
+    return default_rest_registry().get(name)
+
+
+def provide_rest_client(name: str) -> Callable[[], RestClient]:
+    def _dep() -> RestClient:
+        return get_rest_client(name)
+
+    return _dep
+
+
+def reset_rest_registry() -> None:
+    global _default_registry_instance
+    _init_registry.cache_clear()
+    _default_registry_instance = None

{persistence_kit-3.5.0 → persistence_kit-3.6.0}/persistence_kit/restclient/registry.py

@@ -103,6 +103,10 @@ class RestClientRegistry:
     def names(self) -> tuple[str, ...]:
         return tuple(self._services)
 
+    def clear(self) -> None:
+        self._services.clear()
+        self._clients.clear()
+
     async def aclose_all(self) -> None:
         for client in self._clients.values():
             await client.aclose()

{persistence_kit-3.5.0 → persistence_kit-3.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "persistence-kit"
-version = "3.5.0"
+version = "3.6.0"
 description = "Reusable persistence and repository toolkit"
 authors = ["Andres Felipe Serrano Barrios <andresfserrano1@gmail.com>"]
 readme = "README.md"