#Airtest 框架详解：游戏/Android App 图像+UI树双引擎自动化

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

# 连接设备（USB 连接串号可通过 adb devices 查看）
connect_device("Android:///")
poco = AndroidUiautomationPoco()
print(poco.device)

from airtest.core.api import *

# 连接设备
connect_device("Android:///")

# 点击（模板图需提前截图放在项目目录）
touch(Template("like_btn.png"))

# 滑动
swipe((500, 1500), (500, 500))   # 从下往上滑

# 文本输入
text("很棒的视频")

# 按键事件
keyevent("HOME")

# 等待元素出现
wait(Template("comment_input.png"), timeout=10)

# 断言是否存在
assert_exists(Template("success_tip.png"), "点赞成功")

from poco.drivers.android.uiautomation import AndroidUiautomationPoco

poco = AndroidUiautomationPoco(use_airtest_input=True, screenshot_each_action=False)

# 通过文本查找
poco(text="首页").click()

# 通过 ID 或层级关系
poco("android.widget.ListView").child("android.widget.TextView")[0].click()

# 滑动列表
poco("com.ss.android.ugc.aweme:id/recycler_view").swipe([0, -0.3])

# 获取属性
name = poco(text="视频标题").attr("text")

# 先用图像识别点击“评论”按钮
touch(Template("comment_icon.png"))

# 再通过 Poco 定位评论输入框输入文字
poco("android.widget.EditText").set_text("加油！")

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

class BaseOperator:
    def __init__(self, device_uri="Android:///"):
        self.device = connect_device(device_uri)
        self.poco = AndroidUiautomationPoco(
            use_airtest_input=True,
            screenshot_each_action=False
        )
        self.package = "com.ss.android.ugc.aweme"
    
    def start_app(self):
        """启动抖音"""
        stop_app(self.package)
        time.sleep(1)
        start_app(self.package)
        sleep(5)  # 等待首页加载
    
    def swipe_video(self, direction="up", duration=0.3):
        """滑动视频，up 向上滑看下一个，down 向下滑看上一个"""
        if direction == "up":
            swipe((540, 1600), (540, 400), duration=duration)
        else:
            swipe((540, 400), (540, 1600), duration=duration)
        sleep(2)  # 停留观看时间
    
    def like_current_video(self):
        """点赞当前视频（双引擎保底）"""
        try:
            # 优先用 Poco 查找点赞按钮
            like_btn = self.poco(name="com.ss.android.ugc.aweme:id/ae8")
            if like_btn.exists():
                like_btn.click()
                return
        except Exception:
            pass
        # 降级为图像识别
        touch(Template("images/like_btn.png", threshold=0.7))
    
    def comment_current_video(self, text_content):
        """评论当前视频"""
        # 点击评论图标
        try:
            self.poco(desc="评论").click()
        except Exception:
            touch(Template("images/comment_btn.png"))
        
        sleep(1)
        # 输入评论内容
        self.poco("android.widget.EditText").set_text(text_content)
        sleep(0.5)
        # 发送按钮（不同版本可能是“发送”文本或发送图标）
        send_btn = self.poco(text="发送")
        if send_btn.exists():
            send_btn.click()
        else:
            self.poco(type="android.widget.ImageView", desc="发送").click()
        sleep(1)
        # 关闭评论面板
        self.close_comment_panel()
    
    def close_comment_panel(self):
        """关闭评论面板，回退到短视频界面"""
        keyevent("BACK")
        sleep(0.5)
        # 如果 BACK 键无效，用图像点击关闭按钮
        if exists(Template("images/close_comment.png")):
            touch(Template("images/close_comment.png"))
    
    def random_comment(self):
        """生成随机正面评论"""
        comments = ["太厉害了👍", "学到了", "支持创作者", "已点赞", "这内容绝了",
                    "幽默风趣哈哈", "每天必看", "这个系列能多出吗"]
        return random.choice(comments)
    
    def run_interaction_flow(self, swipe_count=3):
        """完整交互流程"""
        self.start_app()
        # 前 swipe_count 个视频只滑动
        for i in range(swipe_count):
            self.swipe_video("up")
        # 第 swipe_count+1 个视频执行点赞+评论
        sleep(1)
        self.like_current_video()
        sleep(0.5)
        self.comment_current_video(self.random_comment())
        # 继续滑动几个视频
        for i in range(2):
            self.swipe_video("up")

from base_operator import BaseOperator

if __name__ == "__main__":
    operator = BaseOperator()
    try:
        operator.run_interaction_flow(swipe_count=3)
        print("自动化流程完成")
    except Exception as e:
        print(f"脚本运行出错: {e}")

# 滑动距离加入随机偏移
import random
x1, y1 = 540, 1600
x2, y2 = 540 + random.randint(-50, 50), 400 + random.randint(-30, 30)
swipe((x1, y1), (x2, y2), duration=random.uniform(0.2, 0.5))

# 每条评论间隔随机等待
sleep(random.uniform(3, 8))

# 降低阈值（默认 0.7）以适应轻微变化
touch(Template("btn.png", threshold=0.6, rgb=True))

# 使用灰度模板，忽略色彩差异（适用于按钮背景色渐变的情况）
touch(Template("btn.png", threshold=0.8, rgb=False))

def safe_touch(template, max_retry=3):
    for i in range(max_retry):
        if exists(template):
            touch(template)
            return True
        sleep(1)
    raise Exception(f"未找到元素: {template}")

try:
    touch(Template("comment.png"))
except TargetNotFoundError:
    snapshot(filename="error_screen.png")
    # 上报到监控平台，通知人工介入