Final rename

custom_components/badnest/api.py

@@ -0,0 +1,160 @@
+from datetime import time, timedelta, datetime
+import json
+
+import requests
+
+API_URL = 'https://home.nest.com'
+
+class NestAPI():
+    def __init__(self, email, password):
+        self._user_id = None
+        self._access_token = None
+        self._device_id = None
+        self._shared_id = None
+        self._czfe_url = None
+        self._compressor_lockout_enabled = None
+        self._compressor_lockout_time = None
+        self._hvac_ac_state = None
+        self._hvac_heater_state = None
+        self.mode = None
+        self._time_to_target = None
+        self._fan_timer_timeout = None
+        self.can_heat = None
+        self.can_cool = None
+        self.has_fan = None
+        self.fan = None
+        self.away = None
+        self.current_temperature = None
+        self.target_temperature = None
+        self.target_temperature_high = None
+        self.target_temperature_low = None
+        self.current_humidity = None
+
+        self._login(email, password)
+        self.update()
+
+    def _login(self, email = 'jacob.a.mcswain@gmail.com', password = 'ttlshiwwyaJ@'):
+        r = requests.post(f'{API_URL}/session', json={
+            'email': email,
+            'password': password
+        })
+        self._user_id = r.json()['userid']
+        self._access_token = r.json()['access_token']
+
+    def get_action(self):
+        if self._hvac_ac_state:
+            return 'cooling'
+        elif self._hvac_heater_state:
+            return 'heating'
+        else:
+            return 'off'
+
+    def update(self):
+        r = requests.post(f'{API_URL}/api/0.1/user/{self._user_id}/app_launch', json={
+            'known_bucket_types': ['shared', 'device'],
+            'known_bucket_versions': []
+        },
+        headers={
+            'Authorization': f'Basic {self._access_token}'
+        })
+
+        self._czfe_url = r.json()['service_urls']['urls']['czfe_url']
+
+        for bucket in r.json()['updated_buckets']:
+            if bucket['object_key'].startswith('shared.'):
+                self._shared_id = bucket['object_key']
+                thermostat_data = bucket['value']
+                self.current_temperature = thermostat_data['current_temperature']
+                self.target_temperature = thermostat_data['target_temperature']
+                self._compressor_lockout_enabled = thermostat_data['compressor_lockout_enabled']
+                self._compressor_lockout_time = thermostat_data['compressor_lockout_timeout']
+                self._hvac_ac_state = thermostat_data['hvac_ac_state']
+                self._hvac_heater_state = thermostat_data['hvac_heater_state']
+                self.mode = thermostat_data['target_temperature_type']
+                self.target_temperature_high = thermostat_data['target_temperature_high']
+                self.target_temperature_low = thermostat_data['target_temperature_low']
+                self.can_heat = thermostat_data['can_heat']
+                self.can_cool = thermostat_data['can_cool']
+            elif bucket['object_key'].startswith('device.'):
+                self._device_id = bucket['object_key']
+                thermostat_data = bucket['value']
+                self._time_to_target = thermostat_data['time_to_target']
+                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']
+                self.away = thermostat_data['home_away_input']
+
+    def set_temp(self, temp, temp_high = None):
+        if temp_high is None:
+            requests.post(f'{self._czfe_url}/v5/put', json={
+                'objects': [{
+                    'object_key': self._shared_id,
+                    'op': 'MERGE',
+                    'value':{
+                        'target_temperature': temp
+                    }
+                }]
+            },
+            headers={
+                'Authorization': f'Basic {self._access_token}'
+            })
+        else:
+            requests.post(f'{self._czfe_url}/v5/put', json={
+                'objects': [{
+                    'object_key': self._shared_id,
+                    'op': 'MERGE',
+                    'value': {
+                        'target_temperature_low': temp,
+                        'target_temperature_high': temp_high
+                    }
+                }]
+            },
+            headers={
+                'Authorization': f'Basic {self._access_token}'
+            })
+            
+
+    def set_mode(self, mode):
+        requests.post(f'{self._czfe_url}/v5/put', json={
+            'objects': [{
+                'object_key': self._shared_id,
+                'op': 'MERGE',
+                'value':{
+                    'target_temperature_type': mode
+                }
+            }]
+        },
+        headers={
+            'Authorization': f'Basic {self._access_token}'
+        })
+
+    def set_fan(self, date):
+        requests.post(f'{self._czfe_url}/v5/put', json={
+            'objects': [{
+                'object_key': self._device_id,
+                'op': 'MERGE',
+                'value':{
+                    'fan_timer_timeout': date
+                }
+            }]
+        },
+        headers={
+            'Authorization': f'Basic {self._access_token}'
+        })
+
+    def set_eco_mode(self):
+        requests.post(f'{self._czfe_url}/v5/put', json={
+            'objects': [{
+                'object_key': self._device_id,
+                'op': 'MERGE',
+                'value':{
+                    'eco': {
+                        'mode': 'manual-eco'
+                    }
+                }
+            }]
+        },
+        headers={
+            'Authorization': f'Basic {self._access_token}'
+        })