nucliadb-telemetry 5.2.1.post1869__py3-none-any.whl → 5.2.1.post1873__py3-none-any.whl

nucliadb_telemetry/fastapi/context.py

@@ -16,13 +16,16 @@
 #
 # You should have received a copy of the GNU Affero General Public License
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
+from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
+from starlette.requests import Request
+from starlette.responses import Response
 
 from nucliadb_telemetry import context
 
 from .utils import get_path_template
 
 
-class ContextInjectorMiddleware:
+class ContextInjectorMiddleware(BaseHTTPMiddleware):
     """
     Automatically inject context values for the current request's path parameters
 
@@ -30,13 +33,9 @@ class ContextInjectorMiddleware:
         - `/api/v1/kb/{kbid}` would inject a context value for `kbid`
     """
 
-    def __init__(self, app):
-        self.app = app
+    async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response:
+        found_path_template = get_path_template(request.scope)
+        if found_path_template.match:
+            context.add_context(found_path_template.scope.get("path_params", {}))  # type: ignore
 
-    async def __call__(self, scope, receive, send):
-        if scope["type"] == "http":
-            found_path_template = get_path_template(scope)
-            if found_path_template.match:
-                context.add_context(found_path_template.scope.get("path_params", {}))  # type: ignore
-
-        return await self.app(scope, receive, send)
+        return await call_next(request)

