import cv2
import lxml
import wda
from lxml import etree

from script.AiTools import AiTools


# 脚本管理类
class ScriptManager():
    def __init__(self):
        super().__init__()

    # 养号
    @classmethod
    def growAccount(self, udid):
        client = wda.USBClient(udid)
        session = client.session()
        session.appium_settings({"snapshotMaxDepth": 0})

        # deviceWidth = client.window_size().width
        # deviceHeight = client.window_size().height

        img = client.screenshot()
        tempPath = "resources/bgv.jpg"
        img.save(tempPath)

        smallImage = AiTools.pathWithName("like")
        bigImage = AiTools.pathWithName("bgv")

        x, y = AiTools.find_image_in_image(bigImage, smallImage)
        print(x, y)
        # client.tap(x, y)


        # xml = session.source()
        # print(xml)
        # root = etree.fromstring(xml.encode('utf-8'))
        # try:
        #     msg = client.xpath('label="收件箱"')
        #     msg.click()
        #     print(msg)
        # except Exception as e:
        #     print(e)