Mirrors
/
ha_gehome
mirror of https://github.com/simbaja/ha_gehome.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

- changed to not reset appliance apis

This commit is contained in:
Jack Simbach 2021-01-04 13:42:17 -05:00
parent 7d95de1532
commit 5c8d2eac6d
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ge_kitchen/devices/base.py
View File

@ -45,6 +45,10 @@ class ApplianceApi:
def appliance(self) -> GeAppliance: def appliance(self) -> GeAppliance:
return self._appliance return self._appliance
@appliance.setter
def appliance(self, value: GeAppliance):
self._appliance = value
@property @property
def available(self) -> bool: def available(self) -> bool:
#Note - online will be there since we're using the GE coordinator #Note - online will be there since we're using the GE coordinator

11
ge_kitchen/update_coordinator.py
View File

@ -47,13 +47,18 @@ class GeKitchenUpdateCoordinator(DataUpdateCoordinator):
self._config_entry = config_entry self._config_entry = config_entry
self._username = config_entry.data[CONF_USERNAME] self._username = config_entry.data[CONF_USERNAME]
self._password = config_entry.data[CONF_PASSWORD] self._password = config_entry.data[CONF_PASSWORD]
self._appliance_apis = {} # type: Dict[str, ApplianceApi]
self._reset_initialization() self._reset_initialization()
super().__init__(hass, _LOGGER, name=DOMAIN) super().__init__(hass, _LOGGER, name=DOMAIN)
def _reset_initialization(self): def _reset_initialization(self):
self.client = None # type: Optional[GeWebsocketClient] self.client = None # type: Optional[GeWebsocketClient]
self._appliance_apis = {} # type: Dict[str, ApplianceApi]
# Mark all appliances as not initialized yet
for a in self.appliance_apis.values():
a.appliance.initialized = False
# Some record keeping to let us know when we can start generating entities # Some record keeping to let us know when we can start generating entities
self._got_roster = False self._got_roster = False
@ -117,6 +122,10 @@ class GeKitchenUpdateCoordinator(DataUpdateCoordinator):
api = self._get_appliance_api(appliance) api = self._get_appliance_api(appliance)
api.build_entities_list() api.build_entities_list()
self.appliance_apis[mac_addr] = api self.appliance_apis[mac_addr] = api
else:
#if we already have the API, switch out its appliance reference for this one
api = self.appliance_apis[mac_addr]
api.appliance = appliance
async def get_client(self) -> GeWebsocketClient: async def get_client(self) -> GeWebsocketClient:
"""Get a new GE Websocket client.""" """Get a new GE Websocket client."""