{nucliadb_telemetry-5.2.1.post1869.dist-info → nucliadb_telemetry-5.2.1.post1873.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_telemetry
-Version: 5.2.1.post1869
+Version: 5.2.1.post1873
 Summary: NucliaDB Telemetry Library Python process
 Home-page: https://github.com/nuclia/nucliadb
 Author: nucliadb Authors
@@ -26,24 +26,24 @@ Requires-Dist: prometheus-client>=0.12.0
 Requires-Dist: orjson>=3.6.7
 Requires-Dist: wrapt>=1.14.1
 Provides-Extra: all
-Requires-Dist: grpcio-testing<1.63.0,>=1.44.0; extra == "all"
-Requires-Dist: grpcio-reflection<1.63.0,>=1.44.0; extra == "all"
+Requires-Dist: grpcio-health-checking<1.63.0,>=1.44.0; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-fastapi>=0.42b0; extra == "all"
-Requires-Dist: grpcio<1.63.0,>=1.44.0; extra == "all"
-Requires-Dist: grpcio-channelz<1.63.0,>=1.44.0; extra == "all"
-Requires-Dist: opentelemetry-proto==1.21.0; extra == "all"
-Requires-Dist: PyNaCl; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-aiohttp-client>=0.42b0; extra == "all"
-Requires-Dist: opentelemetry-exporter-jaeger==1.21.0; extra == "all"
-Requires-Dist: nats-py[nkeys]>=2.5.0; extra == "all"
+Requires-Dist: grpcio<1.63.0,>=1.44.0; extra == "all"
 Requires-Dist: opentelemetry-propagator-b3==1.21.0; extra == "all"
-Requires-Dist: opentelemetry-api==1.21.0; extra == "all"
 Requires-Dist: opentelemetry-semantic-conventions>=0.42b0; extra == "all"
-Requires-Dist: grpcio-status<1.63.0,>=1.44.0; extra == "all"
+Requires-Dist: grpcio-reflection<1.63.0,>=1.44.0; extra == "all"
+Requires-Dist: nats-py[nkeys]>=2.5.0; extra == "all"
+Requires-Dist: grpcio-testing<1.63.0,>=1.44.0; extra == "all"
 Requires-Dist: fastapi; extra == "all"
 Requires-Dist: grpcio-tools<1.63.0,>=1.44.0; extra == "all"
+Requires-Dist: opentelemetry-exporter-jaeger==1.21.0; extra == "all"
+Requires-Dist: opentelemetry-api==1.21.0; extra == "all"
+Requires-Dist: grpcio-channelz<1.63.0,>=1.44.0; extra == "all"
+Requires-Dist: opentelemetry-proto==1.21.0; extra == "all"
 Requires-Dist: opentelemetry-sdk==1.21.0; extra == "all"
-Requires-Dist: grpcio-health-checking<1.63.0,>=1.44.0; extra == "all"
+Requires-Dist: grpcio-status<1.63.0,>=1.44.0; extra == "all"
+Requires-Dist: PyNaCl; extra == "all"
 Provides-Extra: fastapi
 Requires-Dist: fastapi; extra == "fastapi"
 Requires-Dist: opentelemetry-sdk==1.21.0; extra == "fastapi"

{nucliadb_telemetry-5.2.1.post1869.dist-info → nucliadb_telemetry-5.2.1.post1873.dist-info}/RECORD

@@ -15,7 +15,7 @@ nucliadb_telemetry/settings.py,sha256=Gr5rC9tBKHwck6sMMP76rn4Zgwh1w81dSzJJNcgkZv
 nucliadb_telemetry/tracerprovider.py,sha256=ycVqSMo0NTa7EEv4QLC8qAt0NYKbOy8uC9k14PjAiuk,3744
 nucliadb_telemetry/utils.py,sha256=lqbW5SjpGcfEhiJUScYeaNrtt5BFnOTaJn5dwnjCVMM,4765
 nucliadb_telemetry/fastapi/__init__.py,sha256=NRtQtyE1jZzTK5bZETYqb0d6bpjthTAt-UXiO8DAQWQ,3716
-nucliadb_telemetry/fastapi/context.py,sha256=qhl3rVDtqDfnsCHBAlv-eX8SUSlooMnsryBcV5FMZvU,1530
+nucliadb_telemetry/fastapi/context.py,sha256=441ggcZuI6aFYrAJpw1jCTzMy-1gUoBpIvNz1QtB7HQ,1651
 nucliadb_telemetry/fastapi/metrics.py,sha256=15GDaTNslQ00AZEAZvQFC8OyVIKdKLjzhafnWbsfytw,3906
 nucliadb_telemetry/fastapi/tracing.py,sha256=U85I3QIFb27aj4v3Cl6l5evEgskP12LuEXW4pAsvGCc,15052
 nucliadb_telemetry/fastapi/utils.py,sha256=HtfhBSaYs_HWHx7SBp0pPAsoig5CLcHNFIFnzb-RXOE,2979
@@ -30,7 +30,7 @@ nucliadb_telemetry/tests/grpc/helloworld_pb2.py,sha256=ht4dmi0pAy6qDrwcjkbtSf_hO
 nucliadb_telemetry/tests/grpc/helloworld_pb2.pyi,sha256=mBZCQE6z7riQw6fOegJgZ5lHmXqO23_zxL8iQMqSqms,1191
 nucliadb_telemetry/tests/grpc/helloworld_pb2_grpc.py,sha256=_jxUNxl4Fx-JztK9RO5R6osjNP2sVNVPAxLnmczEYOc,2677
 nucliadb_telemetry/tests/grpc/helloworld_pb2_grpc.pyi,sha256=Y6teCx-PhPU-rI6w5ItLBKaTb34FLpngPnuDVWtNve4,958
-nucliadb_telemetry-5.2.1.post1869.dist-info/METADATA,sha256=1w2QRJuveau2N2Wq263-WCezWm0ehWlhnHlxOlJ3mm8,10708
-nucliadb_telemetry-5.2.1.post1869.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-nucliadb_telemetry-5.2.1.post1869.dist-info/top_level.txt,sha256=3qEHI_5ttqQIL2gkNYwSlKsFyBBiEzDiIy9UKISzOaQ,19
-nucliadb_telemetry-5.2.1.post1869.dist-info/RECORD,,
+nucliadb_telemetry-5.2.1.post1873.dist-info/METADATA,sha256=KmFDzJmg-BrVjZdlPXRBSewDaqJRjUOr61fQ2Dqxq-4,10708
+nucliadb_telemetry-5.2.1.post1873.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+nucliadb_telemetry-5.2.1.post1873.dist-info/top_level.txt,sha256=3qEHI_5ttqQIL2gkNYwSlKsFyBBiEzDiIy9UKISzOaQ,19
+nucliadb_telemetry-5.2.1.post1873.dist-info/RECORD,,