def test_get_modes_v2(self):
"""
"""
mocked_url = "https://my.arlo.com/hmsweb/users/devices/automation/active"
mock_json = {}
with open('./responses/expected_get_modes_v2_response.json') as json_file:
mock_json = json.load(json_file)
responses.add(
method = responses.GET,
url = mocked_url,
json = mock_json,
status = 200
)
arlo = Arlo(USERNAME, PASSWORD)
response = arlo.GetModesV2()
assert response[0]["uniqueId"] == mock_json["data"][0]["uniqueId"]
class ArloMqtt:
def __init__(self, args):
self.log = logging.getLogger(type(self).__name__)
if args.debug:
self.log.setLevel(logging.DEBUG)
self.report_interval = args.report_interval
self.report_thread = threading.Thread(target=self.report_thread_fn,
daemon=True)
self.stop = threading.Event()
self.last_reported_ts = time.time()
self.arlo_user = args.arlo_user
self.arlo_pass = args.arlo_pass
self.mqtt_settings = {
'MQTT_BROKER': args.broker,
'MQTT_PORT': args.port,
'MQTT_USERNAME': args.mqtt_user,
'MQTT_PASSWORD': args.mqtt_pass,
}
def report_thread_fn(self):
try:
while True:
self.report()
self.last_reported_ts = time.time()
if self.stop.wait(self.report_interval):
break
except Exception:
self.log.exception('Report thread failed')
finally:
self.log.info('Report thread stopped')
self.stop.set()
def run(self):
self.log.info('Connecting to Arlo')
self.arlo = Arlo(self.arlo_user, self.arlo_pass)
self.arlo_bases = {}
self.arlo_cams = {}
self.report_thread.start()
try:
while True:
if self.stop.wait(self.report_interval * 2):
break
if time.time(
) - self.last_reported_ts > self.report_interval * 2:
self.log.error('Report timeout')
break
finally:
self.log.info('Main thread stopped')
self.stop.set()
def report(self):
self.log.info('Reporting status')
# Fetching devices and states from Arlo
devices = [
dev for dev in self.arlo.GetDevices() if dev['state'] != 'removed'
]
bases = {
dev['deviceId']: dev
for dev in devices if dev['deviceType'] == 'basestation'
}
cams = {
dev['deviceId']: dev
for dev in devices if dev['deviceType'] == 'camera'
}
for mode in self.arlo.GetModesV2():
id = mode['gatewayId']
if id in bases:
bases[id]['modeState'] = mode
for base in bases.values():
for state in self.arlo.GetCameraState(base)['properties']:
id = state['serialNumber']
if id in cams:
cams[id]['cameraState'] = state
# Syncing devices and states to MQTT Homie
for id, base in bases.items():
if id not in self.arlo_bases:
self.arlo_bases[id] = HomieArloBaseStation(
id=id,
name=base['deviceName'],
mqtt_settings=self.mqtt_settings,
arlo=self.arlo,
base_station=base,
log=self.log.getChild('homie'),
)
self.arlo_bases[id].node.active_modes = ','.join(
base['modeState']['activeModes'])
for id in [id for id in self.arlo_bases if id not in bases]:
del self.arlo_bases[id]
for id, cam in cams.items():
if id not in self.arlo_cams:
self.arlo_cams[id] = HomieArloCamera(
id=id,
name=cam['deviceName'],
mqtt_settings=self.mqtt_settings,
arlo=self.arlo,
base_station=bases[cam['parentId']],
log=self.log.getChild('homie'),
)
self.arlo_cams[id].node.privacy_active = cam['cameraState'][
'privacyActive']
self.arlo_cams[id].node.battery_level = cam['cameraState'][
'batteryLevel']
self.arlo_cams[id].node.charging_state = cam['cameraState'][
'chargingState']
self.arlo_cams[id].node.connection_state = cam['cameraState'][
'connectionState']
self.arlo_cams[id].node.signal_strength = cam['cameraState'][
'signalStrength']
self.arlo_cams[id].node.last_image = cam['presignedLastImageUrl']
for id in [id for id in self.arlo_cams if id not in cams]:
del self.arlo_cams[id]
