import os
import sys
import time

import uiautomator2 as u2
from cnocr import CnOcr


def spider_poc(serial):
    d = u2.connect(serial)
    d.screen_on()
    d.unlock()
    d.debug = False

    d.app_stop("com.facebook.katana")
    d.app_start("com.facebook.katana", use_monkey=True)
    time.sleep(3)

    # 搜索
    d.xpath('//*[@content-desc="搜索"]').click(timeout=3)

    # 点击搜索框
    d.xpath('//*[@text="搜索"]').click(timeout=3)

    # 输入搜索关键词
    d(focused=True).set_text('电子元器件')
    # 点击搜索
    d.press('enter')
    time.sleep(6)

    # 点击帖子
    d.xpath('//*[@content-desc="帖子个搜索结果"]').click(timeout=3)
    time.sleep(6)

    # 点击展开
    d.click(0.461, 0.29)

    # 图片识别
    root_dir = os.path.dirname(sys.argv[0])
    screenshot_name = f'screenshots/{time.time()}.png'
    screenshot_dir = os.path.join(root_dir, screenshot_name)
    d.screenshot().save(screenshot_dir)

    ocr = CnOcr()
    res = ocr.ocr(screenshot_dir)
    # print(res)

    # 找到包含“你的快拍”的元素
    text_before_join = ""
    for item in res:
        if item['text'] in '加入':
            break
        text_before_join += item['text']

    print(text_before_join)


if __name__ == "__main__":
    spider_poc('5D7HSG5XXCKVBI85')