refactor: ignore type at assignment

encode · Jan 10, 2023 · e5b7b75 · e5b7b75
1 parent aacd0bd
commit e5b7b75
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 12 deletions.
diff --git a/uvicorn/protocols/http/h11_impl.py b/uvicorn/protocols/http/h11_impl.py
@@ -240,10 +240,8 @@ def handle_events(self) -> None:
                     "raw_path": raw_path,
                     "query_string": query_string,
                     "headers": self.headers,
-                    "extensions": {},
+                    "extensions": {"tls": self.tls} if self.scheme == "https" else {},
                 }
-                if self.scheme == "https":
-                    self.scope["extensions"]["tls"] = self.tls
 
                 upgrade = self._get_upgrade()
                 if upgrade == b"websocket" and self._should_upgrade_to_ws():

diff --git a/uvicorn/protocols/http/httptools_impl.py b/uvicorn/protocols/http/httptools_impl.py
@@ -255,10 +255,8 @@ def on_message_begin(self) -> None:
             "scheme": self.scheme,
             "root_path": self.root_path,
             "headers": self.headers,
-            "extensions": {},
+            "extensions": {"tls": self.tls} if self.scheme == "https" else {},
         }
-        if self.scheme == "https":
-            self.scope["extensions"]["tls"] = self.tls
 
     # Parser callbacks
     def on_url(self, url: bytes) -> None:

diff --git a/uvicorn/protocols/websockets/websockets_impl.py b/uvicorn/protocols/websockets/websockets_impl.py
@@ -208,10 +208,8 @@ async def process_request(
             "query_string": query_string.encode("ascii"),
             "headers": asgi_headers,
             "subprotocols": subprotocols,
-            "extensions": {},
+            "extensions": {"tls": self.tls} if self.scheme == "wss" else {},
         }
-        if self.scheme == "wss":
-            self.scope["extensions"]["tls"] = self.tls
         task = self.loop.create_task(self.run_asgi())
         task.add_done_callback(self.on_task_complete)
         self.tasks.add(task)

diff --git a/uvicorn/protocols/websockets/wsproto_impl.py b/uvicorn/protocols/websockets/wsproto_impl.py
@@ -191,10 +191,8 @@ def handle_connect(self, event: events.Request) -> None:
             "query_string": query_string.encode("ascii"),
             "headers": headers,
             "subprotocols": event.subprotocols,
-            "extensions": {},
+            "extensions": {"tls": self.tls} if self.scheme == "wss" else {},
         }
-        if self.scheme == "wss":
-            self.scope["extensions"]["tls"] = self.tls
         self.queue.put_nowait({"type": "websocket.connect"})
         task = self.loop.create_task(self.run_asgi())
         task.add_done_callback(self.on_task_complete)