wyzeapy.services.hms_service

 1#  Copyright (c) 2021. Mulliken, LLC - All Rights Reserved
 2#  You may use, distribute and modify this code under the terms
 3#  of the attached license. You should have received a copy of
 4#  the license with this file. If not, please write to:
 5#  katie@mulliken.net to receive a copy
 6from enum import Enum
 7from typing import Optional
 8
 9from ..wyze_auth_lib import WyzeAuthLib
10from .base_service import BaseService
11
12
13class HMSMode(Enum):
14    CHANGING = 'changing'
15    DISARMED = 'disarm'
16    AWAY = 'away'
17    HOME = 'home'
18
19
20class HMSService(BaseService):
21    async def update(self, hms_id: str):
22        hms_mode = await self._monitoring_profile_state_status(hms_id)
23        return HMSMode(hms_mode['message'])
24
25    def __init__(self, auth_lib: WyzeAuthLib):
26        super().__init__(auth_lib)
27
28        self._hms_id = None
29
30    @classmethod
31    async def create(cls, auth_lib: WyzeAuthLib):
32        hms_service = cls(auth_lib)
33        hms_service._hms_id = await hms_service._get_hms_id()
34
35        return hms_service
36
37    @property
38    def hms_id(self) -> Optional[str]:
39        return self._hms_id
40
41    @property
42    async def has_hms(self):
43        if self._hms_id is None:
44            self._hms_id = self.hms_id
45
46        return self._hms_id is not None
47
48    async def set_mode(self, mode: HMSMode):
49        if mode == HMSMode.DISARMED:
50            await self._disable_reme_alarm(self.hms_id)
51            await self._monitoring_profile_active(self.hms_id, 0, 0)
52        elif mode == HMSMode.AWAY:
53            await self._monitoring_profile_active(self.hms_id, 0, 1)
54        elif mode == HMSMode.HOME:
55            await self._monitoring_profile_active(self.hms_id, 1, 0)
56
57    async def _get_hms_id(self) -> Optional[str]:
58        """
59        Processes the response from _get_plan_binding_list_by_user to get the hms_id
60
61        :return: The hms_id or nothing if there is no hms in the account
62        """
63
64        await self._auth_lib.refresh_if_should()
65
66        if self._hms_id is not None:
67            return self._hms_id
68
69        response = await self._get_plan_binding_list_by_user()
70        hms_subs = response['data']
71
72        if len(hms_subs) >= 1:
73            for sub in hms_subs:
74                if (devices := sub.get('deviceList')) is not None and len(devices) >= 1:
75                    for device in devices:
76                        self._hms_id = str(device['device_id'])
77                        return self._hms_id
78
79        return None