diff --git a/src/wyzeapy/base_client.py b/src/wyzeapy/base_client.py index 1e370b9..3ee606b 100644 --- a/src/wyzeapy/base_client.py +++ b/src/wyzeapy/base_client.py @@ -96,7 +96,7 @@ class BaseClient: access_token = "" refresh_token = "" - def login(self, email, password) -> None: + def login(self, email, password) -> bool: email = email password = password @@ -111,9 +111,15 @@ def login(self, email, password) -> None: response_json = requests.post("https://auth-prod.api.wyze.com/user/login", headers=headers, json=login_payload).json() - - self.access_token = response_json['access_token'] - self.refresh_token = response_json['refresh_token'] + try: + self.access_token = response_json['access_token'] + self.refresh_token = response_json['refresh_token'] + return True + except KeyError: + return False + + def can_login(self, username, password): + return self.login(username, password) @staticmethod def create_password(password) -> str: