Update base_device.py

pull/255/head
Keyvan Hardani 9 months ago committed by GitHub
parent 4d66490f02
commit 0a81c665ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -291,18 +291,18 @@ class Device:
# Receive authentication request from the server # Receive authentication request from the server
auth_request = await websocket.recv() auth_request = await websocket.recv()
auth_data = json.loads(auth_request) auth_data = json.loads(auth_request)
if auth_data["type"] == "auth_request": if auth_data["type"] == "auth_request":
# Send authentication response with the token # Send authentication response with the token
token = os.getenv("WS_TOKEN") token = os.getenv("WS_TOKEN")
if token: if token:
auth_response = {"token": token} auth_response = {"token": token}
await websocket.send(json.dumps(auth_response)) await websocket.send(json.dumps(auth_response))
# Receive authentication result from the server # Receive authentication result from the server
auth_result = await websocket.recv() auth_result = await websocket.recv()
result_data = json.loads(auth_result) result_data = json.loads(auth_result)
if result_data["type"] == "auth_success": if result_data["type"] == "auth_success":
# Authentication successful # Authentication successful
return True return True

Loading…
Cancel
Save