apitally 0.16.0__tar.gz → 0.16.1__tar.gz

{apitally-0.16.0 → apitally-0.16.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apitally
-Version: 0.16.0
+Version: 0.16.1
 Summary: Simple API monitoring & analytics for REST APIs built with FastAPI, Flask, Django, Starlette, Litestar and BlackSheep.
 Project-URL: Homepage, https://apitally.io
 Project-URL: Documentation, https://docs.apitally.io

{apitally-0.16.0 → apitally-0.16.1}/apitally/blacksheep.py

@@ -1,4 +1,3 @@
-import asyncio
 import time
 from typing import Any, Awaitable, Callable, Dict, List, Optional, Tuple, Union
 
@@ -60,15 +59,14 @@ class ApitallyMiddleware:
         identify_consumer_callback: Optional[Callable[[Request], Union[str, ApitallyConsumer, None]]] = None,
     ) -> None:
         self.app = app
+        self.app_version = app_version
         self.identify_consumer_callback = identify_consumer_callback
         self.client = ApitallyClient(
             client_id=client_id,
             env=env,
             request_logging_config=request_logging_config,
         )
-        self.client.start_sync_loop()
-        self._delayed_set_startup_data_task: Optional[asyncio.Task] = None
-        self.delayed_set_startup_data(app_version)
+        self.app.on_start += self.after_start
         self.app.on_stop += self.on_stop
 
         self.capture_request_body = (
@@ -78,13 +76,10 @@ class ApitallyMiddleware:
             self.client.request_logger.config.enabled and self.client.request_logger.config.log_response_body
         )
 
-    def delayed_set_startup_data(self, app_version: Optional[str] = None) -> None:
-        self._delayed_set_startup_data_task = asyncio.create_task(self._delayed_set_startup_data(app_version))
-
-    async def _delayed_set_startup_data(self, app_version: Optional[str] = None) -> None:
-        await asyncio.sleep(1.0)  # Short delay to allow app routes to be registered first
-        data = _get_startup_data(self.app, app_version=app_version)
+    async def after_start(self, application: Application) -> None:
+        data = _get_startup_data(application, app_version=self.app_version)
         self.client.set_startup_data(data)
+        self.client.start_sync_loop()
 
     async def on_stop(self, application: Application) -> None:
         await self.client.handle_shutdown()

{apitally-0.16.0 → apitally-0.16.1}/tests/test_blacksheep.py

@@ -21,7 +21,6 @@ async def app(module_mocker: MockerFixture) -> Application:
     module_mocker.patch("apitally.client.client_asyncio.ApitallyClient._instance", None)
     module_mocker.patch("apitally.client.client_asyncio.ApitallyClient.start_sync_loop")
     module_mocker.patch("apitally.client.client_asyncio.ApitallyClient.set_startup_data")
-    module_mocker.patch("apitally.blacksheep.ApitallyMiddleware.delayed_set_startup_data")
     app = get_app()
     await app.start()
     return app
@@ -46,6 +45,7 @@ def get_app() -> Application:
             log_request_body=True,
             log_response_body=True,
         ),
+        app_version="1.2.3",
         identify_consumer_callback=identify_consumer,
     )