apitally 0.16.0__py3-none-any.whl → 0.16.2__py3-none-any.whl

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/client/client_asyncio.py

@@ -41,7 +41,6 @@ class ApitallyClient(ApitallyClientBase):
         self._stop_sync_loop = False
         self._sync_loop_task: Optional[asyncio.Task] = None
         self._sync_data_queue: asyncio.Queue[Dict[str, Any]] = asyncio.Queue()
-        self._set_startup_data_task: Optional[asyncio.Task] = None
 
     def get_http_client(self) -> httpx.AsyncClient:
         if httpx.__version__ >= "0.26.0":
@@ -67,7 +66,7 @@ class ApitallyClient(ApitallyClientBase):
                 try:
                     async with self.get_http_client() as client:
                         tasks = [self.send_sync_data(client), self.send_log_data(client)]
-                        if not self._startup_data_sent and last_sync_time > 0:  # not on first sync
+                        if not self._startup_data_sent:
                             tasks.append(self.send_startup_data(client))
                         await asyncio.gather(*tasks)
                     last_sync_time = now
@@ -92,11 +91,6 @@ class ApitallyClient(ApitallyClientBase):
     def set_startup_data(self, data: Dict[str, Any]) -> None:
         self._startup_data_sent = False
         self._startup_data = self.add_uuids_to_data(data)
-        self._set_startup_data_task = asyncio.create_task(self._set_startup_data())
-
-    async def _set_startup_data(self) -> None:
-        async with self.get_http_client() as client:
-            await self.send_startup_data(client)
 
     async def send_startup_data(self, client: httpx.AsyncClient) -> None:
         if self._startup_data is not None:

apitally/client/client_threading.py

@@ -80,7 +80,7 @@ class ApitallyClient(ApitallyClientBase):
                 if (now - last_sync_time) >= self.sync_interval:
                     try:
                         with requests.Session() as session:
-                            if not self._startup_data_sent and last_sync_time > 0:  # not on first sync
+                            if not self._startup_data_sent:
                                 self.send_startup_data(session)
                             self.send_sync_data(session)
                             self.send_log_data(session)
@@ -105,8 +105,6 @@ class ApitallyClient(ApitallyClientBase):
     def set_startup_data(self, data: Dict[str, Any]) -> None:
         self._startup_data_sent = False
         self._startup_data = self.add_uuids_to_data(data)
-        with requests.Session() as session:
-            self.send_startup_data(session)
 
     def send_startup_data(self, session: requests.Session) -> None:
         if self._startup_data is not None:

apitally/django.py

@@ -75,13 +75,13 @@ class ApitallyMiddleware:
             request_logging_config=self.config.request_logging_config,
             proxy=self.config.proxy,
         )
-        self.client.start_sync_loop()
         self.client.set_startup_data(
             _get_startup_data(
                 app_version=self.config.app_version,
                 urlconfs=self.config.urlconfs,
             )
         )
+        self.client.start_sync_loop()
 
         self.capture_request_body = (
             self.client.request_logger.config.enabled and self.client.request_logger.config.log_request_body

apitally/flask.py

@@ -51,7 +51,6 @@ class ApitallyMiddleware:
             request_logging_config=request_logging_config,
             proxy=proxy,
         )
-        self.client.start_sync_loop()
         self.delayed_set_startup_data(app_version, openapi_url)
 
         self.capture_request_body = (
@@ -73,6 +72,7 @@ class ApitallyMiddleware:
     def _delayed_set_startup_data(self, app_version: Optional[str] = None, openapi_url: Optional[str] = None) -> None:
         data = _get_startup_data(self.app, app_version, openapi_url)
         self.client.set_startup_data(data)
+        self.client.start_sync_loop()
 
     def __call__(self, environ: WSGIEnvironment, start_response: StartResponse) -> Iterable[bytes]:
         if not self.client.enabled:

apitally/starlette.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import asyncio
 import time
 from typing import Any, Callable, Dict, List, Optional, Union
 from warnings import warn
@@ -40,6 +39,8 @@ class ApitallyMiddleware:
         proxy: Optional[Union[str, Proxy]] = None,
     ) -> None:
         self.app = app
