Implement authorization with Smart Life / Tuya app via QR code

as shown in https://github.com/tuya/tuya-smart-life

example.env

@@ -10,7 +10,7 @@ no_heat_tag = "!cold!"
 preheat_minutes = 60
 cooloff_minutes = 30
 
-action = "webhooks.py" # or "tuya.py"
+action = "webhooks.py" # or "tuya.py" or "smartlife.py"
 
 webhooks_url="https://maker.ifttt.com/trigger/{action}/with/key/{key}"
 webhooks_key="webhooks_key"
@@ -30,3 +30,12 @@ tuya_schema          = 'tuyaSmart'
 tuya_home            = '<from tuya.py homes>'
 tuya_scene_off       = '<from tuya.py scenes>'
 tuya_scene_on        = '<from tuya.py scenes>'
+
+smartlife_user_code  = '<from SmartLife/Tuya app under Settings/Account/User-Code>'
+smartlife_username   = '<automatically retrieved after QR authorization>'
+smartlife_token_info = '<automatically retrieved after QR authorization>'
+smartlife_terminal_id= '<automatically retrieved after QR authorization>'
+smartlife_endpoint   = '<automatically retrieved after QR authorization>'
+smartlife_home       = '<from smartlife.py scenes>'
+smartlife_scene_off  = '<from smartlife.py scenes>'
+smartlife_scene_on   = '<from smartlife.py scenes>'

requirements.txt

@@ -2,3 +2,6 @@ caldav >= 0.11.0
 pytest-mock
 python-dotenv
 tuya-iot-py-sdk
+tuya-device-sharing-sdk
+pyqrcode
+cryptography

smartlife.py

@@ -0,0 +1,202 @@
+#!/usr/bin/env python3
+# coding: utf-8
+
+import logging
+import json
+import operator
+import os
+import dotenv
+import sys
+from typing import Any
+from pathlib import Path
+import requests
+import pyqrcode
+from tuya_sharing import LoginControl, Manager, SharingTokenListener, logger
+
+EXIT_OK                          = 0
+EXIT_SYNTAX_ERROR                = 1
+EXIT_SMARTLIFE_USER_CODE_MISSING = 2
+EXIT_AUTHENTICATION_FAILED       = 3
+EXIT_SMARTLIFE_HOME_MISSING      = 4
+EXIT_SMARTLIFE_SCENE_MISSING     = 5
+EXIT_TRIGGER_SCENE_FAILED        = 6
+
+URL_PATH           = "apigw.iotbing.com"
+CONF_CLIENT_ID     = "HA_3y9q4ak7g4ephrvke"
+CONF_SCHEMA        = "haauthorize"
+APP_QR_CODE_HEADER = "tuyaSmart--qrLogin/?token="
+
+LOGGER = logging.getLogger(__package__)
+# LOGGER.setLevel(logging.DEBUG)
+
+class TokenListener(SharingTokenListener):
+    def __init__(self, dotenv_file) -> None:
+        super().__init__()
+        self.dotenv_file = dotenv_file
+
+    def update_token(self, new_token_info: [str, Any]):
+        LOGGER.debug("update token info : %s", new_token_info)
+        global token_info
+        token_info = new_token_info
+        dotenv.set_key(self.dotenv_file, 'smartlife_token_info', json.dumps(token_info))
+
+def main() -> int:
+
+    logger.setLevel(LOGGER.getEffectiveLevel())
+
+    dotenv_file = dotenv.find_dotenv()
+
+    if not dotenv_file:
+        source_path = Path(__file__).resolve()
+        dotenv_file = os.path.join(os.getcwd(), '.env')
+
+    dotenv.load_dotenv(dotenv_file)
+
+    try:
+        cmd = sys.argv[1]
+    except IndexError:
+        cmd = ''
+
+    cmds = [ 'login', 'logout', 'homes', 'scenes', 'on', 'off' ]
+    if not cmd in cmds:
+        print('Syntax: smartlife.py (%s)' % '|'.join(cmds), file=sys.stderr)
+        return EXIT_SYNTAX_ERROR;
+
+    session = requests.session()
+
+    user_code = os.environ.get('smartlife_user_code')
+    username = os.environ.get('smartlife_username')
+    terminal_id = os.environ.get('smartlife_terminal_id')
+    endpoint = os.environ.get('smartlife_endpoint')
+
+    global token_info
+    try:
+        token_info = json.loads(os.environ.get('smartlife_token_info'))
+    except TypeError:
+        token_info = None
+
+    if cmd == 'logout':
+        dotenv.unset_key(dotenv_file, 'smartlife_token_info')
+        dotenv.unset_key(dotenv_file, 'smartlife_username')
+        dotenv.unset_key(dotenv_file, 'smartlife_terminal_id')
+        dotenv.unset_key(dotenv_file, 'smartlife_endpoint')
+        return EXIT_OK
+
+    if cmd == 'login':
+        if token_info:
+            print('You are already logged in.')
+            return EXIT_OK
+
+        if not user_code:
+            user_code = input('SmartLife user-code from Settings/Account: ')
+            dotenv.set_key(dotenv_file, 'smartlife_user_code', user_code)
+
+        login_control = LoginControl()
+
+        response = login_control.qr_code(CONF_CLIENT_ID, CONF_SCHEMA, user_code)
+
+        if not response.get("success", False):
+            print('Could not login: %i %s' % (response.get('code'), response.get('msg')), file = sys.stderr)
+            return EXIT_AUTHENTICATION_FAILED
+
+        qr_code = response["result"]["qrcode"]
+
+        while True:
+            print ('Please scan this in Smart-Life and authorize access:')
+            print (pyqrcode.create(APP_QR_CODE_HEADER + qr_code, mode = 'binary').terminal())
+            input ('Then, hit ENTER to continue.')
+
+            ret, info = login_control.login_result(qr_code, CONF_CLIENT_ID, user_code)
+            if not ret:
+                print('Not authorized (yet): %i %s' % (info.get('code'), info.get('msg')), file = sys.stderr)
+                print('Try again')
+                continue
+
+            token_info = {
+                "t": info.get("t"),
+                "uid": info.get("uid"),
+                "expire_time": info.get("expire_time"),
+                "access_token": info.get("access_token"),
+                "refresh_token": info.get("refresh_token"),
+            }
+
+            username = info.get('username')
+            terminal_id = info.get('terminal_id')
+            endpoint = info.get('endpoint')
+            dotenv.set_key(dotenv_file, 'smartlife_username', username)
+            dotenv.set_key(dotenv_file, 'smartlife_token_info', json.dumps(token_info))
+            dotenv.set_key(dotenv_file, 'smartlife_terminal_id', terminal_id)
+            dotenv.set_key(dotenv_file, 'smartlife_endpoint', endpoint)
+            break
+
+        print('You are logged in.')
+        return EXIT_OK
+
+    if not token_info:
+        print('Please log in first!', file = sys.stderr);
+        return EXIT_AUTHENTICATION_FAILED
+
+    token_listener = TokenListener(dotenv_file)
+    smartlife_manager = Manager(
+        CONF_CLIENT_ID,
+        user_code,
+        terminal_id,
+        endpoint,
+        token_info,
+        token_listener
+    )
+
+    try:
+        smartlife_manager.update_device_cache()
+    except Exception as e:
+        print('Cannot access Smartlife (%s)' % (e.args), file=sys.stderr)
+        return EXIT_AUTHENTICATION_FAILED
+
+    smartlife_manager.user_homes.sort(key = operator.attrgetter('name'))
+
+    if cmd == 'homes':
+        print('Homes:')
+        for home in smartlife_manager.user_homes:
+            print('%10s: %s' % (home.id, home.name))
+        return EXIT_OK
+
+    home_id = os.environ.get('smartlife_home')
+
+    if cmd == 'scenes':
+        for home in smartlife_manager.user_homes:
+            print('Scenes in home %s (%s):' % (home.name, home.id));
+            for scene in sorted(smartlife_manager.scene_repository.query_scenes([home.id]),
+                   key = operator.attrgetter('name')):
+                print('    %s: %s' % (scene.scene_id, scene.name));
+
+        return EXIT_OK
+
+    if not home_id:
+        print('Set smartlife_home in .env first, in order to trigger scenes!', file=sys.stderr)
+        return EXIT_SMARTLIFE_HOME_MISSING
+
+    if cmd == 'on':
+        scene_id = os.environ.get('smartlife_scene_on')
+        if not scene_id:
+            print('Set smartlife_scene_on in .env first!', file=sys.stderr)
+            return EXIT_SMARTLIFE_SCENE_MISSING
+    elif cmd == 'off':
+        scene_id = os.environ.get('smartlife_scene_off')
+        if not scene_id:
+            print('Set smartlife_scene_off in .env first!', file=sys.stderr)
+            return EXIT_SMARTLIFE_SCENE_MISSING
+    try:
+        response = smartlife_manager.scene_repository.trigger_scene(home_id, scene_id)
+    except Exception as e:
+        print('Error triggering scene (%s)' % (e.args), file=sys.stderr)
+        return EXIT_TRIGGER_SCENE_FAILED
+
+    if not response:
+        print('Triggering scene reported failure', file=sys.stderr)
+        return EXIT_TRIGGER_SCENE_FAILED
+    print('Smartlife trigger succeeded.')
+    return EXIT_OK
+
+if __name__ == '__main__':
+    sys.exit(main())
+