fastapi-mongo-base 1.4.1__tar.gz → 1.4.3__tar.gz

{fastapi_mongo_base-1.4.1/src/fastapi_mongo_base.egg-info → fastapi_mongo_base-1.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-mongo-base
-Version: 1.4.1
+Version: 1.4.3
 Summary: A simple boilerplate application, including models and schemas and abstract router, for FastAPI with MongoDB
 Author-email: Mahdi Kiani <mahdikiany@gmail.com>
 Maintainer-email: Mahdi Kiani <mahdikiany@gmail.com>

{fastapi_mongo_base-1.4.1 → fastapi_mongo_base-1.4.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fastapi-mongo-base"
-version = "1.4.1"
+version = "1.4.3"
 description = "A simple boilerplate application, including models and schemas and abstract router, for FastAPI with MongoDB"
 readme = "README.md"
 requires-python = ">=3.10"

{fastapi_mongo_base-1.4.1 → fastapi_mongo_base-1.4.3}/src/fastapi_mongo_base/core/prometheus/mongo.py

@@ -65,6 +65,8 @@ class DatabasePoolMonitor(monitoring.ConnectionPoolListener):
         self.in_use_connections = 0
 
     def _update_metrics(self) -> None:
+        if pool_connections is None:
+            return
         pool_connections.labels(
             database=self.database_name,
             state="available",
@@ -109,6 +111,8 @@ class DatabasePoolMonitor(monitoring.ConnectionPoolListener):
         """Record a MongoDB connection creation event."""
         del event
 
+        if connections_created_total is None:
+            return
         with self._lock:
             connections_created_total.labels(
                 database=self.database_name,
@@ -121,6 +125,8 @@ class DatabasePoolMonitor(monitoring.ConnectionPoolListener):
         """Record a MongoDB connection readiness event."""
         del event
 
+        if connections_ready_total is None:
+            return
         with self._lock:
             self.available_connections += 1
 
@@ -137,6 +143,8 @@ class DatabasePoolMonitor(monitoring.ConnectionPoolListener):
         """Record a MongoDB connection closure event."""
         del event
 
+        if connections_closed_total is None:
+            return
         with self._lock:
             if self.in_use_connections > 0:
                 self.in_use_connections -= 1
@@ -159,6 +167,8 @@ class DatabasePoolMonitor(monitoring.ConnectionPoolListener):
         """Record the start of a MongoDB connection checkout attempt."""
         del event
 
+        if checkouts_started_total is None:
+            return
         with self._lock:
             checkouts_started_total.labels(
                 database=self.database_name,
@@ -169,6 +179,8 @@ class DatabasePoolMonitor(monitoring.ConnectionPoolListener):
         event: monitoring.ConnectionCheckOutFailedEvent,
     ) -> None:
         """Record a failed MongoDB connection checkout attempt."""
+        if checkouts_failed_total is None:
+            return
         with self._lock:
             checkouts_failed_total.labels(
                 database=self.database_name,

{fastapi_mongo_base-1.4.1 → fastapi_mongo_base-1.4.3}/src/fastapi_mongo_base/utils/basic.py

@@ -328,9 +328,21 @@ async def gather_sync(
 R = TypeVar("R")
 
 
+def _resolve_mock(
+    mock_response: R | Callable[..., R | Awaitable[R]],
+    *args: object,
+    **kwargs: object,
+) -> R | Awaitable[R]:
+    if callable(mock_response):
+        return mock_response(*args, **kwargs)
+    return mock_response
+
+
 def debug_mode_mock(
     mock_response: R | Callable[..., R | Awaitable[R]],
-) -> Callable[[Callable[..., Awaitable[R]]], Callable[..., Awaitable[R]]]:
+) -> Callable[
+    [Callable[..., R | Awaitable[R]]], Callable[..., R | Awaitable[R]]
+]:
     """
     Return a mock response if debug is enabled.
 
@@ -340,22 +352,39 @@ def debug_mode_mock(
     Returns:
         A decorator that returns a mock response if debug is enabled.
     """
-    from ..core.config import Settings
 
     def decorator(
-        func: Callable[..., Awaitable[R]],
-    ) -> Callable[..., Awaitable[R]]:
+        func: Callable[..., R | Awaitable[R]],
+    ) -> Callable[..., R | Awaitable[R]]:
+
+        from ..core.config import Settings
+
         @functools.wraps(func)
-        async def wrapper(*args: object, **kwargs: object) -> R:
+        async def async_wrapper(*args: object, **kwargs: object) -> R:
             if Settings.debug:
-                if callable(mock_response):
-                    result = mock_response(*args, **kwargs)
-                    if isinstance(result, Awaitable):
-                        return await result
-                    return result
-                return mock_response
+                result = _resolve_mock(mock_response, *args, **kwargs)
+                if isinstance(result, Awaitable):
+                    return await result
+                return result
             return await func(*args, **kwargs)
 
-        return wrapper
+        @functools.wraps(func)
+        def sync_wrapper(*args: object, **kwargs: object) -> R:
+            if Settings.debug:
+                result = _resolve_mock(mock_response, *args, **kwargs)
+                if isinstance(result, Awaitable):
+                    msg = (
+                        "debug_mode_mock callable returned "
+                        "Awaitable for sync function"
+                    )
+                    raise TypeError(msg)
+                return result
+            return func(*args, **kwargs)
+
+        return (
+            async_wrapper
+            if inspect.iscoroutinefunction(func)
+            else sync_wrapper
+        )
 
     return decorator

{fastapi_mongo_base-1.4.1 → fastapi_mongo_base-1.4.3/src/fastapi_mongo_base.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-mongo-base
-Version: 1.4.1
+Version: 1.4.3
 Summary: A simple boilerplate application, including models and schemas and abstract router, for FastAPI with MongoDB
 Author-email: Mahdi Kiani <mahdikiany@gmail.com>
 Maintainer-email: Mahdi Kiani <mahdikiany@gmail.com>