+        self.app_version = app_version
+        self.openapi_url = openapi_url
         self.identify_consumer_callback = identify_consumer_callback
         self.client = ApitallyClient(
             client_id=client_id,
@@ -47,10 +48,6 @@ class ApitallyMiddleware:
             request_logging_config=request_logging_config,
             proxy=proxy,
         )
-        self.client.start_sync_loop()
-        self._delayed_set_startup_data_task: Optional[asyncio.Task] = None
-        self.delayed_set_startup_data(app_version, openapi_url)
-        _register_shutdown_handler(app, self.client.handle_shutdown)
 
         self.capture_request_body = (
             self.client.request_logger.config.enabled and self.client.request_logger.config.log_request_body
@@ -59,17 +56,13 @@ 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, openapi_url: Optional[str] = None) -> None:
-        self._delayed_set_startup_data_task = asyncio.create_task(
-            self._delayed_set_startup_data(app_version, openapi_url)
-        )
+        _register_event_handler(app, "startup", self.on_startup)
+        _register_event_handler(app, "shutdown", self.client.handle_shutdown)
 
-    async def _delayed_set_startup_data(
-        self, app_version: Optional[str] = None, openapi_url: 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, openapi_url)
+    def on_startup(self) -> None:
+        data = _get_startup_data(self.app, app_version=self.app_version, openapi_url=self.openapi_url)
         self.client.set_startup_data(data)
+        self.client.start_sync_loop()
 
     async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
         if self.client.enabled and scope["type"] == "http" and scope["method"] != "OPTIONS":
@@ -296,8 +289,8 @@ def _get_routes(app: Union[ASGIApp, Router]) -> List[BaseRoute]:
     return []  # pragma: no cover
 
 
-def _register_shutdown_handler(app: Union[ASGIApp, Router], shutdown_handler: Callable[[], Any]) -> None:
+def _register_event_handler(app: Union[ASGIApp, Router], event: str, handler: Callable[[], Any]) -> None:
     if isinstance(app, Router):
-        app.add_event_handler("shutdown", shutdown_handler)
+        app.add_event_handler(event, handler)
     elif hasattr(app, "app"):
-        _register_shutdown_handler(app.app, shutdown_handler)
+        _register_event_handler(app.app, event, handler)

{apitally-0.16.0.dist-info → apitally-0.16.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apitally
-Version: 0.16.0
+Version: 0.16.2
 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.dist-info → apitally-0.16.2.dist-info}/RECORD

@@ -1,18 +1,18 @@
 apitally/__init__.py,sha256=ShXQBVjyiSOHxoQJS2BvNG395W4KZfqMxZWBAR0MZrE,22
-apitally/blacksheep.py,sha256=LSdg5LG-bnXdhKtPkp71CBQKGVtUi3t_EirovoaRy-s,9973
+apitally/blacksheep.py,sha256=KvcPFeiwQgWZmRglbm8SLaN6_WRs5kZ3SymB1IuLR-A,9616
 apitally/common.py,sha256=azDxepViH0QW0MuufTHxeSQyLGzCkocAX_KPziWTx8A,1605
-apitally/django.py,sha256=zwe8svC8rfo7TyHfOlkYTeXptxPFoRjvt0bbYvgtJKM,16892
+apitally/django.py,sha256=f_k7yYlvvvhJMR53NcXCfmlLxLX3CeLO9ephF4bzKbo,16892
 apitally/django_ninja.py,sha256=-CmrwFFRv7thFOUK_OrOSouhHL9bm5sIBNIQlpyE_2c,166
 apitally/django_rest_framework.py,sha256=-CmrwFFRv7thFOUK_OrOSouhHL9bm5sIBNIQlpyE_2c,166
 apitally/fastapi.py,sha256=IfKfgsmIY8_AtnuMTW2sW4qnkya61CAE2vBoIpcc9tk,169
