瀏覽代碼

add more logs

Khaleel Al-Adhami 1 周之前
父節點
當前提交
224db72591
共有 1 個文件被更改,包括 22 次插入0 次删除
  1. 22 0
      reflex/app.py

+ 22 - 0
reflex/app.py

@@ -658,6 +658,28 @@ class App(MiddlewareMixin, LifespanMixin):
         top_asgi_app.mount("", asgi_app)
         top_asgi_app.mount("", asgi_app)
         App._add_cors(top_asgi_app)
         App._add_cors(top_asgi_app)
 
 
+        class LogRequestsMiddleware:
+            def __init__(self, app: ASGIApp):
+                self.app = app
+
+            async def __call__(self, scope: Scope, receive: Receive, send: Send):
+                async def log_receive():
+                    message = await receive()
+                    print(  # noqa: T201
+                        f"Receiving Message: {message} with scope {scope}"
+                    )
+                    return message
+
+                async def log_send(message: Message):
+                    print(  # noqa: T201
+                        f"Sending Message: {message} with scope {scope}"
+                    )
+                    await send(message)
+
+                await self.app(scope, log_receive, log_send)
+
+        top_asgi_app.add_middleware(LogRequestsMiddleware)
+
         return top_asgi_app
         return top_asgi_app
 
 
     def _add_default_endpoints(self):
     def _add_default_endpoints(self):