mirror of
https://github.com/kennethreitz/responder.git
synced 2026-06-05 06:46:14 +00:00
kennethreitz
6f9c87d71c
- Call future.result() instead of bare property access in test (#596) - Catch (ValueError, TypeError) instead of broad Exception in response model serialization (#597) - Catch WebSocketDisconnect instead of broad Exception in websocket chat example (#598) Closes #596, closes #597, closes #598 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
# WebSocket chat room example.
|
|
# https://responder.kennethreitz.org/tutorial-websockets.html
|
|
from starlette.websockets import WebSocketDisconnect
|
|
|
|
import responder
|
|
|
|
api = responder.API()
|
|
|
|
connected = set()
|
|
|
|
|
|
@api.route("/")
|
|
def index(req, resp):
|
|
resp.html = """
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<body>
|
|
<h1>Chat Room</h1>
|
|
<div id="messages" style="height:300px;overflow-y:scroll;border:1px solid #ccc;padding:10px;"></div>
|
|
<input id="input" placeholder="Type a message..." style="width:300px;" />
|
|
<script>
|
|
const ws = new WebSocket(`ws://${location.host}/chat`);
|
|
const messages = document.getElementById("messages");
|
|
const input = document.getElementById("input");
|
|
ws.onmessage = (e) => {
|
|
const p = document.createElement("p");
|
|
p.textContent = e.data;
|
|
messages.appendChild(p);
|
|
messages.scrollTop = messages.scrollHeight;
|
|
};
|
|
input.addEventListener("keypress", (e) => {
|
|
if (e.key === "Enter" && input.value) {
|
|
ws.send(input.value);
|
|
input.value = "";
|
|
}
|
|
});
|
|
</script>
|
|
</body>
|
|
</html>
|
|
""" # noqa: E501
|
|
|
|
|
|
@api.route("/chat", websocket=True)
|
|
async def chat(ws):
|
|
await ws.accept()
|
|
connected.add(ws)
|
|
try:
|
|
while True:
|
|
message = await ws.receive_text()
|
|
for client in connected:
|
|
await client.send_text(message)
|
|
except WebSocketDisconnect:
|
|
pass
|
|
finally:
|
|
connected.discard(ws)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
api.run()
|