middleware now skips non-http/ws requests

alex-oleshkevich · Nov 15, 2022 · 8ea0cb6 · 8ea0cb6
1 parent 41ee54c
commit 8ea0cb6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/starlette_babel/locale.py b/starlette_babel/locale.py
@@ -154,6 +154,10 @@ def __init__(
         ]
 
     async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+        if scope['type'] not in ('http', 'websocket'):
+            await self.app(scope, receive, send)
+            return
+
         async def send_wrapper(message: Message) -> None:
             if message["type"] == "http.response.start":
                 message["headers"].append(tuple([b"content-language", scope["state"]["language"].encode()]))

diff --git a/starlette_babel/timezone.py b/starlette_babel/timezone.py
@@ -125,6 +125,10 @@ def __init__(
         ]
 
     async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+        if scope['type'] not in ('http', 'websocket'):
+            await self.app(scope, receive, send)
+            return
+
         conn = HTTPConnection(scope)
         tz = self.detect_timezone(conn) or self.fallback
         try: