import pyautogui
import time
import keyboard
import sys

# 评级功能函数
def do_rating():
    # 等待 0.5 秒
    time.sleep(0.5)

    # 单击坐标 1198, 233
    pyautogui.click(1198, 233)
    time.sleep(3.5)

    # 按住左键，从 (277,194) 拖动到 (412,309)，持续 0.3 秒
    pyautogui.moveTo(277, 194)
    pyautogui.mouseDown()
    pyautogui.moveTo(412, 309, duration=0.3)
    pyautogui.mouseUp()

    # 模拟 Ctrl + C 复制
    keyboard.press_and_release('ctrl+c')
    time.sleep(0.3)

    # 鼠标向下滑动 10 下（幅度大、持续 2 秒）
    for i in range(10):
        pyautogui.scroll(-800)  # 数值越大，滑动越大
        time.sleep(2/10)

    # 点击坐标 335, 644
    pyautogui.click(335, 644)
    time.sleep(0.2)

    # 粘贴 Ctrl + V
    keyboard.press_and_release('ctrl+v')
    time.sleep(0.2)

    # 输入文本
    text = (
        "客户为淮安人，从事自由职业，常驻:  ,客户在我行开立一张社保卡账户，主要开立为了日常以及医疗使用，"
        "同时开通了微信银行功能且账户分类为关注类，查询客户账户明细无异常，故结合系统评级认定客户较低风险。"
    )

    pyautogui.write(text, interval=0.02)

    print("✓ 本次评级执行完毕\n")


# 主循环
def main():
    print("=== 评级自动化程序 ===")
    print("输入 1 回车：执行一次评级")
    print("输入 0 回车：退出程序\n")

    while True:
        cmd = input("请输入指令（1=执行，0=退出）：").strip()

        if cmd == "1":
            print("开始执行评级...\n")
            do_rating()

        elif cmd == "0":
            print("程序已退出。")
            sys.exit(0)

        else:
            print("无效输入，请重新输入。\n")


if __name__ == "__main__":
    main()