Compare commits

..

No commits in common. "c3cb93d903b9e0a6bc60c7f1e3bb7dae8a71118e" and "a00cfb688c5de58afca0df8c3a5eec2cfd4f1566" have entirely different histories.

40
main.py
View file

@ -3,7 +3,7 @@ import json
import sqlite3 import sqlite3
from typing import Dict from typing import Dict
from fastapi import FastAPI, Header, HTTPException, Query, WebSocket, WebSocketDisconnect from fastapi import FastAPI, Header, HTTPException, Query
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from fastapi.routing import APIRoute from fastapi.routing import APIRoute
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
@ -11,8 +11,6 @@ from pydantic import BaseModel
from typing import Optional from typing import Optional
from db import init_db, save_snapshot, DB_FILE from db import init_db, save_snapshot, DB_FILE
import asyncio
from starlette.concurrency import run_in_threadpool
# ------------------------------------------------------------------ # ------------------------------------------------------------------
app = FastAPI() app = FastAPI()
@ -196,42 +194,6 @@ def get_trails(
] ]
return JSONResponse(content={"trails": trails}) return JSONResponse(content={"trails": trails})
# -------------------- WebSocket endpoints -----------------------
browser_conns: set[WebSocket] = set()
async def _broadcast_to_browser_clients(snapshot: dict):
for ws in list(browser_conns):
try:
await ws.send_json(snapshot)
except WebSocketDisconnect:
browser_conns.remove(ws)
@app.websocket("/ws/position")
async def ws_receive_snapshots(websocket: WebSocket, secret: str = Query(...)):
await websocket.accept()
if secret != SHARED_SECRET:
await websocket.close(code=1008)
return
try:
while True:
data = await websocket.receive_json()
snap = TelemetrySnapshot.parse_obj(data)
live_snapshots[snap.character_name] = snap.dict()
await run_in_threadpool(save_snapshot, snap.dict())
await _broadcast_to_browser_clients(snap.dict())
except WebSocketDisconnect:
pass
@app.websocket("/ws/live")
async def ws_live_updates(websocket: WebSocket):
await websocket.accept()
browser_conns.add(websocket)
try:
while True:
await asyncio.sleep(3600)
except WebSocketDisconnect:
browser_conns.remove(websocket)
# -------------------- static frontend --------------------------- # -------------------- static frontend ---------------------------
app.mount("/", StaticFiles(directory="static", html=True), name="static") app.mount("/", StaticFiles(directory="static", html=True), name="static")