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

Merge pull request #20 from USA-RedDragon/start-error-handling

Browse Source 
Add preliminary error handling support
This commit is contained in:
Jacob McSwain 2019-10-22 13:34:59 -05:00 committed by GitHub
commit 5f4b14514c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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.headers.update({"Referer": "https://home.nest.com/"})
self._device_id = device_id
if not email and not password:
self._login_google(issue_token, cookie, api_key)
self._email = email
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:
self._login_nest(email, password)
self._login_nest(self._email, self._password)
def _login_nest(self, email, password):
r = self._session.post(
@ -100,6 +108,7 @@ class NestThermostatAPI(NestAPI):
self.update()
def get_devices(self):
try:
r = self._session.post(
f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
json={
@ -115,6 +124,11 @@ class NestThermostatAPI(NestAPI):
devices.append(bucket.replace('device.', ''))
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):
if self._hvac_ac_state:
@ -125,6 +139,7 @@ class NestThermostatAPI(NestAPI):
return "off"
def update(self):
try:
r = self._session.post(
f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
json={
@ -138,11 +153,13 @@ class NestThermostatAPI(NestAPI):
temp_mode = None
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"]
self.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[
"compressor_lockout_enabled"
]
@ -150,7 +167,8 @@ class NestThermostatAPI(NestAPI):
"compressor_lockout_timeout"
]
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"]
self.target_temperature_high = thermostat_data[
"target_temperature_high"
@ -159,10 +177,12 @@ class NestThermostatAPI(NestAPI):
thermostat_data["target_temperature_low"]
self.can_heat = thermostat_data["can_heat"]
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"]
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.fan = thermostat_data["fan_timer_timeout"] > 0
self.current_humidity = thermostat_data["current_humidity"]
@ -170,6 +190,11 @@ class NestThermostatAPI(NestAPI):
thermostat_data["eco"]["mode"] == 'auto-eco':
temp_mode = 'eco'
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):
if temp_high is None: