apitally 0.19.0__py3-none-any.whl → 0.19.1__py3-none-any.whl

apitally/client/sentry.py

@@ -6,17 +6,22 @@ from typing import Callable, Set
 _tasks: Set[asyncio.Task] = set()
 
 
-def get_sentry_event_id_async(cb: Callable[[str], None]) -> None:
+def get_sentry_event_id_async(cb: Callable[[str], None], raise_on_error: bool = False) -> None:
     try:
-        from sentry_sdk.hub import Hub
+        import sentry_sdk
         from sentry_sdk.scope import Scope
     except ImportError:
+        if raise_on_error:
+            raise
         return  # pragma: no cover
     if not hasattr(Scope, "get_isolation_scope") or not hasattr(Scope, "_last_event_id"):
-        # sentry-sdk < 2.2.0 is not supported
+        if raise_on_error:
+            raise RuntimeError("sentry-sdk < 2.2.0 is not supported")
         return  # pragma: no cover
-    if Hub.current.client is None:
-        return  # sentry-sdk not initialized
+    if not sentry_sdk.is_initialized():
+        if raise_on_error:
+            raise RuntimeError("sentry-sdk not initialized")
+        return
 
     scope = Scope.get_isolation_scope()
     if event_id := scope._last_event_id:

apitally/fastapi.py

@@ -1,4 +1,4 @@
-from apitally.starlette import ApitallyConsumer, ApitallyMiddleware, RequestLoggingConfig
+from apitally.starlette import ApitallyConsumer, ApitallyMiddleware, RequestLoggingConfig, set_consumer
 
 
-__all__ = ["ApitallyMiddleware", "ApitallyConsumer", "RequestLoggingConfig"]
+__all__ = ["ApitallyMiddleware", "ApitallyConsumer", "RequestLoggingConfig", "set_consumer"]

apitally/flask.py

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
     from werkzeug.routing.map import Map
 
 
-__all__ = ["ApitallyMiddleware", "ApitallyConsumer", "RequestLoggingConfig"]
+__all__ = ["ApitallyMiddleware", "ApitallyConsumer", "RequestLoggingConfig", "set_consumer"]
 
 
 class ApitallyMiddleware:
@@ -216,6 +216,10 @@ class ApitallyMiddleware:
         return None
 
 
+def set_consumer(identifier: str, name: Optional[str] = None, group: Optional[str] = None) -> None:
+    g.apitally_consumer = ApitallyConsumer(identifier, name=name, group=group)
+
+
 def _get_startup_data(
     app: Flask, app_version: Optional[str] = None, openapi_url: Optional[str] = None
 ) -> Dict[str, Any]:

apitally/litestar.py

@@ -24,7 +24,7 @@ from apitally.client.request_logging import (
 from apitally.common import get_versions, parse_int, try_json_loads
 
 
-__all__ = ["ApitallyPlugin", "ApitallyConsumer", "RequestLoggingConfig"]
+__all__ = ["ApitallyPlugin", "ApitallyConsumer", "RequestLoggingConfig", "set_consumer"]
 
 
 class ApitallyPlugin(InitPluginProtocol):
@@ -287,6 +287,10 @@ class ApitallyPlugin(InitPluginProtocol):
         return None
 
 
+def set_consumer(request: Request, identifier: str, name: Optional[str] = None, group: Optional[str] = None) -> None:
+    request.state.apitally_consumer = ApitallyConsumer(identifier, name=name, group=group)
+
+
 def _get_openapi(app: Litestar) -> str:
     schema = app.openapi_schema.to_schema()
     return json.dumps(schema)

apitally/starlette.py

@@ -25,7 +25,7 @@ from apitally.client.request_logging import (
 from apitally.common import get_versions, parse_int, try_json_loads
 
 
-__all__ = ["ApitallyMiddleware", "ApitallyConsumer", "RequestLoggingConfig"]
+__all__ = ["ApitallyMiddleware", "ApitallyConsumer", "RequestLoggingConfig", "set_consumer"]
 
 
 class ApitallyMiddleware:
@@ -268,6 +268,10 @@ class ApitallyMiddleware:
         return None
 
 
+def set_consumer(request: Request, identifier: str, name: Optional[str] = None, group: Optional[str] = None) -> None:
+    request.state.apitally_consumer = ApitallyConsumer(identifier, name=name, group=group)
+
+
 def _get_startup_data(
     app: ASGIApp, app_version: Optional[str] = None, openapi_url: Optional[str] = None
 ) -> Dict[str, Any]:

{apitally-0.19.0.dist-info → apitally-0.19.1.dist-info}/METADATA

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

@@ -4,11 +4,11 @@ apitally/common.py,sha256=azDxepViH0QW0MuufTHxeSQyLGzCkocAX_KPziWTx8A,1605
 apitally/django.py,sha256=7eSh1tzuu0wRl9PxJgXMBSz9DfyohCDIqd4ecQTOJ1M,18499
 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=OoCEnjtnD51GUGq-adK80ebuiLj-5HXubxffCv5XTCM,9622
-apitally/litestar.py,sha256=mHoMqBO_gyoopeHljY8e8GTcV29UDf3uhQMxY3GeNpA,13451
+apitally/fastapi.py,sha256=813dGFl2eOSSYOYwuubFQYvpRikV3shu7bLR2TlwUC8,199
+apitally/flask.py,sha256=sqcsVmbfqpVSN409xDF8DeGOSNkWHI4SoyTFxNfWJ4w,9819
+apitally/litestar.py,sha256=fmcWNmvGvrjGG6J_q5lqB_qJpTk6PXFK9jam6bTWU8U,13678
 apitally/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-apitally/starlette.py,sha256=Ep7n_yAqoleFpuLk43kMONiFJVRsFsu-G2_TIjfiCHQ,13878
+apitally/starlette.py,sha256=PnEwplG7i0qJpfEWKdJxGCIioLFp4kQfa-5Y6NWGSSg,14105
 apitally/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 apitally/client/client_asyncio.py,sha256=rTsH5wlLHK3RmyIuEiT6vzjquU-l2OPC34JnC2U6uYw,6658
 apitally/client/client_base.py,sha256=DvivGeHd3dyOASRvkIo44Zh8RzdBMfH8_rROa2lFbgw,3799
@@ -17,10 +17,10 @@ apitally/client/consumers.py,sha256=w_AFQhVgdtJVt7pVySBvSZwQg-2JVqmD2JQtVBoMkus,
 apitally/client/logging.py,sha256=QMsKIIAFo92PNBUleeTgsrsQa7SEal-oJa1oOHUr1wI,507
 apitally/client/request_logging.py,sha256=yCF_CH8isp3gpQRVgakk6DNV60mtPZWG6z45pUNACOM,17644
 apitally/client/requests.py,sha256=SDptGOg9XvaEKFj2o3oxJz-JAuZzUrqpHnbOQixf99o,3794
-apitally/client/sentry.py,sha256=qMjHdI0V7c50ruo1WjmjWc8g6oGDv724vSCvcuZ8G9k,1188
+apitally/client/sentry.py,sha256=dXW2zf-wexYSp4CJBsFFRKz9jWkPBy0ftqIk0o5Hkq8,1364
 apitally/client/server_errors.py,sha256=4B2BKDFoIpoWc55UVH6AIdYSgzj6zxCdMNUW77JjhZw,3423
 apitally/client/validation_errors.py,sha256=6G8WYWFgJs9VH9swvkPXJGuOJgymj5ooWA9OwjUTbuM,1964
-apitally-0.19.0.dist-info/METADATA,sha256=9hbiC_z4DINCao5gg0F0Lqd9NVR8hZqNnnjpXkg9lnI,9316
-apitally-0.19.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-apitally-0.19.0.dist-info/licenses/LICENSE,sha256=vbLzC-4TddtXX-_AFEBKMYWRlxC_MN0g66QhPxo8PgY,1065
-apitally-0.19.0.dist-info/RECORD,,
+apitally-0.19.1.dist-info/METADATA,sha256=QE1kA2cHRaEPmxm3feYkfbeowkZMGnuAu5plkvfXDz0,9316
+apitally-0.19.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+apitally-0.19.1.dist-info/licenses/LICENSE,sha256=vbLzC-4TddtXX-_AFEBKMYWRlxC_MN0g66QhPxo8PgY,1065
+apitally-0.19.1.dist-info/RECORD,,