Added code to wait for server connection

Author: makermelissa

MacroPad_RPC_Home_Assistant/code.py

@@ -48,7 +48,7 @@ class MqttError(Exception):
 def rpc_call(function, *args, **kwargs):
     response = rpc.call(function, *args, **kwargs)
     if response["error"]:
-        if response["error_type"]:
+        if response["error_type"] == "mqtt":
             raise MqttError(response["message"])
         raise RpcError(response["message"])
     return response["return_val"]
@@ -64,6 +64,15 @@ def update_key(key_number):
     else:
         macropad.pixels[key_number] = 0
 
+server_is_running = False
+print("Waiting for server...")
+while not server_is_running:
+    try:
+        server_is_running = rpc_call("is_running")
+        print("Connected")
+    except RpcError:
+        pass
+
 mqtt_init()
 last_macropad_encoder_value = macropad.encoder
 

MacroPad_RPC_Home_Assistant/rpc_ha_server.py

@@ -68,10 +68,14 @@ def mqtt_get_last_value(topic):
         return last_mqtt_messages[topic]
     return None
 
+def is_running():
+    return True
+
 def handle_rpc(packet):
     """This function will verify good data in packet,
     call the method with parameters, and generate a response
     packet as the return value"""
+    print("Received packet")
     func_name = packet['function']
     if func_name in PROTECTED_FUNCTIONS:
         return rpc.create_response_packet(error=True, message=f"{func_name}'() is a protected function and can not be called.")