Merge pull request #24 from USA-RedDragon/cameras-multiple-api-instances

Attempt to make the camera's use different API instances

custom_components/badnest/api.py

@@ -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._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:
-            self._login_nest(email, password)
+            self._login_nest(self._email, self._password)
 
     def _login_nest(self, email, password):
         r = self._session.post(
@@ -100,21 +108,30 @@ class NestThermostatAPI(NestAPI):
         self.update()
 
     def get_devices(self):
-        r = self._session.post(
+        try:
-            f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
+            r = self._session.post(
-            json={
+                f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
-                "known_bucket_types": ["buckets"],
+                json={
-                "known_bucket_versions": [],
+                    "known_bucket_types": ["buckets"],
-            },
+                    "known_bucket_versions": [],
-            headers={"Authorization": f"Basic {self._access_token}"},
+                },
-        )
+                headers={"Authorization": f"Basic {self._access_token}"},
-        devices = []
+            )
-        buckets = r.json()['updated_buckets'][0]['value']['buckets']
+            devices = []
-        for bucket in buckets:
+            buckets = r.json()['updated_buckets'][0]['value']['buckets']
-            if bucket.startswith('device.'):
+            for bucket in buckets:
-                devices.append(bucket.replace('device.', ''))
+                if bucket.startswith('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 again')
+            return self.get_devices()
+        except KeyError:
+            _LOGGER.debug('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,51 +142,66 @@ class NestThermostatAPI(NestAPI):
             return "off"
 
     def update(self):
-        r = self._session.post(
+        try:
-            f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
+            r = self._session.post(
-            json={
+                f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
-                "known_bucket_types": ["shared", "device"],
+                json={
-                "known_bucket_versions": [],
+                    "known_bucket_types": ["shared", "device"],
-            },
+                    "known_bucket_versions": [],
-            headers={"Authorization": f"Basic {self._access_token}"},
+                },
-        )
+                headers={"Authorization": f"Basic {self._access_token}"},
+            )
 
-        self._czfe_url = r.json()["service_urls"]["urls"]["czfe_url"]
+            self._czfe_url = r.json()["service_urls"]["urls"]["czfe_url"]
 
-        for bucket in r.json()["updated_buckets"]:
+            temp_mode = None
-            if bucket["object_key"].startswith(f"shared.{self._device_id}"):
+            for bucket in r.json()["updated_buckets"]:
-                thermostat_data = bucket["value"]
+                if bucket["object_key"] \
-                self.current_temperature = \
+                        .startswith(f"shared.{self._device_id}"):
-                    thermostat_data["current_temperature"]
+                    thermostat_data = bucket["value"]
-                self.target_temperature = thermostat_data["target_temperature"]
+                    self.current_temperature = \
-                self._compressor_lockout_enabled = thermostat_data[
+                        thermostat_data["current_temperature"]
-                    "compressor_lockout_enabled"
+                    self.target_temperature = \
-                ]
+                        thermostat_data["target_temperature"]
-                self._compressor_lockout_time = thermostat_data[
+                    self._compressor_lockout_enabled = thermostat_data[
-                    "compressor_lockout_timeout"
+                        "compressor_lockout_enabled"
-                ]
+                    ]
-                self._hvac_ac_state = thermostat_data["hvac_ac_state"]
+                    self._compressor_lockout_time = thermostat_data[
-                self._hvac_heater_state = thermostat_data["hvac_heater_state"]
+                        "compressor_lockout_timeout"
-                if self.mode != 'eco':
+                    ]
-                    self.mode = thermostat_data["target_temperature_type"]
+                    self._hvac_ac_state = thermostat_data["hvac_ac_state"]
-                self.target_temperature_high = thermostat_data[
+                    self._hvac_heater_state = \
-                    "target_temperature_high"
+                        thermostat_data["hvac_heater_state"]
-                ]
+                    if temp_mode is None:
-                self.target_temperature_low = \
+                        temp_mode = thermostat_data["target_temperature_type"]
-                    thermostat_data["target_temperature_low"]
+                    self.target_temperature_high = thermostat_data[
-                self.can_heat = thermostat_data["can_heat"]
+                        "target_temperature_high"
-                self.can_cool = thermostat_data["can_cool"]
+                    ]
-            elif bucket["object_key"].startswith(f"device.{self._device_id}"):
+                    self.target_temperature_low = \
-                thermostat_data = bucket["value"]
+                        thermostat_data["target_temperature_low"]
-                self._time_to_target = thermostat_data["time_to_target"]
+                    self.can_heat = thermostat_data["can_heat"]
-                self._fan_timer_timeout = thermostat_data["fan_timer_timeout"]
+                    self.can_cool = thermostat_data["can_cool"]
-                self.has_fan = thermostat_data["has_fan"]
+                elif bucket["object_key"] \
-                self.fan = thermostat_data["fan_timer_timeout"] > 0
+                        .startswith(f"device.{self._device_id}"):
-                self.current_humidity = thermostat_data["current_humidity"]
+                    thermostat_data = bucket["value"]
-                if thermostat_data["eco"]["mode"] == 'manual-eco':
+                    self._time_to_target = thermostat_data["time_to_target"]
-                    self.mode = 'eco'
+                    self._fan_timer_timeout = \
-                else:
+                        thermostat_data["fan_timer_timeout"]
-                    self.mode = 'unknown'
+                    self.has_fan = thermostat_data["has_fan"]
+                    self.fan = thermostat_data["fan_timer_timeout"] > 0
+                    self.current_humidity = thermostat_data["current_humidity"]
+                    if thermostat_data["eco"]["mode"] == 'manual-eco' or \
+                            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 again')
+            self.update()
+        except KeyError:
+            _LOGGER.debug('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:
@@ -271,53 +303,79 @@ class NestTemperatureSensorAPI(NestAPI):
         self.update()
 
     def get_devices(self):
-        r = self._session.post(
+        try:
-            f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
+            r = self._session.post(
-            json={
+                f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
-                "known_bucket_types": ["buckets"],
+                json={
-                "known_bucket_versions": [],
+                    "known_bucket_types": ["buckets"],
-            },
+                    "known_bucket_versions": [],
-            headers={"Authorization": f"Basic {self._access_token}"},
+                },
-        )
+                headers={"Authorization": f"Basic {self._access_token}"},
-        devices = []
+            )
-        buckets = r.json()['updated_buckets'][0]['value']['buckets']
+            devices = []
-        for bucket in buckets:
+            buckets = r.json()['updated_buckets'][0]['value']['buckets']
-            if bucket.startswith('kryptonite.'):
+            for bucket in buckets:
-                devices.append(bucket.replace('kryptonite.', ''))
+                if bucket.startswith('kryptonite.'):
+                    devices.append(bucket.replace('kryptonite.', ''))
 
-        return devices
+            return devices
+        except requests.exceptions.RequestException as e:
+            _LOGGER.error(e)
+            _LOGGER.error('Failed to get devices, trying again')
+            return self.get_devices()
+        except KeyError:
+            _LOGGER.debug('Failed to get devices, trying to log in again')
+            self.login()
+            return self.get_devices()
 
     def update(self):
-        r = self._session.post(
+        try:
-            f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
+            r = self._session.post(
-            json={
+                f"{API_URL}/api/0.1/user/{self._user_id}/app_launch",
-                "known_bucket_types": ["kryptonite"],
+                json={
-                "known_bucket_versions": [],
+                    "known_bucket_types": ["kryptonite"],
-            },
+                    "known_bucket_versions": [],
-            headers={"Authorization": f"Basic {self._access_token}"},
+                },
-        )
+                headers={"Authorization": f"Basic {self._access_token}"},
+            )
 
-        for bucket in r.json()["updated_buckets"]:
+            for bucket in r.json()["updated_buckets"]:
-            if bucket["object_key"].startswith(
+                if bucket["object_key"].startswith(
-                    f"kryptonite.{self._device_id}"):
+                        f"kryptonite.{self._device_id}"):
-                sensor_data = bucket["value"]
+                    sensor_data = bucket["value"]
-                self.temperature = sensor_data["current_temperature"]
+                    self.temperature = sensor_data["current_temperature"]
-                self.battery_level = sensor_data["battery_level"]
+                    self.battery_level = sensor_data["battery_level"]
+        except requests.exceptions.RequestException as e:
+            _LOGGER.error(e)
+            _LOGGER.error('Failed to update, trying again')
+            self.update()
+        except KeyError:
+            _LOGGER.debug('Failed to update, trying to log in again')
+            self.login()
+            self.update()
 
 
 class NestCameraAPI(NestAPI):
-    def __init__(self, email, password, issue_token, cookie, api_key):
+    def __init__(self,
+                 email,
+                 password,
+                 issue_token,
+                 cookie,
+                 api_key,
+                 device_id=None):
         super(NestCameraAPI, self).__init__(
             email,
             password,
             issue_token,
             cookie,
-            api_key)
+            api_key,
+            device_id)
         # log into dropcam
         self._session.post(
             f"{API_URL}/dropcam/api/login",
             data={"access_token": self._access_token}
         )
+        self._device_id = device_id
         self.location = None
         self.name = "Nest Camera"
         self.online = None
@@ -327,10 +385,6 @@ class NestCameraAPI(NestAPI):
         self.data_tier = None
         self.update()
 
-    def set_device(self, uuid):
-        self._device_id = uuid
-        self.update()
-
     def update(self):
         if self._device_id:
             props = self.get_properties()

custom_components/badnest/camera.py

@@ -13,7 +13,6 @@ from .const import DOMAIN, CONF_ISSUE_TOKEN, CONF_COOKIE, CONF_APIKEY
 _LOGGER = logging.getLogger(__name__)
 
 DEFAULT_NAME = "Nest Camera"
-DATA_KEY = "camera.badnest"
 
 
 async def async_setup_platform(hass,
@@ -21,8 +20,6 @@ async def async_setup_platform(hass,
                                async_add_entities,
                                discovery_info=None):
     """Set up a Nest Camera."""
-    hass.data[DATA_KEY] = dict()
-
     api = NestCameraAPI(
         hass.data[DOMAIN][CONF_EMAIL],
         hass.data[DOMAIN][CONF_PASSWORD],
@@ -36,9 +33,15 @@ async def async_setup_platform(hass,
     _LOGGER.info("Adding cameras")
     for camera in api.get_cameras():
         _LOGGER.info("Adding nest cam uuid: %s", camera["uuid"])
-        device = NestCamera(camera["uuid"], api)
+        device = NestCamera(camera["uuid"], NestCameraAPI(
+            hass.data[DOMAIN][CONF_EMAIL],
+            hass.data[DOMAIN][CONF_PASSWORD],
+            hass.data[DOMAIN][CONF_ISSUE_TOKEN],
+            hass.data[DOMAIN][CONF_COOKIE],
+            hass.data[DOMAIN][CONF_APIKEY],
+            camera["uuid"]
+        ))
         cameras.append(device)
-        hass.data[DATA_KEY][camera["uuid"]] = device
 
     async_add_entities(cameras)
 
@@ -50,7 +53,6 @@ class NestCamera(Camera):
         """Initialize a Nest camera."""
         super().__init__()
         self._uuid = uuid
-        api.set_device(self._uuid)
         self._device = api
         self._time_between_snapshots = timedelta(seconds=30)
         self._last_image = None

custom_components/badnest/climate.py

@@ -192,9 +192,7 @@ class NestClimate(ClimateDevice):
     @property
     def hvac_mode(self):
         """Return hvac target hvac state."""
-        if self.device.mode == 'unknown' \
+        if self.device.mode is None or self.device.mode == NEST_MODE_ECO:
-            or self.device.mode == NEST_MODE_ECO \
-                or self.device.mode is None:
             # We assume the first operation in operation list is the main one
             return self._operation_list[0]