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

Co-authored-by: Aber <Chen_Ye@ahnu.edu.cn>

diff --git a/starlette/endpoints.py b/starlette/endpoints.py
index 73367c2575d0c719579eee137532e415b9065c5f..f2468a326d53755f38bfd413825c4c6c0672b556 100644 (file)
--- a/starlette/endpoints.py
+++ b/starlette/endpoints.py
@@ -80,7 +80,9 @@ class WebSocketEndpoint:
                     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