Add domain_suffix option

This commit is contained in:
Meatballs1
2024-05-13 12:15:51 +01:00
parent 1935feb02a
commit c1e31e7c1a
2 changed files with 12 additions and 6 deletions

View File

@@ -27,6 +27,7 @@ from HCSocket import HCSocket, now
@click.option("--mqtt_certfile")
@click.option("--mqtt_keyfile")
@click.option("--mqtt_clientname", default="hcpy1")
@click.option("--domain_suffix", default="")
@click_config_file.configuration_option()
def hc2mqtt(
devices_file: str,
@@ -40,6 +41,7 @@ def hc2mqtt(
mqtt_certfile: str,
mqtt_keyfile: str,
mqtt_clientname: str,
domain_suffix: str
):
def on_connect(client, userdata, flags, rc):
@@ -101,6 +103,7 @@ def hc2mqtt(
f"Hello {devices_file=} {mqtt_host=} {mqtt_prefix=} "
f"{mqtt_port=} {mqtt_username=} {mqtt_password=} "
f"{mqtt_ssl=} {mqtt_cafile=} {mqtt_certfile=} {mqtt_keyfile=} {mqtt_clientname=}"
f"{domain_suffix=}"
)
with open(devices_file, "r") as f:
@@ -130,7 +133,7 @@ def hc2mqtt(
for device in devices:
mqtt_topic = mqtt_prefix + device["name"]
thread = Thread(target=client_connect, args=(client, device, mqtt_topic))
thread = Thread(target=client_connect, args=(client, device, mqtt_topic, domain_suffix))
thread.start()
client.loop_forever()
@@ -140,7 +143,7 @@ global dev
dev = {}
def client_connect(client, device, mqtt_topic):
def client_connect(client, device, mqtt_topic, domain_suffix):
host = device["host"]
name = device["name"]
@@ -192,7 +195,7 @@ def client_connect(client, device, mqtt_topic):
time.sleep(3)
try:
print(now(), name, f"connecting to {host}")
ws = HCSocket(host, device["key"], device.get("iv", None))
ws = HCSocket(host, device["key"], device.get("iv", None), domain_suffix)
dev[name] = HCDevice(ws, device)
dev[name].run_forever(on_message=on_message, on_open=on_open, on_close=on_close)
except Exception as e: