Move connect_socket into the context

custom_components/georide/__init__.py

@@ -114,19 +114,6 @@ async def async_unload_entry(hass, entry):
 
     return True
 
-async def connect_socket(context):
-    """subscribe to GeoRide socket"""
-    _LOGGER.info("GeoRide socket connexion")
-    socket = GeoRideSocket()
-    socket.subscribe_locked(context.on_lock_callback)
-    socket.subscribe_device(context.on_device_callback)
-    socket.subscribe_position(context.on_position_callback)
-    socket.subscribe_alarm(context.on_alarm_callback)
-
-    context.socket = socket
-
-    socket.init()
-    socket.connect(await context.get_token())
 
 
 class GeoRideContext:
@@ -172,6 +159,20 @@ class GeoRideContext:
         """ GeoRide tracker list """
         self._georide_trackers = trackers
 
+    async def connect_socket(self):
+        """subscribe to GeoRide socket"""
+        _LOGGER.info("GeoRide socket connexion")
+        socket = GeoRideSocket()
+        socket.subscribe_locked(self.on_lock_callback)
+        socket.subscribe_device(self.on_device_callback)
+        socket.subscribe_position(self.on_position_callback)
+        socket.subscribe_alarm(self.on_alarm_callback)
+
+        self._socket = socket
+
+        socket.init()
+        socket.connect(await self.get_token())
+
     async def get_token(self):
         """ here we return the current valid tocken """
         jwt_data = jwt.decode(self._token, verify=False)
@@ -180,8 +181,8 @@ class GeoRideContext:
         epoch = math.ceil(time.time())
         if (exp_timestamp - TOKEN_SAFE_DAY) < epoch:
             _LOGGER.info("Time reached, renew token")
-
-            account = await self._hass.async_add_executor_job(GeoRideApi.get_authorisation_token, self._email, self._password)
+            account = await self._hass.async_add_executor_job(GeoRideApi.get_authorisation_token,
+                                                              self._email, self._password)
             config = self._hass.data[DOMAIN]["config"]
             config[CONF_TOKEN] = account.auth_token
             self._token = account.auth_token
@@ -200,7 +201,7 @@ class GeoRideContext:
 
         if not self._thread_started:
             _LOGGER.info("Start the thread")
-            self._hass.async_add_executor_job(connect_socket, self)
+            await self.connect_socket()
             # We refresh the tracker list each hours
             self._thread_started = True
 
@@ -212,7 +213,8 @@ class GeoRideContext:
     async def refresh_trackers(self):
         """Used to refresh the tracker list"""
         _LOGGER.info("Tracker list refresh")
-        self._georide_trackers = await self._hass.async_add_executor_job(GeoRideApi.get_trackers, await self.get_token())
+        self._georide_trackers = await self._hass.async_add_executor_job(GeoRideApi.get_trackers,
+                                                                         await self.get_token())
 
     @property
     def socket(self):

custom_components/georide/switch.py

@@ -86,9 +86,9 @@ class GeoRideLockSwitchEntity(SwitchEntity):
 
     async def async_update(self):
         """ update the current tracker"""
-        _LOGGER.info('update')
+        _LOGGER.info('update %s', self._tracker_id)
         self._data = await self._get_tracker_callback(self._tracker_id)
-        _LOGGER.debug('data %s', self._data)
+        _LOGGER.debug('data %s', self._data.is_locked)
 
         self._name = self._data.tracker_name
         self._is_on = self._data.is_locked