Avoid TypeError on websocket.disconnect when code is None (#1574

) Co-authored-by: Aber <Chen_Ye@ahnu.edu.cn>
encode · Apr 6, 2022 · 487f05d · 487f05d
1 parent 20d24a8
commit 487f05d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/starlette/endpoints.py b/starlette/endpoints.py
@@ -80,7 +80,9 @@ async def dispatch(self) -> None:
                     data = await self.decode(websocket, message)
                     await self.on_receive(websocket, data)
                 elif message["type"] == "websocket.disconnect":
-                    close_code = int(message.get("code", status.WS_1000_NORMAL_CLOSURE))
+                    close_code = int(
+                        message.get("code") or status.WS_1000_NORMAL_CLOSURE
+                    )
                     break
         except Exception as exc:
             close_code = status.WS_1011_INTERNAL_ERROR