# 创建时间:2024/9/21 下午7:37
# 创建人:天霄
# 基于 Python 3.11
# ========================================
# 测试智能插座的拉合闸
# ========================================
import sys
from pathlib import Path
import yaml
from miio import Device


def main():
    args = sys.argv[1:]
    if not args:
        print('请输入参数！on 为合闸，off 为拉闸。\n示例：python test_power_control.py on')
        sys.exit(1)

    workPath = Path(__file__).parent.absolute()
    conf_path = workPath / 'config.yaml'
    if not conf_path.is_file():
        print('配置文件 config.yaml 不存在，请从 config-template.yaml 复制一份并重命名为 config.yaml')
        sys.exit(1)

    conf: dict = yaml.safe_load(conf_path.read_text(encoding='utf-8'))
    device_conf = conf.get('device', {})
    power_param = {
        'did': str(device_conf.get('did', '')),
        'siid': device_conf.get('siid'),
        'piid': device_conf.get('piid'),
    }

    if args[0] == 'on':
        power_param['value'] = True
        opt = '合闸'
    elif args[0] == 'off':
        power_param['value'] = False
        opt = '拉闸'
    else:
        print('参数错误！请输入 on 或 off。')
        sys.exit(1)
    device = Device(device_conf.get('ip'), device_conf.get('token'))
    device.send("set_properties", [power_param])
    print(f'已执行{opt}操作！请核对智能插座的电流通断情况。')


if __name__ == '__main__':
    main()