[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2024-03-26 22:50:08 +00:00
parent 935543f81a
commit cc3dc538a3
3 changed files with 16 additions and 10 deletions

View File

@@ -43,9 +43,9 @@
import json import json
import re import re
import sys import sys
import traceback
import threading import threading
import time import time
import traceback
from base64 import urlsafe_b64encode as base64url_encode from base64 import urlsafe_b64encode as base64url_encode
from datetime import datetime from datetime import datetime
@@ -338,7 +338,6 @@ class HCDevice:
}, },
) )
threading.Thread(target=self.reconnect).start() threading.Thread(target=self.reconnect).start()
else: else:
self.print("Unknown resource", resource, file=sys.stderr) self.print("Unknown resource", resource, file=sys.stderr)
@@ -402,17 +401,21 @@ class HCDevice:
def _on_message(ws, message): def _on_message(ws, message):
values = self.handle_message(message) values = self.handle_message(message)
on_message(values) on_message(values)
def _on_open(ws): def _on_open(ws):
self.connected = True self.connected = True
on_open(ws) on_open(ws)
def _on_close(ws, code, message): def _on_close(ws, code, message):
self.connected = False self.connected = False
on_close(ws, code, message) on_close(ws, code, message)
def on_error(ws, message): def on_error(ws, message):
self.print("Websocket error:", message) self.print("Websocket error:", message)
self.ws.run_forever(on_message=_on_message, on_open=_on_open, on_close=_on_close, self.ws.run_forever(
on_error=on_error) on_message=_on_message, on_open=_on_open, on_close=_on_close, on_error=on_error
)
def print(self, *args): def print(self, *args):
print(now(), self.name, *args) print(now(), self.name, *args)

View File

@@ -201,12 +201,13 @@ class HCSocket:
print(now(), "CON:", self.uri) print(now(), "CON:", self.uri)
self.ws = websocket.WebSocketApp( self.ws = websocket.WebSocketApp(
self.uri, self.uri,
socket=sock, socket=sock,
on_open=_on_open, on_open=_on_open,
on_message=_on_message, on_message=_on_message,
on_close=_on_close, on_close=_on_close,
on_error=_on_error) on_error=_on_error,
)
websocket.setdefaulttimeout(30) websocket.setdefaulttimeout(30)

View File

@@ -139,6 +139,7 @@ def hc2mqtt(
global dev global dev
dev = {} dev = {}
def client_connect(client, device, mqtt_topic): def client_connect(client, device, mqtt_topic):
host = device["host"] host = device["host"]
name = device["name"] name = device["name"]
@@ -201,5 +202,6 @@ def client_connect(client, device, mqtt_topic):
time.sleep(57) time.sleep(57)
if __name__ == "__main__": if __name__ == "__main__":
hc2mqtt() hc2mqtt()