-apitally/flask.py,sha256=p_u33_FQq2i5AebWB8wYxXX0CPhcX8OJHGWj5dR4sPY,9622
+apitally/flask.py,sha256=OoCEnjtnD51GUGq-adK80ebuiLj-5HXubxffCv5XTCM,9622
 apitally/litestar.py,sha256=mHoMqBO_gyoopeHljY8e8GTcV29UDf3uhQMxY3GeNpA,13451
 apitally/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-apitally/starlette.py,sha256=iEcN--2eeUW9d78H42WolWEkss2idvXLjK2OQmvULdM,13218
+apitally/starlette.py,sha256=g6YmC186uG6Lc9-Gx3Pp4ke-jn8IeKK0nFLBGDnWYv0,12771
 apitally/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-apitally/client/client_asyncio.py,sha256=9mdi9Hmb6-xn7dNdwP84e4PNAHGg2bYdMEgIfPUAtcQ,7003
+apitally/client/client_asyncio.py,sha256=rTsH5wlLHK3RmyIuEiT6vzjquU-l2OPC34JnC2U6uYw,6658
 apitally/client/client_base.py,sha256=DvivGeHd3dyOASRvkIo44Zh8RzdBMfH8_rROa2lFbgw,3799
-apitally/client/client_threading.py,sha256=7JPu2Uulev7X2RiSLx4HJYfvAP6Z5zB_yuSevMfQC7I,7389
+apitally/client/client_threading.py,sha256=sxMRcxRgk1SxJjSq-qpIcDVmD3Q7Kv4CVT5zEUVt0KM,7257
 apitally/client/consumers.py,sha256=w_AFQhVgdtJVt7pVySBvSZwQg-2JVqmD2JQtVBoMkus,2626
 apitally/client/logging.py,sha256=QMsKIIAFo92PNBUleeTgsrsQa7SEal-oJa1oOHUr1wI,507
 apitally/client/request_logging.py,sha256=SMvQd3WDolnb8u9rHVh2_OgXwFjL2jLZt-GpZNQ1XGk,14115
@@ -20,7 +20,7 @@ apitally/client/requests.py,sha256=SDptGOg9XvaEKFj2o3oxJz-JAuZzUrqpHnbOQixf99o,3
 apitally/client/sentry.py,sha256=qMjHdI0V7c50ruo1WjmjWc8g6oGDv724vSCvcuZ8G9k,1188
 apitally/client/server_errors.py,sha256=4B2BKDFoIpoWc55UVH6AIdYSgzj6zxCdMNUW77JjhZw,3423
 apitally/client/validation_errors.py,sha256=6G8WYWFgJs9VH9swvkPXJGuOJgymj5ooWA9OwjUTbuM,1964
-apitally-0.16.0.dist-info/METADATA,sha256=kWbylw8pJ0Q0xHBt1P7S3h6cTqx9UQ8SBAnrxFew5CM,9321
-apitally-0.16.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-apitally-0.16.0.dist-info/licenses/LICENSE,sha256=vbLzC-4TddtXX-_AFEBKMYWRlxC_MN0g66QhPxo8PgY,1065
-apitally-0.16.0.dist-info/RECORD,,
+apitally-0.16.2.dist-info/METADATA,sha256=Lycf-s6pDP5lO9B8-pRqcDfHZcJTFZQ--wSuquoDZn0,9321
+apitally-0.16.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+apitally-0.16.2.dist-info/licenses/LICENSE,sha256=vbLzC-4TddtXX-_AFEBKMYWRlxC_MN0g66QhPxo8PgY,1065
+apitally-0.16.2.dist-info/RECORD,,