LailaTheElf/rcrf-wifi
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gs: fix conn check

Browse Source
This commit is contained in:
Laila van Reenen 2024-08-12 23:35:07 +02:00
parent b2dfdcc14a
commit 543e67929b
Signed by: LailaTheElf
GPG Key ID: 1F4E6EE3E6DDF769
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ground-station.py
View File

@ -17,13 +17,15 @@ Boats = []
Clients = []
async def sendToClient(client, msg):
if client['state'] != 'terminated' and client['ws'].connected:
await client['ws'].send(msg)
elif client['state'] != 'kicked':
client['state'] = 'terminated'
if client['boat'] is not None:
client['boat']['state'] = BOAT_STATE_AVAILABLE
client['boat'] = None
if client['state'] != 'terminated':
try:
await client['ws'].send(msg)
except ConnectionClosed:
if client['state'] != 'kicked':
client['state'] = 'terminated'
if client['boat'] is not None:
client['boat']['state'] = BOAT_STATE_AVAILABLE
client['boat'] = None
async def sendToBoat(boat, cmd):
boat['log'].append({"t": datetime.timestamp(datetime.now()), "type": "tx", "msg": cmd})
@ -195,10 +197,11 @@ async def new_boat(boatId, name, ws):
"""handler for every new boat connection"""
boat = { "id": boatId, "name": name, "ws": ws, "state": BOAT_STATE_AVAILABLE, "lastMsg": 0, "log": [] }
print("new boat connected: " + boatId)
for bo in Boats:
for i, bo in enumerate(Boats):
if bo['id'] == boat['id']:
bo['state'] = BOAT_STATE_TERMINATED
await bo['ws'].close()
# await bo['ws'].close()
# del Boats[i]
Boats.append(boat)
return boat