sascha/badnest
1
0
Fork 0
mirror of https://github.com/USA-RedDragon/badnest.git synced 2025-01-18 22:50:44 +00:00
Code Issues Releases Wiki Activity

Add preliminary error handling support

Browse Source
This commit is contained in:
Jacob McSwain 2019-10-22 13:34:00 -05:00
parent c4f702b859
commit 98daabc5d7
No known key found for this signature in database
GPG Key ID: 59957AD09B94EDF5
1 changed files with 85 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
custom_components/badnest/api.py
View File

@ -25,10 +25,18 @@ class NestAPI:
self._session = requests.Session() self._session = requests.Session()
self._session.headers.update({"Referer": "https://home.nest.com/"}) self._session.headers.update({"Referer": "https://home.nest.com/"})
self._device_id = device_id self._device_id = device_id
if not email and not password: self._email = email
self._login_google(issue_token, cookie, api_key) self._password = password
self._issue_token = issue_token
self._cookie = cookie
self._api_key = api_key
self.login()
def login(self):
if not self._email and not self._password:
self._login_google(self._issue_token, self._cookie, self._api_key)
else: else:
self._login_nest(email, password) self._login_nest(self._email, self._password)
def _login_nest(self, email, password): def _login_nest(self, email, password):
r = self._session.post( r = self._session.post(
@ -100,6 +108,7 @@ class NestThermostatAPI(NestAPI):
self.update() self.update()
def get_devices(self): def get_devices(self):
try:
r = self._session.post( r = self._session.post(
f"{API_URL}/api/0.1/user/{self._user_id}/app_launch", f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
json={ json={
@ -115,6 +124,11 @@ class NestThermostatAPI(NestAPI):
devices.append(bucket.replace('device.', '')) devices.append(bucket.replace('device.', ''))
return devices return devices
except requests.exceptions.RequestException as e:
_LOGGER.error(e)
_LOGGER.error('Failed to get devices, trying to log in again')
self.login()
return self.get_devices()
def get_action(self): def get_action(self):
if self._hvac_ac_state: if self._hvac_ac_state:
@ -125,6 +139,7 @@ class NestThermostatAPI(NestAPI):
return "off" return "off"
def update(self): def update(self):
try:
r = self._session.post( r = self._session.post(
f"{API_URL}/api/0.1/user/{self._user_id}/app_launch", f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
json={ json={
@ -138,11 +153,13 @@ class NestThermostatAPI(NestAPI):
temp_mode = None temp_mode = None
for bucket in r.json()["updated_buckets"]: for bucket in r.json()["updated_buckets"]:
if bucket["object_key"].startswith(f"shared.{self._device_id}"): if bucket["object_key"] \
.startswith(f"shared.{self._device_id}"):
thermostat_data = bucket["value"] thermostat_data = bucket["value"]
self.current_temperature = \ self.current_temperature = \
thermostat_data["current_temperature"] thermostat_data["current_temperature"]
self.target_temperature = thermostat_data["target_temperature"] self.target_temperature = \
thermostat_data["target_temperature"]
self._compressor_lockout_enabled = thermostat_data[ self._compressor_lockout_enabled = thermostat_data[
"compressor_lockout_enabled" "compressor_lockout_enabled"
] ]
@ -150,7 +167,8 @@ class NestThermostatAPI(NestAPI):
"compressor_lockout_timeout" "compressor_lockout_timeout"
] ]
self._hvac_ac_state = thermostat_data["hvac_ac_state"] self._hvac_ac_state = thermostat_data["hvac_ac_state"]
self._hvac_heater_state = thermostat_data["hvac_heater_state"] self._hvac_heater_state = \
thermostat_data["hvac_heater_state"]
temp_mode = thermostat_data["target_temperature_type"] temp_mode = thermostat_data["target_temperature_type"]
self.target_temperature_high = thermostat_data[ self.target_temperature_high = thermostat_data[
"target_temperature_high" "target_temperature_high"
@ -159,10 +177,12 @@ class NestThermostatAPI(NestAPI):
thermostat_data["target_temperature_low"] thermostat_data["target_temperature_low"]
self.can_heat = thermostat_data["can_heat"] self.can_heat = thermostat_data["can_heat"]
self.can_cool = thermostat_data["can_cool"] self.can_cool = thermostat_data["can_cool"]
elif bucket["object_key"].startswith(f"device.{self._device_id}"): elif bucket["object_key"] \
.startswith(f"device.{self._device_id}"):
thermostat_data = bucket["value"] thermostat_data = bucket["value"]
self._time_to_target = thermostat_data["time_to_target"] self._time_to_target = thermostat_data["time_to_target"]
self._fan_timer_timeout = thermostat_data["fan_timer_timeout"] self._fan_timer_timeout = \
thermostat_data["fan_timer_timeout"]
self.has_fan = thermostat_data["has_fan"] self.has_fan = thermostat_data["has_fan"]
self.fan = thermostat_data["fan_timer_timeout"] > 0 self.fan = thermostat_data["fan_timer_timeout"] > 0
self.current_humidity = thermostat_data["current_humidity"] self.current_humidity = thermostat_data["current_humidity"]
@ -170,6 +190,11 @@ class NestThermostatAPI(NestAPI):
thermostat_data["eco"]["mode"] == 'auto-eco': thermostat_data["eco"]["mode"] == 'auto-eco':
temp_mode = 'eco' temp_mode = 'eco'
self.mode = temp_mode self.mode = temp_mode
except requests.exceptions.RequestException as e:
_LOGGER.error(e)
_LOGGER.error('Failed to update, trying to log in again')
self.login()
self.update()
def set_temp(self, temp, temp_high=None): def set_temp(self, temp, temp_high=None):
if temp_high is None: if temp_high is None: