feat: add callback register for websocket

teslajsonpy/controller.py

@@ -76,6 +76,7 @@ def __init__(
         self.__vin_id_map = {}
         self.__vin_vehicle_id_map = {}
         self.__vehicle_id_vin_map = {}
+        self.__websocket_listeners = []
 
     async def connect(self, test_login=False) -> Tuple[Text, Text]:
         """Connect controller to Tesla."""
@@ -148,6 +149,19 @@ def get_tokens(self) -> Tuple[Text, Text]:
         self.__connection.token_refreshed = False
         return (self.__connection.refresh_token, self.__connection.access_token)
 
+    def register_websocket_callback(self, callback) -> int:
+        """Register callback for websocket messages.
+
+        Args
+            callback (function): function to call with json data
+
+        Returns
+            int: Return index of entry
+
+        """
+        self.__websocket_listeners.append(callback)
+        return len(self.__websocket_listeners) - 1
+
     def wake_up(func):
         #  pylint: disable=no-self-argument
         #  issue is use of wraps on classmethods which should be replaced:
@@ -676,7 +690,6 @@ def _process_websocket_message(self, data):
             update_json = {}
             vehicle_id = int(data["tag"])
             vin = self.__vehicle_id_vin_map[vehicle_id]
-            _LOGGER.debug("Updating %s with websocket", vin[-5:])
             keys = [
                 ("timestamp", int),
                 ("speed", int),
@@ -695,7 +708,7 @@ def _process_websocket_message(self, data):
             values = data["value"].split(",")
             for num, value in enumerate(values):
                 update_json[keys[num][0]] = keys[num][1](value) if value else None
-            _LOGGER.debug("Update_json %s", update_json)
+            _LOGGER.debug("Updating %s with websocket: %s", vin[-5:], update_json)
             self.__driving[vin]["timestamp"] = update_json["timestamp"]
             self.__charging[vin]["timestamp"] = update_json["timestamp"]
             self.__state[vin]["timestamp"] = update_json["timestamp"]
@@ -713,3 +726,5 @@ def _process_websocket_message(self, data):
             self.__charging[vin]["est_battery_range"] = update_json["est_range"]
             # self.__driving[vin]["heading"] = update_json["heading"]
             # est_heading appears more accurate
+        for func in self.__websocket_listeners:
+            func(data)