waylon/learn-sql-model
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

wip

Browse source
This commit is contained in:
Waylon Walker 2023-06-14 09:02:50 -05:00
parent fa64c9aed1
commit 892a3c9a8a
No known key found for this signature in database
GPG key ID: 66E2BF2B4190EFE4
4 changed files with 94 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

30
learn_sql_model/api/websocket.py
View file

@ -101,14 +101,19 @@ async def websocket_endpoint_hero_echo(
):
config = get_config()
await websocket.accept()
last_heros = None
try:
with config.database.engine.connect() as con:
while True:
data = await websocket.receive_text()
hero = HeroUpdate.parse_raw(data)
heros = con.execute("SELECT * FROM hero").fetchall()
heros = Heros.parse_obj({"heros": heros})
# heros = con.execute("SELECT * FROM hero").fetchall()
# heros = Heros.parse_obj({"heros": heros})
heros = Heros.list(session=session)
if heros != last_heros:
await manager.broadcast(heros.json(), "heros")
last_heros = heros
hero.update(session=session)
console.print(heros)
await websocket.send_text(heros.json())
@ -119,21 +124,28 @@ async def websocket_endpoint_hero_echo(
print("connection closed")
@web_socket_router.websocket("/ws-hero-update")
async def websocket_endpoint_hero_update(
@web_socket_router.websocket("/ws-heros'")
async def websocket_endpoint_heros(
websocket: WebSocket,
session: Session = Depends(get_session),
):
await manager.connect(websocket, "heros")
await websocket.accept()
try:
while True:
data = await websocket.receive_text()
hero = HeroUpdate.parse_raw(data)
print(hero)
hero.update(session=session)
print("hero is updated")
...
# data = await websocket.receive_text()
# hero = HeroUpdate.parse_raw(data)
# heros = con.execute("SELECT * FROM hero").fetchall()
# heros = Heros.parse_obj({"heros": heros})
# hero.update(session=session)
# console.print(heros)
# await websocket.send_text(heros.json())
except WebSocketDisconnect:
print("disconnected")
manager.disconnect(websocket, "heros")
except ConnectionClosed:
manager.disconnect(websocket, "heros")
print("connection closed")