from appium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time
import traceback

# 配置 Appium
desired_caps = {
    "platformName": "Windows",
    "deviceName": "WindowsPC",
    "app": r"C:\Windows\System32\notepad.exe",
    "ms:waitForAppLaunch": "25",
    "ms:waitForAppLaunch": "1",  # 减少启动等待时间
    "ms:experimental-webdriver": True,  # 添加实验性能力
    "newCommandTimeout": 3600  # 增加命令超时时间
}

try:
    print("正在连接到 WinAppDriver...")
    driver = webdriver.Remote(
        command_executor='http://127.0.0.1:4723/wd/hub',
        desired_capabilities=desired_caps
    )
    
    print("记事本已启动")
    time.sleep(0.1)
    
    print("正在查找编辑框...")
    wait = WebDriverWait(driver, 10)
    edit_box = wait.until(
        EC.presence_of_element_located((By.NAME, "文本编辑器"))
    )
    
    print("找到编辑框，正在输入文本...")
    # 输入多行文本
    edit_box.send_keys(u"Hello World!\n")
    edit_box.send_keys("这是第二行文本\n")
    edit_box.send_keys(f"当前时间: {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
    edit_box.send_keys("测试成功！")
    
    print("文本输入完成")
    time.sleep(0.1)  # 暂停3秒查看结果
    
except Exception as e:
    print(f"发生错误: {e}")
    print(f"错误类型: {type(e).__name__}")
    print("详细错误信息:")
    print(traceback.format_exc())
    
finally:
    if 'driver' in locals():
        driver.quit()
        print("记事本已关闭")