mirror of
https://github.com/joBr99/nspanel-lovelace-ui.git
synced 2026-02-04 12:28:56 +01:00
.
This commit is contained in:
joBr99
@@ -1,6 +1,6 @@
|
||||
# https://developers.home-assistant.io/docs/add-ons/configuration#add-on-config
|
||||
name: NSPanel Lovelace UI Addon
|
||||
version: "4.7.44"
|
||||
version: "4.7.46"
|
||||
slug: nspanel-lovelace-ui
|
||||
description: NSPanel Lovelace UI Addon
|
||||
services:
|
||||
@@ -17,15 +17,8 @@ init: false
|
||||
map:
|
||||
- addon_config:rw
|
||||
options:
|
||||
loglevel:
|
||||
- DEBUG
|
||||
- INFO
|
||||
- WARNING
|
||||
- ERROR
|
||||
- CRITICAL
|
||||
logtofile:
|
||||
- true
|
||||
- false
|
||||
loglevel: DEBUG
|
||||
logtofile: true
|
||||
schema:
|
||||
loglevel: list(DEBUG|INFO|WARNING|ERROR|CRITICAL)
|
||||
logtofile: bool
|
||||
|
||||
@@ -1,261 +1,261 @@
|
||||
import websocket
|
||||
import ssl
|
||||
import logging
|
||||
import json
|
||||
from threading import Thread
|
||||
import time
|
||||
import os
|
||||
|
||||
home_assistant_url = ""
|
||||
home_assistant_token = ""
|
||||
settings = {}
|
||||
auth_ok = False
|
||||
next_id = 0
|
||||
request_all_states_id = 0
|
||||
ws_connected = False
|
||||
home_assistant_entity_state_cache = {}
|
||||
template_cache = {}
|
||||
response_buffer = {}
|
||||
|
||||
|
||||
ON_CONNECT_HANDLER = None
|
||||
ON_DISCONNECT_HANDLER = None
|
||||
|
||||
|
||||
def init(settings_from_manager, on_ha_update_from_manager):
|
||||
global home_assistant_url, home_assistant_token, settings, on_ha_update
|
||||
settings = settings_from_manager
|
||||
on_ha_update = on_ha_update_from_manager
|
||||
home_assistant_url = settings["home_assistant_address"]
|
||||
home_assistant_token = settings["home_assistant_token"]
|
||||
# Disable logging from underlying "websocket"
|
||||
logging.getLogger("websocket").propagate = False
|
||||
# Disable logging from underlying "websocket"
|
||||
logging.getLogger("websockets").propagate = False
|
||||
|
||||
|
||||
def register_on_connect_handler(handler):
|
||||
global ON_CONNECT_HANDLER
|
||||
ON_CONNECT_HANDLER = handler
|
||||
|
||||
|
||||
def register_on_disconnect_handler(handler):
|
||||
global ON_DISCONNECT_HANDLER
|
||||
ON_DISCONNECT_HANDLER = handler
|
||||
|
||||
|
||||
def on_message(ws, message):
|
||||
global auth_ok, request_all_states_id, home_assistant_entity_state_cache, response_buffer, template_cache
|
||||
json_msg = json.loads(message)
|
||||
if json_msg["type"] == "auth_required":
|
||||
authenticate_client()
|
||||
elif json_msg["type"] == "auth_ok":
|
||||
auth_ok = True
|
||||
logging.info("Home Assistant auth OK. Requesting existing states.")
|
||||
subscribe_to_events()
|
||||
_get_all_states()
|
||||
if ON_CONNECT_HANDLER is not None:
|
||||
ON_CONNECT_HANDLER()
|
||||
# for templates
|
||||
elif json_msg["type"] == "event" and json_msg["id"] in response_buffer:
|
||||
template_cache[response_buffer[json_msg["id"]]] = {
|
||||
"result": json_msg["event"]["result"],
|
||||
"listener-entities": json_msg["event"]["listeners"]["entities"]
|
||||
}
|
||||
elif json_msg["type"] == "event" and json_msg["event"]["event_type"] == "state_changed":
|
||||
entity_id = json_msg["event"]["data"]["entity_id"]
|
||||
home_assistant_entity_state_cache[entity_id] = json_msg["event"]["data"]["new_state"]
|
||||
send_entity_update(entity_id)
|
||||
# rerender template
|
||||
for template, template_cache_entry in template_cache.items():
|
||||
if entity_id in template_cache_entry.get("listener-entities", []):
|
||||
cache_template(template)
|
||||
|
||||
elif json_msg["type"] == "result" and not json_msg["success"]:
|
||||
logging.error("Failed result: ")
|
||||
logging.error(json_msg)
|
||||
elif json_msg["type"] == "result" and json_msg["success"]:
|
||||
if json_msg["id"] == request_all_states_id:
|
||||
for entity in json_msg["result"]:
|
||||
home_assistant_entity_state_cache[entity["entity_id"]] = entity
|
||||
else:
|
||||
if json_msg["id"] in response_buffer and json_msg.get("result"):
|
||||
response_buffer[json_msg["id"]] = json_msg["result"]
|
||||
return None # Ignore success result messages
|
||||
else:
|
||||
logging.debug(message)
|
||||
|
||||
|
||||
def _ws_connection_open(ws):
|
||||
global ws_connected
|
||||
ws_connected = True
|
||||
logging.info("WebSocket connection to Home Assistant opened.")
|
||||
if ON_CONNECT_HANDLER is not None:
|
||||
ON_CONNECT_HANDLER()
|
||||
|
||||
|
||||
def _ws_connection_close(ws, close_status_code, close_msg):
|
||||
global ws_connected
|
||||
ws_connected = False
|
||||
logging.error("WebSocket connection closed!")
|
||||
if ON_DISCONNECT_HANDLER is not None:
|
||||
ON_DISCONNECT_HANDLER()
|
||||
|
||||
|
||||
def connect():
|
||||
Thread(target=_do_connection, daemon=True).start()
|
||||
|
||||
|
||||
def _do_connection():
|
||||
global home_assistant_url, ws, settings
|
||||
ws_url = home_assistant_url.replace(
|
||||
"https://", "wss://").replace("http://", "ws://")
|
||||
if settings["is_addon"]:
|
||||
ws_url += "/core/websocket"
|
||||
else:
|
||||
ws_url += "/api/websocket"
|
||||
ws = websocket.WebSocketApp(F"{ws_url}", on_message=on_message,
|
||||
on_open=_ws_connection_open, on_close=_ws_connection_close)
|
||||
while True:
|
||||
logging.info(F"Connecting to Home Assistant at {ws_url}")
|
||||
ws.close()
|
||||
time.sleep(1)
|
||||
ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})
|
||||
time.sleep(10)
|
||||
|
||||
|
||||
def authenticate_client():
|
||||
global home_assistant_token
|
||||
logging.info("Sending auth to Home Assistant")
|
||||
msg = {
|
||||
"type": "auth",
|
||||
"access_token": home_assistant_token
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
|
||||
|
||||
def subscribe_to_events():
|
||||
global next_id
|
||||
msg = {
|
||||
"id": next_id,
|
||||
"type": "subscribe_events",
|
||||
"event_type": "state_changed"
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
|
||||
|
||||
def _get_all_states():
|
||||
global next_id, request_all_states_id
|
||||
msg = {
|
||||
"id": next_id,
|
||||
"type": "get_states",
|
||||
}
|
||||
request_all_states_id = next_id
|
||||
send_message(json.dumps(msg))
|
||||
|
||||
# Got new value from Home Assistant, send update to callback method
|
||||
def send_entity_update(entity_id):
|
||||
global on_ha_update
|
||||
on_ha_update(entity_id)
|
||||
|
||||
def call_service(entity_name: str, domain: str, service: str, service_data: dict) -> bool:
|
||||
global next_id
|
||||
try:
|
||||
msg = {
|
||||
"id": next_id,
|
||||
"type": "call_service",
|
||||
"domain": domain,
|
||||
"service": service,
|
||||
"service_data": service_data,
|
||||
"target": {
|
||||
"entity_id": entity_name
|
||||
},
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
return True
|
||||
except Exception as e:
|
||||
logging.exception("Failed to call Home Assisatant service.")
|
||||
return False
|
||||
|
||||
def execute_script(entity_name: str, domain: str, service: str, service_data: dict) -> str:
|
||||
global next_id, response_buffer
|
||||
try:
|
||||
call_id = next_id
|
||||
# request answer for this call
|
||||
response_buffer[call_id] = True
|
||||
msg = {
|
||||
"id": call_id,
|
||||
"type": "execute_script",
|
||||
"sequence": [
|
||||
{
|
||||
"service": f"{domain}.{service}",
|
||||
"data": service_data,
|
||||
"target": {
|
||||
"entity_id": [entity_name]
|
||||
},
|
||||
"response_variable": "service_result"
|
||||
},
|
||||
{
|
||||
"stop": "done",
|
||||
"response_variable": "service_result"
|
||||
}
|
||||
]
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
# busy waiting for response with a timeout of 0.2 seconds - maybe there's a better way of doing this
|
||||
mustend = time.time() + 0.4
|
||||
while time.time() < mustend:
|
||||
if response_buffer[call_id] == True:
|
||||
#print(f'loooooooooop {time.time()}')
|
||||
time.sleep(0.0001)
|
||||
else:
|
||||
return response_buffer[call_id]["response"]
|
||||
raise TimeoutError("Did not recive respose in time to HA script call")
|
||||
except Exception as e:
|
||||
logging.exception("Failed to call Home Assisatant script.")
|
||||
return False
|
||||
|
||||
def cache_template(template):
|
||||
global next_id, response_buffer
|
||||
try:
|
||||
call_id = next_id
|
||||
response_buffer[call_id] = template
|
||||
msg = {
|
||||
"id": call_id,
|
||||
"type": "render_template",
|
||||
"template": template
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
return True
|
||||
except Exception as e:
|
||||
logging.exception("Failed to render template.")
|
||||
return False
|
||||
|
||||
def get_template(template):
|
||||
global template_cache
|
||||
if template in template_cache:
|
||||
return template_cache[template].get("result")
|
||||
else:
|
||||
mustend = time.time() + 0.5
|
||||
while time.time() < mustend:
|
||||
if template not in template_cache:
|
||||
time.sleep(0.0001)
|
||||
else:
|
||||
return template_cache.get(template, []).get("result", "404")
|
||||
|
||||
def get_entity_data(entity_id: str):
|
||||
if entity_id in home_assistant_entity_state_cache:
|
||||
return home_assistant_entity_state_cache[entity_id]
|
||||
else:
|
||||
return None
|
||||
def is_existent(entity_id: str):
|
||||
if entity_id in home_assistant_entity_state_cache:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
def send_message(message):
|
||||
global ws, next_id
|
||||
next_id += 1
|
||||
ws.send(message)
|
||||
import websocket
|
||||
import ssl
|
||||
import logging
|
||||
import json
|
||||
from threading import Thread
|
||||
import time
|
||||
import os
|
||||
|
||||
home_assistant_url = ""
|
||||
home_assistant_token = ""
|
||||
settings = {}
|
||||
auth_ok = False
|
||||
next_id = 0
|
||||
request_all_states_id = 0
|
||||
ws_connected = False
|
||||
home_assistant_entity_state_cache = {}
|
||||
template_cache = {}
|
||||
response_buffer = {}
|
||||
|
||||
|
||||
ON_CONNECT_HANDLER = None
|
||||
ON_DISCONNECT_HANDLER = None
|
||||
|
||||
|
||||
def init(settings_from_manager, on_ha_update_from_manager):
|
||||
global home_assistant_url, home_assistant_token, settings, on_ha_update
|
||||
settings = settings_from_manager
|
||||
on_ha_update = on_ha_update_from_manager
|
||||
home_assistant_url = settings["home_assistant_address"]
|
||||
home_assistant_token = settings["home_assistant_token"]
|
||||
# Disable logging from underlying "websocket"
|
||||
logging.getLogger("websocket").propagate = False
|
||||
# Disable logging from underlying "websocket"
|
||||
logging.getLogger("websockets").propagate = False
|
||||
|
||||
|
||||
def register_on_connect_handler(handler):
|
||||
global ON_CONNECT_HANDLER
|
||||
ON_CONNECT_HANDLER = handler
|
||||
|
||||
|
||||
def register_on_disconnect_handler(handler):
|
||||
global ON_DISCONNECT_HANDLER
|
||||
ON_DISCONNECT_HANDLER = handler
|
||||
|
||||
|
||||
def on_message(ws, message):
|
||||
global auth_ok, request_all_states_id, home_assistant_entity_state_cache, response_buffer, template_cache
|
||||
json_msg = json.loads(message)
|
||||
if json_msg["type"] == "auth_required":
|
||||
authenticate_client()
|
||||
elif json_msg["type"] == "auth_ok":
|
||||
auth_ok = True
|
||||
logging.info("Home Assistant auth OK. Requesting existing states.")
|
||||
subscribe_to_events()
|
||||
_get_all_states()
|
||||
if ON_CONNECT_HANDLER is not None:
|
||||
ON_CONNECT_HANDLER()
|
||||
# for templates
|
||||
elif json_msg["type"] == "event" and json_msg["id"] in response_buffer:
|
||||
template_cache[response_buffer[json_msg["id"]]] = {
|
||||
"result": json_msg["event"]["result"],
|
||||
"listener-entities": json_msg["event"]["listeners"]["entities"]
|
||||
}
|
||||
elif json_msg["type"] == "event" and json_msg["event"]["event_type"] == "state_changed":
|
||||
entity_id = json_msg["event"]["data"]["entity_id"]
|
||||
home_assistant_entity_state_cache[entity_id] = json_msg["event"]["data"]["new_state"]
|
||||
send_entity_update(entity_id)
|
||||
# rerender template
|
||||
for template, template_cache_entry in template_cache.items():
|
||||
if entity_id in template_cache_entry.get("listener-entities", []):
|
||||
cache_template(template)
|
||||
|
||||
elif json_msg["type"] == "result" and not json_msg["success"]:
|
||||
logging.error("Failed result: ")
|
||||
logging.error(json_msg)
|
||||
elif json_msg["type"] == "result" and json_msg["success"]:
|
||||
if json_msg["id"] == request_all_states_id:
|
||||
for entity in json_msg["result"]:
|
||||
home_assistant_entity_state_cache[entity["entity_id"]] = entity
|
||||
else:
|
||||
if json_msg["id"] in response_buffer and json_msg.get("result"):
|
||||
response_buffer[json_msg["id"]] = json_msg["result"]
|
||||
return None # Ignore success result messages
|
||||
else:
|
||||
logging.debug(message)
|
||||
|
||||
|
||||
def _ws_connection_open(ws):
|
||||
global ws_connected
|
||||
ws_connected = True
|
||||
logging.info("WebSocket connection to Home Assistant opened.")
|
||||
if ON_CONNECT_HANDLER is not None:
|
||||
ON_CONNECT_HANDLER()
|
||||
|
||||
|
||||
def _ws_connection_close(ws, close_status_code, close_msg):
|
||||
global ws_connected
|
||||
ws_connected = False
|
||||
logging.error("WebSocket connection closed!")
|
||||
if ON_DISCONNECT_HANDLER is not None:
|
||||
ON_DISCONNECT_HANDLER()
|
||||
|
||||
|
||||
def connect():
|
||||
Thread(target=_do_connection, daemon=True).start()
|
||||
|
||||
|
||||
def _do_connection():
|
||||
global home_assistant_url, ws, settings
|
||||
ws_url = home_assistant_url.replace(
|
||||
"https://", "wss://").replace("http://", "ws://")
|
||||
if settings["is_addon"]:
|
||||
ws_url += "/core/websocket"
|
||||
else:
|
||||
ws_url += "/api/websocket"
|
||||
ws = websocket.WebSocketApp(F"{ws_url}", on_message=on_message,
|
||||
on_open=_ws_connection_open, on_close=_ws_connection_close)
|
||||
while True:
|
||||
logging.info(F"Connecting to Home Assistant at {ws_url}")
|
||||
ws.close()
|
||||
time.sleep(1)
|
||||
ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})
|
||||
time.sleep(10)
|
||||
|
||||
|
||||
def authenticate_client():
|
||||
global home_assistant_token
|
||||
logging.info("Sending auth to Home Assistant")
|
||||
msg = {
|
||||
"type": "auth",
|
||||
"access_token": home_assistant_token
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
|
||||
|
||||
def subscribe_to_events():
|
||||
global next_id
|
||||
msg = {
|
||||
"id": next_id,
|
||||
"type": "subscribe_events",
|
||||
"event_type": "state_changed"
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
|
||||
|
||||
def _get_all_states():
|
||||
global next_id, request_all_states_id
|
||||
msg = {
|
||||
"id": next_id,
|
||||
"type": "get_states",
|
||||
}
|
||||
request_all_states_id = next_id
|
||||
send_message(json.dumps(msg))
|
||||
|
||||
# Got new value from Home Assistant, send update to callback method
|
||||
def send_entity_update(entity_id):
|
||||
global on_ha_update
|
||||
on_ha_update(entity_id)
|
||||
|
||||
def call_service(entity_name: str, domain: str, service: str, service_data: dict) -> bool:
|
||||
global next_id
|
||||
try:
|
||||
msg = {
|
||||
"id": next_id,
|
||||
"type": "call_service",
|
||||
"domain": domain,
|
||||
"service": service,
|
||||
"service_data": service_data,
|
||||
"target": {
|
||||
"entity_id": entity_name
|
||||
},
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
return True
|
||||
except Exception as e:
|
||||
logging.exception("Failed to call Home Assisatant service.")
|
||||
return False
|
||||
|
||||
def execute_script(entity_name: str, domain: str, service: str, service_data: dict) -> str:
|
||||
global next_id, response_buffer
|
||||
try:
|
||||
call_id = next_id
|
||||
# request answer for this call
|
||||
response_buffer[call_id] = True
|
||||
msg = {
|
||||
"id": call_id,
|
||||
"type": "execute_script",
|
||||
"sequence": [
|
||||
{
|
||||
"service": f"{domain}.{service}",
|
||||
"data": service_data,
|
||||
"target": {
|
||||
"entity_id": [entity_name]
|
||||
},
|
||||
"response_variable": "service_result"
|
||||
},
|
||||
{
|
||||
"stop": "done",
|
||||
"response_variable": "service_result"
|
||||
}
|
||||
]
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
# busy waiting for response with a timeout of 0.2 seconds - maybe there's a better way of doing this
|
||||
mustend = time.time() + 0.4
|
||||
while time.time() < mustend:
|
||||
if response_buffer[call_id] == True:
|
||||
#print(f'loooooooooop {time.time()}')
|
||||
time.sleep(0.0001)
|
||||
else:
|
||||
return response_buffer[call_id]["response"]
|
||||
raise TimeoutError("Did not recive respose in time to HA script call")
|
||||
except Exception as e:
|
||||
logging.exception("Failed to call Home Assisatant script.")
|
||||
return {}
|
||||
|
||||
def cache_template(template):
|
||||
global next_id, response_buffer
|
||||
try:
|
||||
call_id = next_id
|
||||
response_buffer[call_id] = template
|
||||
msg = {
|
||||
"id": call_id,
|
||||
"type": "render_template",
|
||||
"template": template
|
||||
}
|
||||
send_message(json.dumps(msg))
|
||||
return True
|
||||
except Exception as e:
|
||||
logging.exception("Failed to render template.")
|
||||
return False
|
||||
|
||||
def get_template(template):
|
||||
global template_cache
|
||||
if template in template_cache:
|
||||
return template_cache[template].get("result")
|
||||
else:
|
||||
mustend = time.time() + 0.5
|
||||
while time.time() < mustend:
|
||||
if template not in template_cache:
|
||||
time.sleep(0.0001)
|
||||
else:
|
||||
return template_cache.get(template, []).get("result", "404")
|
||||
|
||||
def get_entity_data(entity_id: str):
|
||||
if entity_id in home_assistant_entity_state_cache:
|
||||
return home_assistant_entity_state_cache[entity_id]
|
||||
else:
|
||||
return None
|
||||
def is_existent(entity_id: str):
|
||||
if entity_id in home_assistant_entity_state_cache:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
def send_message(message):
|
||||
global ws, next_id
|
||||
next_id += 1
|
||||
ws.send(message)
|
||||
|
||||
Reference in New Issue
Block a user