mduval/GeoRideHA
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add get_tracker to await

Browse Source
This commit is contained in:
Matthieu DUVAL
2021-04-10 12:05:38 +02:00
parent 1e21de655f
commit f72481236d
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
custom_components/georide/__init__.py
View File

@@ -85,7 +85,7 @@ async def async_setup_entry(hass, entry):
hass.data[DOMAIN]["context"] = context hass.data[DOMAIN]["context"] = context
# We add trackers to the context # We add trackers to the context
context.refresh_trackers() await context.refresh_trackers()
hass.async_create_task( hass.async_create_task(
hass.config_entries.async_forward_entry_setup(entry, "device_tracker")) hass.config_entries.async_forward_entry_setup(entry, "device_tracker"))
@@ -187,13 +187,13 @@ class GeoRideContext:
_LOGGER.info("Token exp data: %s", exp_timestamp) _LOGGER.info("Token exp data: %s", exp_timestamp)
return self._token return self._token
def get_tracker(self, tracker_id): async def get_tracker(self, tracker_id):
""" here we return last tracker by id""" """ here we return last tracker by id"""
epoch_min = math.floor(time.time()/60) epoch_min = math.floor(time.time()/60)
if (epoch_min % MIN_UNTIL_REFRESH) == 0: if (epoch_min % MIN_UNTIL_REFRESH) == 0:
if epoch_min != self._previous_refresh: if epoch_min != self._previous_refresh:
self._previous_refresh = epoch_min self._previous_refresh = epoch_min
self.refresh_trackers() await self.refresh_trackers()
if not self._thread_started: if not self._thread_started:
_LOGGER.info("Start the thread") _LOGGER.info("Start the thread")
@@ -206,10 +206,10 @@ class GeoRideContext:
return tracker return tracker
return None return None
def refresh_trackers(self): async def refresh_trackers(self):
"""Used to refresh the tracker list""" """Used to refresh the tracker list"""
_LOGGER.info("Tracker list refresh") _LOGGER.info("Tracker list refresh")
self._georide_trackers = GeoRideApi.get_trackers(self.get_token()) self._georide_trackers = await GeoRideApi.get_trackers(await self.get_token())
@property @property
def socket(self): def socket(self):

2
custom_components/georide/binary_sensor.py
View File

@@ -19,7 +19,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): # pylint: d
if token is None: if token is None:
return False return False
trackers = GeoRideApi.get_trackers(token) trackers = await GeoRideApi.get_trackers(token)
binary_sensor_entities = [] binary_sensor_entities = []
for tracker in trackers: for tracker in trackers:

2
custom_components/georide/device_tracker.py
View File

@@ -23,7 +23,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): # pylint: d
_LOGGER.debug('Current GeoRide token: %s', token) _LOGGER.debug('Current GeoRide token: %s', token)
trackers = GeoRideApi.get_trackers(token) trackers = await GeoRideApi.get_trackers(token)
tracker_entities = [] tracker_entities = []

2
custom_components/georide/sensor.py
View File

@@ -21,7 +21,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): # pylint: d
if token is None: if token is None:
return False return False
trackers = GeoRideApi.get_trackers(token) trackers = await GeoRideApi.get_trackers(token)
odometer_switch_entities = [] odometer_switch_entities = []
for tracker in trackers: for tracker in trackers: