#UI自动化控制技术 - 极简实战指南

import uiautomator2 as u2
import time

# 1. 连接默认 USB 设备（WiFi 连接可用 u2.connect("192.168.1.100")）
d = u2.connect()
# 2. 启动指定 App（包名可通过 adb shell dumpsys window | grep mCurrentFocus 获取）
d.app_start("com.taobao.taobao")
time.sleep(3)  # 硬等待仅用于演示，实际建议用 exists() 做条件等待

# 3. 通过 resourceId 精准定位搜索框（resourceId 最稳定、最推荐）
search_box = d(resourceId="com.taobao.taobao:id/searchEdit")
# 4. 点击并输入关键词
search_box.click()
search_box.set_text("学生党备用机")
# 5. 上滑页面并点击「搜索」按钮
d.swipe_ext("up", scale=0.8)
d(text="搜索").click()

from airtest.core.api import *
from poco.drivers.android.uiautomation import AndroidUiautomationPoco

# 1. 连接默认 USB 设备
connect_device("Android:///")
# 2. 初始化 Poco 解析器
poco = AndroidUiautomationPoco(use_airtest_input=True)

# 3. 图像识别定位（需提前截图保存为 like.png，建议只截纯图标，不要带动态文字）
# Template(r"like.png", record_pos=(0.7, 0.5), resolution=(1080, 2400)).click()

# 4. Poco UI 树定位（当元素有稳定属性时优先使用）
poco(text="点赞").click()

# 5. 执行滑动操作
swipe((500, 1800), (500, 600), duration=0.8)

import uiautomator2 as u2
import time
import random

# ----------------------- 配置区（可根据实际需求修改） -----------------------
DOUYIN_PKG = "com.ss.android.ugc.aweme"
LIKE_PROB = 0.4                # 40% 的概率给视频点赞
COMMENT_PROB = 0.1             # 10% 的概率发送评论
MAX_VIDEOS = 15                # 最多刷 15 个视频后停止
COMMENT_LIST = [
    "很棒的内容～",
    "学到了，收藏收藏！",
    "支持一下博主！",
    "这个好有意思哈哈哈哈",
    "666666"
]

# ----------------------- 连接设备 -----------------------
print("🔗 正在连接安卓设备...")
try:
    d = u2.connect()           # WiFi 调试可写 u2.connect("IP地址")
    print(f"✅ 连接成功！设备序列号：{d.serial}")
except Exception as e:
    print(f"❌ 连接失败，请检查 USB 调试或网络连接：{e}")
    exit()

# ----------------------- 安全操作工具函数 -----------------------
def safe_click(element, timeout=2):
    """带超时和随机偏移的安全点击"""
    if not element.exists(timeout=timeout):
        return False
    bounds = element.bounds
    # 避开元素边缘，防止点到无效区域
    x = random.randint(bounds['left'] + 6, bounds['right'] - 6)
    y = random.randint(bounds['top'] + 6, bounds['bottom'] - 6)
    d.click(x, y)
    return True

def random_delay(base, variance):
    """生成随机延迟，base 为基准秒数，variance 为上下浮动范围"""
    actual_delay = max(0.5, base + random.uniform(-variance, variance))
    time.sleep(actual_delay)

# ----------------------- 主运行逻辑 -----------------------
total_videos = 0
total_likes = 0
total_comments = 0

print(f"\n🚀 脚本启动！目标：最多处理 {MAX_VIDEOS} 个视频")
try:
    while total_videos < MAX_VIDEOS:
        total_videos += 1
        print(f"\n🎬 正在处理第 {total_videos}/{MAX_VIDEOS} 个视频...")
        # 模拟观看时间
        random_delay(2.5, 1)

        # 1. 随机点赞
        if random.random() < LIKE_PROB:
            # 多重降级定位，提高成功率
            like_element = (
                d(resourceId="com.ss.android.ugc.aweme:id/aweme_like_layout")
                or d(desc="赞")
                or d(resourceId="com.ss.android.ugc.aweme:id/dv9")  # 备用 ID（不同版本可能不同）
            )
            if safe_click(like_element):
                total_likes += 1
                print(f"👍 点赞成功！累计点赞：{total_likes}")
                random_delay(0.8, 0.3)

        # 2. 随机评论
        if random.random() < COMMENT_PROB:
            comment_btn = (
                d(resourceId="com.ss.android.ugc.aweme:id/aweme_comment_layout")
                or d(desc="评论")
                or d(resourceId="com.ss.android.ugc.aweme:id/dvb")
            )
            if safe_click(comment_btn):
                random_delay(1.5, 0.5)
                # 定位输入框
                input_box = (
                    d(resourceId="com.ss.android.ugc.aweme:id/chat_input_view")
                    or d(className="android.widget.EditText")
                )
                if input_box.exists(timeout=2):
                    random_comment = random.choice(COMMENT_LIST)
                    input_box.set_text(random_comment)
                    random_delay(0.9, 0.2)
                    # 点击发送按钮
                    send_btn = d(text="发送") or d(desc="发送")
                    if safe_click(send_btn):
                        total_comments += 1
                        print(f"💬 评论成功！内容：{random_comment} | 累计评论：{total_comments}")
                        random_delay(1, 0.3)
                # 返回视频播放页
                d.press("back")
                random_delay(0.8, 0.3)

        # 3. 上滑到下一个视频
        print("⬆️  滑动到下一个视频...")
        d.swipe_ext(
            "up",
            scale=0.7 + random.random() * 0.1,
            duration=0.8 + random.random() * 0.2
        )
        # 等待下一个视频加载完成
        random_delay(1.2, 0.5)

except KeyboardInterrupt:
    print("\n⏸️ 用户手动停止了脚本")
except Exception as e:
    print(f"\n❌ 脚本运行出错：{e}")
finally:
    print("\n📊 本次运行统计：")
    print(f"  处理视频数：{total_videos}")
    print(f"  成功点赞数：{total_likes}")
    print(f"  成功评论数：{total_comments}")