Add fastapi source to support 0.118.0+

Author: tomaisthorpe

aikido_zen/__init__.py

@@ -61,6 +61,7 @@ def protect(mode="daemon", token=""):
     import aikido_zen.sources.flask
     import aikido_zen.sources.quart
     import aikido_zen.sources.starlette
+    import aikido_zen.sources.fastapi
     import aikido_zen.sources.xml_sources.xml
     import aikido_zen.sources.xml_sources.lxml
 

aikido_zen/sources/fastapi/__init__.py

@@ -0,0 +1 @@
+import aikido_zen.sources.fastapi.fastapi_routing

aikido_zen/sources/fastapi/fastapi_routing.py

@@ -0,0 +1,18 @@
+from ...sinks import on_import, patch_function
+from ..starlette.starlette_routing import _request_response
+
+
+@on_import("fastapi.routing", "fastapi")
+def patch(m):
+    """
+    patching module fastapi.routing
+    - patches: request_response
+
+    Newer FastAPI defines its own request_response rather than importing from
+    starlette.routing, so the starlette patch alone doesn't cover APIRoute endpoints.
+    In older versions FastAPI imports from starlette, so we skip to avoid double-wrapping.
+    """
+    import starlette.routing
+
+    if m.request_response is not starlette.routing.request_response:
+        patch_function(m, "request_response", _request_response)