1. # -*- coding: utf-8 -*-#
    2. # file:         BasePage
    3. # Author:       ShunZhe
    4. # Date:         2021/1/9
    5. import os
    6. import shutil
    7. import time
    8. from datetime import datetime
    10. from func_timeout.exceptions import FunctionTimedOut
    11. from selenium.webdriver.common.by import By
    12. from selenium.webdriver.support import expected_conditions as EC
    13. from selenium.webdriver.support.wait import WebDriverWait
    15. from common import logger
    17. DelayTime = 1
    20. class BasePage:
    21.     def __init__(self, driver):
    22.         self.driver = driver
    24.     # ----------------------------------------------------
    26.     def get_element(self, style, delay=DelayTime, describe=''):
    27.         """
    29.         :param style: 元素定位
    30.         :param delay:
    31.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    32.         :return:
    33.         """
    35.         if "&&" not in style:
    36.             raise NameError("语法错误，style缺少'&&'")
    38.         else:
    39.             by = style.split("&&")[0]
    40.             value = style.split("&&")[1]
    41.             for i in range(5):
    42.                 try:
    43.                     if by == "id":
    44.                         element = self.driver.find_element_by_id(value)
    45.                         logger.info('get element: ' + str(value) + '>>' + str(describe))
    46.                         time.sleep(delay)
    48.                     elif by == "name":
    49.                         element = self.driver.find_element_by_name(value)
    50.                         logger.info('get element: ' + str(value) + '>>' + str(describe))
    51.                         time.sleep(delay)
    53.                     elif by == "class":
    54.                         element = self.driver.find_element_by_class_name(value)
    55.                         logger.info('get element: ' + str(value) + '>>' + str(describe))
    56.                         time.sleep(delay)
    58.                     elif by == "link_text":
    59.                         element = self.driver.find_element_by_link_text(value)
    60.                         logger.info('get element: ' + str(value) + '>>' + str(describe))
    61.                         time.sleep(delay)
    63.                     elif by == "xpath":
    64.                         element = self.driver.find_element_by_xpath(value)
    65.                         logger.info('get element: ' + str(value) + '>>' + str(describe))
    66.                         time.sleep(delay)
    68.                     elif by == "css":
    69.                         element = self.driver.find_element_by_css_selector(value)
    70.                         logger.info('get element: ' + str(value) + '>>' + str(describe))
    71.                         time.sleep(delay)
    72.                     # if self.driver:
    73.                     #     try:
    74.                     #         style_red = 'arguments[0].style.border="2px solid #FF0000"'
    75.                     #         style_green = 'arguments[0].style.border="2px solid     #00FF00"'
    76.                     #         style_null = 'arguments[0].style.border=""'
    77.                     #
    78.                     #         for _ in range(2):
    79.                     #             self.driver.execute_script(style_red, element)
    80.                     #             time.sleep(0.1)
    81.                     #             self.driver.execute_script(style_green, element)
    82.                     #             time.sleep(0.1)
    83.                     #         self.driver.execute_script(style_green, element)
    84.                     #         time.sleep(0.2)
    85.                     #         self.driver.execute_script(style_null, element)
    86.                     #     except WebDriverException:
    87.                     #         pass
    88.                     return element
    89.                 except FunctionTimedOut:
    90.                     element = []
    91.             else:
    92.                 NameError(
    93.                     "请输入正确的定位元素：'id','name','class','link_text','xpath','css'")
    95.     def get_elements(self, style, delay=DelayTime, describe=''):
    96.         """
    98.         :param style: 元素定位
    99.         :param delay:
    100.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    101.         :return:
    102.         """
    104.         if "&&" not in style:
    105.             raise NameError("语法错误，style缺少'&&'")
    107.         else:
    108.             by = style.split("&&")[0]
    109.             value = style.split("&&")[1]
    110.             for i in range(5):
    111.                 if by == "id":
    112.                     element = self.driver.find_elements_by_id(value)
    113.                     logger.info('get element: ' + str(value) + '>>' + str(describe))
    114.                     time.sleep(delay)
    116.                 elif by == "name":
    117.                     element = self.driver.find_elements_by_name(value)
    118.                     logger.info('get element: ' + str(value) + '>>' + str(describe))
    119.                     time.sleep(delay)
    121.                 elif by == "class":
    122.                     element = self.driver.find_elements_by_class_name(value)
    123.                     logger.info('get element: ' + str(value) + '>>' + str(describe))
    124.                     time.sleep(delay)
    126.                 elif by == "link_text":
    127.                     element = self.driver.find_elements_by_link_text(value)
    128.                     logger.info('get element: ' + str(value) + '>>' + str(describe))
    129.                     time.sleep(delay)
    131.                 elif by == "xpath":
    132.                     element = self.driver.find_elements_by_xpath(value)
    133.                     logger.info('get element: ' + str(value) + '>>' + str(describe))
    134.                     time.sleep(delay)
    136.                 elif by == "css":
    137.                     element = self.driver.find_elements_by_css_selector(value)
    138.                     logger.info('get element: ' + str(value) + '>>' + str(describe))
    139.                     time.sleep(delay)
    140.                 return element
    141.             else:
    142.                 NameError(
    143.                     "请输入正确的定位元素：'id','name','class','link_text','xpath','css'")
    145.     # ----------------------------------------------------
    147.     def click_elem(self, style, describe=''):
    148.         """
    150.         :param style: 元素定位
    151.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    152.         :return:
    153.         """
    154.         if "&&" not in style:
    155.             raise NameError("语法错误，style缺少'&&'")
    157.         else:
    158.             value = style.split("&&")[1]
    159.             element = self.get_element(style).click()
    160.             logger.info('click element: ' + str(value) + '>>' + str(describe))
    161.         return element
    163.     # ----------------------------------------------------
    165.     def get_elem_send_keys(self, style, send_key, describe=None):
    166.         """
    168.         :param style:元素定位
    169.         :param send_key:
    170.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    171.         :return:
    172.         """
    173.         if "&&" not in style:
    174.             raise NameError("语法错误，style缺少'&&'")
    176.         else:
    177.             element = self.get_element(style).send_keys(send_key)
    178.             logger.info('input keys: ' + str(send_key) + '>>' + str(describe))
    179.             return element
    181.     # ----------------------------------------------------
    183.     def show_wait_visibility(self, style, timeout=20, delay=DelayTime):
    185.         """
    186.         show_wait_visibility:显示等待函数（该方法需要找到元素，并且该元素也可见）
    187.         提供6种显示等待方法，开发者随使用情况调用
    188.         style必须为 XX&&XX的格式，以字符串形式传参
    189.         && 前面是定位的方式，后面是定位的元素。例如：id&&#kk
    190.         timeout:显示等待超时时间
    191.         delay:延迟时间
    192.         # presence_of_element_located： 当我们不关心元素是否可见，只关心元素是否存在在页面中。
    193.         # visibility_of_element_located： 当我们需要找到元素，并且该元素也可见。
    194.         :return:
    195.         """
    196.         # 显示等待方法导入
    197.         if "&&" not in style:
    198.             raise NameError("语法错误，style缺少'&&'")
    200.         else:
    202.             by = style.split("&&")[0]
    203.             value = style.split("&&")[1]
    205.             if by == "id":
    206.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    207.                     EC.visibility_of_element_located((By.ID, value)))
    208.                 time.sleep(delay)
    209.                 return expected
    211.             elif by == "name":
    212.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    213.                     EC.visibility_of_element_located((By.NAME, value)))
    214.                 time.sleep(delay)
    215.                 return expected
    217.             elif by == "class":
    218.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    219.                     EC.visibility_of_element_located((By.CLASS_NAME, value)))
    220.                 time.sleep(delay)
    221.                 return expected
    223.             elif by == "link_text":
    224.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    225.                     EC.visibility_of_element_located((By.LINK_TEXT, value)))
    226.                 time.sleep(delay)
    227.                 return expected
    229.             elif by == "xpath":
    230.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    231.                     EC.visibility_of_element_located((By.XPATH, value)))
    232.                 time.sleep(delay)
    233.                 return expected
    235.             elif by == "css":
    236.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    237.                     EC.visibility_of_element_located((By.CSS_SELECTOR, value)))
    238.                 time.sleep(delay)
    239.                 return expected
    241.             else:
    242.                 raise NameError(
    243.                     "请输入正确的定位元素：'id','name','class','link_text','xpath','css'")
    245.     # ----------------------------------------------------
    247.     def show_wait_v_until_not(self, style, timeout=20, delay=DelayTime):
    249.         if "&&" not in style:
    250.             raise NameError("语法错误，style缺少'&&'")
    251.         else:
    252.             by = style.split("&&")[0]
    253.             value = style.split("&&")[1]
    255.             if by == "id":
    256.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    257.                     EC.visibility_of_element_located((By.ID, value)))
    258.                 time.sleep(delay)
    259.                 return expected
    261.             elif by == "name":
    262.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    263.                     EC.visibility_of_element_located((By.NAME, value)))
    264.                 time.sleep(delay)
    265.                 return expected
    267.             elif by == "class":
    268.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    269.                     EC.visibility_of_element_located((By.CLASS_NAME, value)))
    270.                 time.sleep(delay)
    271.                 return expected
    273.             elif by == "link_text":
    274.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    275.                     EC.visibility_of_element_located((By.LINK_TEXT, value)))
    276.                 time.sleep(delay)
    277.                 return expected
    279.             elif by == "xpath":
    280.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    281.                     EC.visibility_of_element_located((By.XPATH, value)))
    282.                 time.sleep(delay)
    283.                 return expected
    285.             elif by == "css":
    286.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    287.                     EC.visibility_of_element_located((By.CSS_SELECTOR, value)))
    288.                 time.sleep(delay)
    289.                 return expected
    291.             else:
    292.                 raise NameError(
    293.                     "请输入正确的定位元素：'id','name','class','link_text','xpath','css'")
    295.     # ----------------------------------------------------
    297.     def show_wait_presence(self, style, timeout=20, delay=DelayTime):
    299.         """
    300.         show_wait_presence:显示等待函数（该方法不关心元素是否可见，只关心元素是否存在在页面中）
    301.         提供6种显示等待方法，开发者随使用情况调用
    302.         style必须为 XX&&XX的格式，以字符串形式传参
    303.         && 前面是定位的方式，后面是定位的元素。例如：id&&#kk
    304.         timeout:显示等待超时时间
    305.         delay:延迟时间
    306.         # presence_of_element_located： 当我们不关心元素是否可见，只关心元素是否存在在页面中。
    307.         # visibility_of_element_located： 当我们需要找到元素，并且该元素也可见。
    308.         :return:
    309.         """
    310.         # 显示等待方法导入
    311.         if "&&" not in style:
    312.             raise NameError("语法错误，style缺少'&&'")
    314.         else:
    316.             by = style.split("&&")[0]
    317.             value = style.split("&&")[1]
    319.             if by == "id":
    320.                 expected = WebDriverWait(self.driver, timeout, 1).until(EC.presence_of_element_located((By.ID, value)))
    321.                 time.sleep(delay)
    322.                 return expected
    324.             elif by == "name":
    325.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    326.                     EC.presence_of_element_located((By.NAME, value)))
    327.                 time.sleep(delay)
    328.                 return expected
    330.             elif by == "class":
    331.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    332.                     EC.presence_of_element_located((By.CLASS_NAME, value)))
    333.                 time.sleep(delay)
    334.                 return expected
    336.             elif by == "link_text":
    337.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    338.                     EC.presence_of_element_located((By.LINK_TEXT, value)))
    339.                 time.sleep(delay)
    340.                 return expected
    342.             elif by == "xpath":
    343.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    344.                     EC.presence_of_element_located((By.XPATH, value)))
    345.                 time.sleep(delay)
    346.                 return expected
    348.             elif by == "css":
    349.                 expected = WebDriverWait(self.driver, timeout, 1).until(
    350.                     EC.presence_of_element_located((By.CSS_SELECTOR, value)))
    351.                 time.sleep(delay)
    352.                 return expected
    354.             else:
    355.                 raise NameError(
    356.                     "请输入正确的定位元素：'id','name','class','link_text','xpath','css'")
    358.     # ----------------------------------------------------
    360.     def show_wait_p_until_not(self, style, timeout=20, delay=DelayTime):
    361.         if "&&" not in style:
    362.             raise NameError("语法错误，style缺少'&&'")
    364.         else:
    366.             by = style.split("&&")[0]
    367.             value = style.split("&&")[1]
    369.             if by == "id":
    370.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    371.                     EC.presence_of_element_located((By.ID, value)))
    372.                 time.sleep(delay)
    373.                 return expected
    375.             elif by == "name":
    376.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    377.                     EC.presence_of_element_located((By.NAME, value)))
    378.                 time.sleep(delay)
    379.                 return expected
    381.             elif by == "class":
    382.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    383.                     EC.presence_of_element_located((By.CLASS_NAME, value)))
    384.                 time.sleep(delay)
    385.                 return expected
    387.             elif by == "link_text":
    388.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    389.                     EC.presence_of_element_located((By.LINK_TEXT, value)))
    390.                 time.sleep(delay)
    391.                 return expected
    393.             elif by == "xpath":
    394.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    395.                     EC.presence_of_element_located((By.XPATH, value)))
    396.                 time.sleep(delay)
    397.                 return expected
    399.             elif by == "css":
    400.                 expected = WebDriverWait(self.driver, timeout, 1).until_not(
    401.                     EC.presence_of_element_located((By.CSS_SELECTOR, value)))
    402.                 time.sleep(delay)
    403.                 return expected
    405.             else:
    406.                 raise NameError(
    407.                     "请输入正确的定位元素：'id','name','class','link_text','xpath','css'")
    409.     @staticmethod
    410.     def get_datetime_str():
    411.         """获取日期字符串"""
    412.         datetime_str = datetime.now().strftime('%Y%m%d%H%M%S%f')
    413.         return datetime_str
    415.     @classmethod
    416.     def create_folder(cls, folder_path, folder):
    417.         """新建文件夹并返回文件路径"""
    418.         work_file = os.path.join(folder_path, folder)
    419.         if not os.path.exists(work_file):
    420.             os.mkdir(work_file)
    421.         return work_file
    423.     """
    424.         文件操作
    425.         """
    427.     @classmethod
    428.     def clear_files(cls, path):
    429.         """清除文件夹中的数据"""
    430.         for root, dirs, files in os.walk(path):
    431.             for f in files:
    432.                 os.unlink(os.path.join(root, f))
    433.             for d in dirs:
    434.                 shutil.rmtree(os.path.join(root, d))
    1. # -*- coding: utf-8 -*-#
    2. # file:         web_page
    3. # Author:       ShunZhe
    4. # Date:         2021/1/20
    5. import os
    6. import platform
    7. import time
    9. import pyautogui
    10. import pyperclip
    11. from airtest import aircv
    12. from selenium.webdriver import ActionChains
    13. from selenium.webdriver.common.keys import Keys
    15. from common import logger
    16. from lib.base_page import BasePage
    17. from lib.ocv.cv import Template
    18. from lib.ocv.error import TargetNotFoundError
    19. from lib.ocv.settings import Settings
    20. from lib.path import Path
    22. DelayTime = 1
    23. pyautogui.PAUSE = 1  # 调用在动作执行后暂停的秒数
    24. pyautogui.FAILSAFE = True  # 启动自动防故障功能，将鼠标移动到左上坐标(0，0)，抛出failSafeException异常
    27. class Webpage(BasePage):
    28.     """
    29.     web端页面基类,提供web端定位元素的api和各种操作元素的api，包括图像识别的方法
    30.     """
    32.     def get(self, url):
    33.         """
    34.         打开连接
    35.         """
    36.         self.driver.get(url)
    37.         logger.info("open url: " + url)
    39.     def switch_to_iframe(self, frame_id):
    40.         """切换进入iframe页面"""
    41.         logger.info('switch to the iframe')
    42.         self.driver.switch_to_frame(frame_id)
    44.     def switch_to_default_content(self):
    45.         """从iframe切回主界面"""
    46.         self.driver.switch_to_default_content()
    47.         logger.info("switch to default content")
    49.     def get_current_url(self):
    50.         """
    51.         获取当前窗口url地址
    52.         """
    53.         return self.driver.current_url
    55.     def get_elem_text(self, style):
    56.         """
    57.         获取文本
    58.         :param style:
    59.         :return:
    60.         """
    61.         return self.get_element(style).text
    63.     def get_elem_attribute(self, style, val):
    64.         """
    65.         获取某个元素值
    66.         :param style:
    67.         :param val: 获取的属性
    68.         :return:
    69.         """
    70.         return self.get_element(style).get_attribute(val)
    72.     @classmethod
    73.     def upload(cls, file):
    74.         """
    75.         上传文件，兼容mac和windows系统
    76.         :param file:上传文件路径
    77.         :return:
    78.         """
    80.         if platform.system() == "Darwin":
    81.             # mac系统
    82.             x_dim, y_dim = pyautogui.size()
    83.             pyautogui.click(x_dim // 2, y_dim // 2, duration=0.5)
    84.             pyautogui.click(x_dim // 2 + 50, y_dim // 2 + 50, duration=0.5)
    85.             time.sleep(3)
    86.             pyautogui.hotkey('command', 'shift', 'g')
    87.             time.sleep(2)
    88.             pyperclip.copy('/')
    89.             pyautogui.hotkey('command', 'v')
    90.             time.sleep(1)
    91.             pyautogui.hotkey('command', 'a')
    92.             pyautogui.hotkey('command', 'delete')
    93.             time.sleep(1)
    94.             pyperclip.copy(file)
    95.             pyautogui.hotkey('command', 'v')
    96.         elif platform.system() == "Windows":
    97.             # windows系统
    98.             time.sleep(3)
    99.             pyperclip.copy(file)
    100.             pyautogui.hotkey('ctrl', 'v')
    101.         time.sleep(1)
    102.         pyautogui.press('enter', presses=2)
    103.         time.sleep(2)
    105.     def switch_to_last_window(self):
    106.         """
    107.         切换到当前浏览器最新打开的窗口
    108.         """
    109.         all_h = self.driver.window_handles
    110.         return self.driver.switch_to.window(all_h[-1])
    112.     def switch_to_first_window(self):
    113.         """
    114.         切换到当前浏览器首次打开的窗口
    115.         """
    116.         all_h = self.driver.window_handles
    117.         return self.driver.switch_to.window(all_h[0])
    119.     def get_elements_counts(self, style):
    120.         """
    121.         获取一组元素并返回元素个数
    122.         """
    123.         sum_counts = self.get_elements(style, describe="获取一组元素")
    124.         logger.info('获取一组元素个数：' + str(len(sum_counts)))
    125.         return len(sum_counts)
    127.     def select_all(self, style):
    128.         """
    129.         模拟全选ctrl+a
    130.         """
    131.         if platform.system().lower() == "darwin":
    132.             self.get_element(style).send_keys(Keys.COMMAND, "a")
    133.         else:
    134.             self.get_element(style).send_keys(Keys.CONTROL, "a")
    135.         logger.info("select_all：" + str(style.split("&&")[1]))
    137.     def keyboard_cut(self, style):
    138.         """
    139.         模拟剪切ctrl+x
    140.         """
    141.         if platform.system().lower() == "darwin":
    142.             self.get_element(style).send_keys(Keys.COMMAND, "x")
    143.         else:
    144.             self.get_element(style).send_keys(Keys.CONTROL, "x")
    145.         logger.info("keyboard_cut：" + str(style.split("&&")[1]))
    147.     def keyboard_copy(self, style):
    148.         """
    149.         模拟复制ctrl+c
    150.         """
    151.         if platform.system().lower() == "darwin":
    152.             self.get_element(style).send_keys(Keys.COMMAND, "c")
    153.         else:
    154.             self.get_element(style).send_keys(Keys.CONTROL, "c")
    155.         logger.info("keyboard_copy：" + str(style.split("&&")[1]))
    157.     def keyboard_paste(self, style):
    158.         """
    159.         模拟粘贴ctrl+v
    160.         """
    161.         if platform.system().lower() == "darwin":
    162.             self.get_element(style).send_keys(Keys.COMMAND, "v")
    163.         else:
    164.             self.get_element(style).send_keys(Keys.CONTROL, "v")
    165.         logger.info("keyboard_paste：" + str(style.split("&&")[1]))
    167.     def keyboard_delete(self, style):
    168.         """
    169.         模拟delete键
    170.         """
    171.         self.get_element(style).send_keys(Keys.DELETE)
    172.         logger.info("keyboard_delete：" + str(style.split("&&")[1]))
    174.     def keyboard_enter(self, style):
    175.         """
    176.         模拟回车键
    177.         """
    178.         self.get_element(style).send_keys(Keys.ENTER)
    179.         logger.info("keyboard_enter：" + str(style.split("&&")[1]))
    181.     """
    182.     鼠标操作
    183.     """
    185.     def move_to_element(self, style):
    186.         """
    187.         Moving the mouse to the middle of an element
    188.         """
    189.         ActionChains(self.driver).move_to_element(self.get_element(style)).perform()
    190.         logger.info("focus element:" + str(style.split("&&")[1]))
    192.     def right_click(self, style):
    193.         """右键"""
    194.         ActionChains(self.driver).context_click(self.get_element(style)).perform()
    195.         logger.info("right click element:" + str(style.split("&&")[1]))
    197.     def move_by_offset(self, x=0, y=0):
    198.         """按住鼠标移动"""
    199.         e = ActionChains(self.driver).move_by_offset(x, y)
    200.         logger.info("click_and_hold to the coordinate is ({}, {})".format(x, y))
    201.         return e
    203.     def click_and_hold(self, style):
    204.         """按住鼠标"""
    205.         res = ActionChains(self.driver).click_and_hold(on_element=self.get_element(style))
    206.         logger.info("click and hold the mouse")
    207.         return res
    209.     def click_after_focus(self, focus_style, style):
    210.         """
    211.         hover元素并点击操作
    212.         :param focus_style: hover的元素
    213.         :param style: 点击的元素
    214.         :return:
    215.         """
    216.         focus_elem = self.get_element(focus_style)
    217.         click_elem = self.get_element(style)
    218.         ActionChains(self.driver).move_to_element(focus_elem).click(click_elem).perform()
    220.     def scroll_down(self, style, amount_to_scroll=-200):
    221.         """
    222.         向下滚动指定格数
    223.         style：元素定位
    224.         amount_to_scroll：滚动格数
    225.         eg：
    226.         scroll_down(style, -10)   # 向下滚动10格
    227.         """
    228.         self.move_to_element(style)
    229.         time.sleep(2)
    230.         if abs(amount_to_scroll) > 200:
    231.             for i in range(abs(amount_to_scroll) // 200):
    232.                 time.sleep(2)
    233.                 pyautogui.scroll(-200)
    235.     def scroll_up(self, style, amount_to_scroll=200):
    236.         """
    237.         向上滚动指定格数
    238.         style：元素定位
    239.         amount_to_scroll：滚动格数
    240.         eg：
    241.         scroll_down(style, 10)   # 向上滚动10格
    243.         """
    244.         self.move_to_element(style)
    245.         time.sleep(2)
    246.         if amount_to_scroll > 200:
    247.             for i in range(int(amount_to_scroll) // 200):
    248.                 time.sleep(2)
    249.                 pyautogui.scroll(200)
    251.     """
    252.     js操作浏览器
    253.     """
    255.     def js_click(self, style, delay=DelayTime, index=0, describe=''):
    256.         """
    258.         :param style: 元素定位
    259.         :param delay:
    260.         :param index: 获取定位元素项数，默认为0，第一项
    261.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    262.         :return:
    263.         """
    264.         if "&&" not in style:
    265.             raise NameError("语法错误，style缺少'&&'")
    266.         else:
    267.             by = style.split("&&")[0]
    268.             value = style.split("&&")[1]
    269.             if by == "css":
    270.                 js = """document.querySelectorAll('{css}')[{index}].click();""".format(css=value, index=index)
    271.                 self.driver.execute_script(js)
    272.                 time.sleep(delay)
    273.                 logger.info("click element by js:" + value + '>>' + describe)
    275.     def js_get_attribute(self, style, attribute, index=0, describe=''):
    276.         """
    277.         获取指定元素属性
    278.         :param attribute: 被获取元素
    279.         :param style: 元素定位
    280.         :param index: 获取定位元素项数，默认为0，第一项
    281.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    282.         """
    283.         if "&&" not in style:
    284.             raise NameError("语法错误，style缺少'&&'")
    285.         else:
    286.             by = style.split("&&")[0]
    287.             value = style.split("&&")[1]
    288.             if by == "css":
    289.                 time.sleep(2)
    290.                 js = """return document.querySelectorAll('{css}')[{index}].getAttribute('{attr}');""". \
    291.                     format(css=value, index=index, attr=attribute)
    292.                 return self.driver.execute_script(js)
    293.             logger.info("get_attribute by js:" + value + '>>' + describe)
    295.     def js_update_inner_text(self, style, text, index=0, describe=''):
    296.         """
    297.         修改文本框
    298.         :param text:
    299.         :param style:
    300.         :param index: 获取定位元素项数，默认为0，第一项
    301.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    302.         """
    303.         if "&&" not in style:
    304.             raise NameError("语法错误，style缺少'&&'")
    305.         else:
    306.             by = style.split("&&")[0]
    307.             value = style.split("&&")[1]
    308.             if by == "css":
    309.                 js = """var elm = document.querySelectorAll('{css}')[{index}];
    310.                             elm.innerText='{text}';""".format(css=value, index=index, text=text)
    311.                 self.driver.execute_script(js)
    312.             logger.info("update text by js:" + value + '>>' + describe)
    314.     def js_get_text_content(self, style, index=0, describe=''):
    315.         """
    316.         获取文本
    317.         :param style:
    318.         :param index: 获取定位元素项数，默认为0，第一项
    319.         :param describe: 描述用例执行，主要用于日志调试记录，非必填
    320.         """
    321.         if "&&" not in style:
    322.             raise NameError("语法错误，style缺少'&&'")
    323.         else:
    324.             by = style.split("&&")[0]
    325.             value = style.split("&&")[1]
    326.             if by == "css":
    327.                 time.sleep(2)
    328.                 js = """return document.querySelectorAll('{css}')[{index}].textContent;""".format(css=value,
    329.                                                                                                   index=index)
    330.                 return self.driver.execute_script(js)
    331.             logger.info("get textContent by js:" + value + '>>' + describe)
    333.     def clear_local_storage(self):
    334.         """清除浏览器localStorage"""
    335.         js = """window.localStorage.clear();"""
    336.         self.driver.execute_script(js)
    337.         logger.info("clear localStorage successful~")
    339.     def add_cookies_and_refresh(self, cookies):
    340.         """添加浏览器cookies"""
    341.         self.driver.delete_all_cookies()
    342.         self.driver.add_cookie(cookies)
    343.         self.driver.refresh()
    344.         time.sleep(2)
    345.         logger.info("add cookies successful and refresh the browser")
    347.     @staticmethod
    348.     def get_screen_size():
    349.         """
    350.         获取当前屏幕分辨率
    351.         """
    352.         screen_size = pyautogui.size()
    353.         return screen_size
    355.     def full_screenshot(self):
    356.         """
    357.         全屏幕截图
    358.         """
    359.         screenshot_path = os.path.abspath(Path().SCREENSHOT_PATH)
    360.         if not os.path.exists(screenshot_path):
    361.             os.mkdir(screenshot_path)
    362.         time_str = self.get_datetime_str()
    363.         file_path = f'{screenshot_path}/{time_str}.png'
    364.         pyautogui.screenshot(file_path)
    365.         return file_path
    367.     def get_cv_obj(self):
    368.         """根据图片路径，将图片处理为cv2的图片处理格式"""
    369.         file_path = self.full_screenshot()
    370.         cv_obj = aircv.imread(file_path)
    371.         return cv_obj
    373.     def _try_log_screen(self, screen=None):
    374.         """
    375.         记录图像识别的屏幕截图
    377.         Args:
    378.             screen: screenshot to be saved
    380.         Returns:
    381.             None
    383.         """
    384.         # 没有默认配置，则使用临时目录存放识别的图片
    385.         if not Settings.SS_DIR:
    386.             Settings.SS_DIR = Path().TMP_IMG_PATH
    387.         if screen is None:
    388.             screen = self.get_cv_obj()
    389.         datetime_str = self.get_datetime_str()
    390.         filename = f"{datetime_str}.jpg"
    391.         filepath = os.path.join(Settings.SS_DIR, filename)
    392.         aircv.imwrite(filepath, screen)
    394.     def get_template(self, filename, threshold=None, record_pos=None, resolution=(), rgb=False):
    395.         """picture as touch/swipe/wait/exists target and extra info for cv match"""
    396.         if not resolution:
    397.             resolution = self.get_screen_size()
    398.         if not os.path.exists(filename):
    399.             filename = os.path.join(Settings.IMG_DIR_PATH, filename)
    400.         logger.info(f"查找图片: {filename}")
    401.         template = Template(filename, threshold=threshold, record_pos=record_pos, resolution=resolution, rgb=rgb)
    402.         return template
    404.     def loop_find(self, filename, timeout=Settings.FIND_TIMEOUT, threshold=None, interval=0.5, intervalfunc=None):
    405.         """
    406.         在当前屏幕查找预期图片，直到超时
    408.         Args:
    409.             filename:  mage name or image file path
    410.             timeout: time interval how long to look for the image template
    411.             threshold: default is None
    412.             interval: sleep interval before next attempt to find the image template
    413.             intervalfunc: function that is executed after unsuccessful attempt to find the image template
    415.         Raises:
    416.             TargetNotFoundError: when image template is not found in screenshot
    418.         Returns:
    419.             TargetNotFoundError if image template not found, otherwise returns the position where the image template has
    420.             been found in screenshot
    422.         """
    423.         query = self.get_template(filename, threshold=threshold)
    424.         start_time = time.time()
    425.         while True:
    426.             screen = self.get_cv_obj()
    427.             query.resolution = self.get_resolution(screen)
    428.             if screen is None:
    429.                 logger.warn("Screen is None, may be locked")
    430.             else:
    431.                 if threshold:
    432.                     query.threshold = threshold
    433.                 match_pos = query.match_in(screen)
    434.                 if match_pos:
    435.                     self._try_log_screen(screen)
    436.                     return match_pos
    437.             if intervalfunc is not None:
    438.                 intervalfunc()
    439.             # 超时则raise，未超时则进行下次循环:
    440.             if (time.time() - start_time) > timeout:
    441.                 self._try_log_screen(screen)
    442.                 raise TargetNotFoundError('Picture %s not found in screen' % query)
    443.             else:
    444.                 time.sleep(interval)
    446.     @staticmethod
    447.     def get_resolution(img):
    448.         h, w = img.shape[:2]
    449.         return w, h
    451.     @staticmethod
    452.     def click_mouse_coordinate(*pos):
    453.         """点击鼠标坐标"""
    454.         # 关闭安全特性
    455.         pyautogui.FAILSAFE = False
    456.         pyautogui.moveTo(*pos)
    457.         # 鼠标双击
    458.         pyautogui.doubleClick()
    460.     def touch(self, filename, threshold=Settings.THRESHOLD, times=1, timeout=Settings.FIND_TIMEOUT, **kwargs):
    461.         """
    462.         点击识别到的图片位置
    464.         :param threshold: 识别图片的匹配度，默认0.8
    465.         :param filename:  识别图片的文件名或文件路径，直接传文件名，会拼接上ocv/settings文件配置的IMG_DIR_PATH路径
    466.         :param times: how many touches to be performed
    467.         :param timeout: 查找图片最多等待时间
    468.         :param kwargs: platform specific `kwargs`, please refer to corresponding docs
    469.         :return: finial position to be clicked
    470.         :platforms: Android, iOS
    472.         用法：
    473.             页面类继承MobilePage
    474.             self.touch('img_name.png', threshold=0.8)
    475.         """
    476.         _pos = self.loop_find(filename, timeout=timeout, threshold=threshold)
    477.         logger.info(f'{filename},获取到的图像坐标位置：{_pos}')
    478.         for _ in range(times):
    479.             if platform.system() == 'Darwin':
    480.                 pos = (_pos[0] / 2, _pos[1] / 2)
    481.                 self.click_mouse_coordinate(pos)
    482.             else:
    483.                 self.click_mouse_coordinate(_pos)
    484.             time.sleep(1)
    485.             return _pos
    487.     def screenshots_and_return_Coordinates(self, folder, filename, _x, _y, width, height, threshold=Settings.THRESHOLD,
    488.                                            timeout=Settings.FIND_TIMEOUT):
    489.         """截图并返回截图坐标"""
    490.         folder_file = self.create_folder(Path().PLAYER_PATH, folder)
    491.         time.sleep(4)
    492.         pyautogui.moveTo(_x, _y, duration=0.1)
    493.         if platform.system() == "Darwin":
    494.             image = pyautogui.screenshot(region=(_x * 2, _y * 2, width * 2, height * 2))
    495.             image.save(os.path.join(folder_file, filename))
    496.         else:
    497.             image = pyautogui.screenshot(region=(_x, _y, width, height))
    498.             image.save(os.path.join(folder_file, filename))
    499.         logger.info("Picture {} saved successfully！传入参数x坐标{},y坐标{},宽{},高{}".format(filename, _x, _y, width, height))
    500.         start_time = time.time()
    501.         if time.time() - start_time > timeout:
    502.             raise TargetNotFoundError("Picture is not Found!")
    503.         else:
    504.             logger.info(" The image {} has been identified".format(filename))
    505.             _pos = pyautogui.locateCenterOnScreen(os.path.join(folder_file, filename), confidence=threshold)
    506.             if platform.system() == 'Darwin':
    507.                 _pos = (_pos[0] / 2, _pos[1] / 2)
    508.             else:
    509.                 _pos = _pos
    510.             logger.info("图片坐标：{}".format(_pos))
    511.             return _pos
    513.     def is_img_exists(self, filename, threshold=Settings.THRESHOLD_STRICT, timeout=Settings.FIND_TIMEOUT, msg=""):
    514.         """
    515.         验证预期图片是否存在当前屏幕
    517.         :param threshold: 识别图片的匹配度，默认0.8
    518.         :param filename:  识别图片的文件名或文件路径，直接传文件名，会拼接上ocv/settings文件配置的IMG_DIR_PATH路径
    519.         :param timeout: 查找图片最多等待时间
    520.         :param msg: short description of assertion, it will be recorded in the report
    521.         :raise AssertionError: if assertion fails
    522.         :return: coordinates of the target
    523.         """
    524.         try:
    525.             pos = self.loop_find(filename, timeout=timeout, threshold=threshold)
    526.             logger.info(f'{filename}获取到的图像坐标位置：{pos}')
    527.             return True
    528.         except TargetNotFoundError:
    529.             logger.error(f"没有匹配到图像：{filename}, 匹配度设置：{threshold} message: {msg}")
    530.             return False
    532.     def is_element_exists(self, style):
    533.         """检查元素是否存在"""
    534.         try:
    535.             self.get_element(style)
    536.             logger.info(style.split("&&")[1] + " element exists")
    537.             return True
    538.         except:
    539.             logger.info(style.split("&&")[1] + " element does not exist")
    540.             return False