import os

import pyautogui

import time, os

import pyperclip  # 复制

pyautogui.FAILSAFE = False

class Auto:

    def get_xs(self, x_imgs):

        """

        :param x_imgs: 用来确定表格x坐标的图片 路径列表

        :return:

        """

        xs = []

        for img in x_imgs:

            button7location = None

            while not button7location:

                button7location = pyautogui.locateOnScreen(img, grayscale=True)

                if button7location:

                    x, _ = [i / 2 for i in pyautogui.center(button7location)]

                    xs.append(x)

                else:

                    if img == x_imgs[0]:

                        return xs

        return xs

    def to_find_page(self):

        print('切换网页')

        pyautogui.hotkey('ctrl', 'tab')  # 切换网页

        # pyautogui.hotkey('command', 'tab')  # 切换网页

        time.sleep(1)

        return True

    def get_x_imgs(self, x_path):

        abs_file = os.path.abspath('.')

        if not abs_file.endswith('img'):

            abs_file = os.path.join(abs_file, 'utils/img')

        print(abs_file)

        path = os.path.join(abs_file, x_path)

        print(path)

        # x_imgs = [f'utils/img/{x_path}/x1.png', f'utils/img/{x_path}/x2.png',

        #           f'utils/img/{x_path}/x3.png', f'utils/img/{x_path}/x4.png']

        # x_imgs = [os.path.join(abs_file, i) for i in x_imgs]

        l = os.walk(path)

        x_imgs = []

        for i, v, files in l:

            for file in files:

                if file.split('.')[0].startswith('x') and file.split('.')[-1] in ['png', 'jpeg']:

                    x_imgs.append(os.path.join(path, file))

        return x_imgs

    def auto_write(self, xs, datas, x_path):

        """

        :param xs: 表格每列的x坐标列表

        :param datas: 需要填写的数据  {y1_2: 100}  y1:代表第一行, 2：代表第二列， 100为填写的值

        :param x_path: 文件夹名字（以表格名字为文件夹名）

        :return:

        """

        keys = datas.keys()

        # yes = False

        for key in keys:

            name, index = key.split('_')

            # file = f'utils/img/{x_path}/{name}.png'

            file = f'{x_path}/{name}.png'

            abs_file = os.path.abspath('.')

            if not abs_file.endswith('img'):

                abs_file = os.path.join(abs_file, 'utils/img')

            file = os.path.join(abs_file, file)

            button7location = None

            while button7location is None:

                button7location = pyautogui.locateOnScreen(file, grayscale=True, confidence=.8)

                # 滚动条下滑

                if not button7location:

                    # if not yes:

                    #     return False

                    pyautogui.scroll(-1)

            # yes = True

            _, button7y = [i / 2 for i in pyautogui.center(button7location)]

            print(key)

            pyautogui.moveTo(xs[int(index)-1], button7y, duration=.1)

            pyautogui.click()

            # pyautogui.click()

            # pyautogui.typewrite(str(datas[key]))

            # 复制粘贴效果好点

            # pyperclip.copy(datas[key])  # 先复制

            # pyautogui.hotkey('command', 'v')  # 再粘贴

        return True

def main(datas, table_name):

    """

    :param datas:

    :param table_name: 文件名＝表名

    :return:

    """

    auto = Auto()

    x_imgs = auto.get_x_imgs(x_path=table_name)

    xs = None

    n = 0

    while not xs:

        n += 1

        print(f'n:{n}')

        auto.to_find_page()

        xs = auto.get_xs(x_imgs=x_imgs)

    auto.auto_write(xs=xs, datas=datas, x_path=table_name)

if __name__ == '__main__':

    datas = {'y1_2': 1}

    main(datas=datas, table_name='test')