WebDriverCmd.py

# 导入selenium的浏览器驱动接口 

import sys
import os
import json
o_path = os.getcwd()  # 返回当前工作目录
sys.path.append(o_path)  # 添加自己指定的搜索路径
import platform 
import time
import pyperclip
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.keys import Keys
from selenium import webdriver 
from selenium.webdriver import ActionChains 

# pip install PyUserInput
from pykeyboard import PyKeyboard

import pyautogui 
from pynput import mouse
import pynput, time
from Common.Platform import Platform

# webdriver 下载 http://chromedriver.storage.googleapis.com/index.html

# mac mini m1 arm cpu keyboard python crash bug
if 'Darwin' not in platform.system():
    import keyboard #Using module keyboard
 

class CmdType(object): 
    CLICK = "click"
    CLICK_SCRIPT = "click_script"
    CLICK_Action = "CLICK_Action"
    CLICK_List_ALL = "CLICK_List_ALL"
    CLICK_List_Item = "CLICK_List_Item"
    INPUT = "input" 
    INPUT_CLEAR = "input_clear" 
    ENTER = "enter"
    # 粘贴
    CTR_V = "control_v"
    

class CmdInfo(object):  
    type:None
    type2:None
    cmd:None
    value:None
    delay:None
    isWaiting:None
    item:None
    index=0




class WebDriverCmd():  
    listCmd:None
    driver: None
    isMouseClick:False

    #构造函数
    def __init__(self,webdv):
        self.listCmd= []
        self.driver = webdv

    def AddCmd(self,type,cmd,value="",delay=1):
        info = CmdInfo()
        info.type = type
        info.cmd = cmd
        info.value = value
        info.delay = delay
        info.isWaiting = False
        return self.AddCmdInfo(info)

    def AddCmdWait(self,type,cmd,value="",delay=1):
        info = CmdInfo()
        info.type = type
        info.cmd = cmd
        info.value = value
        info.delay = delay
        info.isWaiting = True
        return self.AddCmdInfo(info)

    def AddCmdList(self,type2,cmd,index=0,delay=1):
        info = CmdInfo()
        info.type = CmdType.CLICK_List_Item
        info.type2 = type2
        info.cmd = cmd
        info.index = index
        info.delay = delay
        info.isWaiting = False
        return self.AddCmdInfo(info)

    def AddCmd2(self,type,cmd):
        info = CmdInfo()
        info.type = type
        info.cmd = cmd
        info.value = ""
        info.delay = 1
        info.isWaiting = False

        return self.AddCmdInfo(info)

    def AddCmdInfo(self,info): 
        self.listCmd.append(info) 
        item = None
        if self.IsElementExist(info.cmd):
            item = self.driver.find_element(By.XPATH, info.cmd)
            
        info.item = item

        return item
    
    def Clear(self):
        self.listCmd.clear()

    # 让元素在可见范围 可以点击操作
    def SetItemVisible(self,item,delay=1):
        ActionChains(self.driver).move_to_element(item).perform()
        time.sleep(delay)


    def IsElementExist(self,element):
        flag=True
        browser=self.driver
        try:
            # browser.find_element_by_css_selector(element)
            browser.find_element(By.XPATH, element)
            return flag 
        except:
            flag=False
            return flag

    def IsElementChildExist(self,parent,key):
        flag=True 
        try:
            # browser.find_element_by_css_selector(element)
            parent.find_element(By.XPATH, key)
            return flag 
        except:
            flag=False
            return flag

    def Find(self,key,isWait=False):
        item = None
        if isWait:
            if self.IsElementExist(key):
                item = self.driver.find_element(By.XPATH, key)
            else:
                # waiting
                while True:
                    time.sleep(1) 
                    print("waiting key=", key)
                    if self.IsElementExist(key): 
                        item = self.driver.find_element(By.XPATH, key)
                        break

        else: 
            item = self.driver.find_element(By.XPATH, key)  
        
        return item
 
    def GetParent(self,item,level =1):
        parent = item
        for i in range(level):
            index = i
            parent = self.GetParentInternal(parent)
        
        return parent 

    def GetParentInternal(self,item):
        return item.find_element(By.XPATH, "./..")


    
    # key = "span[@class = 'info-item-content']"
    def FindBrother(self,item,key): 
        # .//../span[@class = 'info-item-content']
        str = ".//../"+key
        return item.find_element_by_xpath(str)

    def FindChild(self,item,key,isWait=False):
        ret = None
        if isWait:
            if self.IsElementChildExist(item,key):
                ret = item.find_element(By.XPATH, key)
            else:
                # waiting
                while True:
                    time.sleep(1) 
                    print("FindChild waiting key=", key)
                    if self.IsElementChildExist(item,key): 
                        ret = item.find_element(By.XPATH, key)
                        break

        else: 
            ret = item.find_element(By.XPATH, key) 

        return ret

    def FindList(self,key,isWait=False):
        item = None
        if isWait:
            if self.IsElementExist(key):
                item = self.driver.find_elements(By.XPATH, key)
            else:
                # waiting
                while True:
                    time.sleep(1) 
                    print("waiting key=", key)
                    if self.IsElementExist(key): 
                        item = self.driver.find_elements(By.XPATH, key)
                        break

        else: 
            item = self.driver.find_elements(By.XPATH, key)  
        
        return item

    def FindListChild(self,item,key):
        return item.find_elements(By.XPATH, key)

 
    def on_click(self,x, y, button, pressed):
        print(button)
        # Button.middle left right
        if button==pynput.mouse.Button.left:
            # pyautogui.rightClick()
            self.isMouseClick = True
            return False

        return True

    def WaitKeyBoard(self,key_press):
        # text = pyautogui.confirm('这个消息弹窗是文字+OK+Cancel按钮')
        # print(text)
        # return
        if Platform.isMacSystem():
            self.isMouseClick = False
            with pynput.mouse.Listener(on_click=self.on_click) as listener:
                listener.join()
            
            while not self.isMouseClick :#making a loop
                time.sleep(1)
                print('waiting for Mouse Middle Click = ')
        else:
            while True:#making a loop
                time.sleep(1)
                print('waiting for key press = ',key_press)
                # try:  
                if keyboard.is_pressed(key_press):
                    print('You Pressed A Key!')
                    break

                
                k = PyKeyboard()
 

    def SetInputText(self, key,title): 
        webcmd = WebDriverCmd(self.driver)
        if Platform.isWindowsSystem():
            pyperclip.copy(title) 
            pyperclip.paste()
            webcmd.AddCmd2(CmdType.CLICK_Action, key)
            webcmd.AddCmd2(CmdType.CTR_V, key) 
            webcmd.Run(True)
        else:
            webcmd.AddCmd(CmdType.INPUT, key,title) 
            webcmd.Run(True)


    def DoCmd(self,item,type,value="",cmd="",index=0):
        if type == CmdType.CLICK:
                # self.driver.execute_script("arguments[0].click();", item)
                item.click()
        if type == CmdType.CLICK_SCRIPT:
            # 有些item.click() 会报InvalidArgumentException: Message: invalid argument
            self.driver.execute_script("arguments[0].click();", item) 
            
        if type == CmdType.CLICK_Action:
            # 有些item.click() 无响应 用这个鼠标模拟点击 
            action= ActionChains(self.driver)
            action.click(item).perform()

        if type == CmdType.INPUT:
            item.clear()
            item.send_keys(value)
            # item.clear()
            # item.send_keys(info.value)
            # item.text = info.value

        if type == CmdType.INPUT_CLEAR:
            item.clear()

        if type == CmdType.ENTER:
            item.send_keys(Keys.ENTER)
        if type == CmdType.CTR_V:
            item.send_keys(Keys.CONTROL,"v")
                
        if type == CmdType.CLICK_List_ALL:
            list =  self.driver.find_elements(By.XPATH, cmd)
            for item in list:
                item.click()

        # if type == CmdType.CLICK_List_Item:
        #     list =  self.driver.find_elements(By.XPATH, cmd)
        #     item = list[index]
        #     if type2 == CmdType.CLICK:
        #         item.click()
        #     if type2 == CmdType.CLICK_SCRIPT:
        #         self.driver.execute_script("arguments[0].click();", item)
        #     if type2 == CmdType.CLICK_Action:
        #         action= ActionChains(self.driver)
        #         action.click(item).perform()



# 组合查找 https://blog.csdn.net/qq_32189701/article/details/100176577
# find_element_by_xpath("//input[@class=‘s_ipt’ and @name=‘wd’]")
    def Run(self,isClear):
        for info in self.listCmd:
            if info.isWaiting:
                if self.IsElementExist(info.cmd):
                    item = self.driver.find_element(By.XPATH, info.cmd)
                else:
                    # waiting
                    while True:
                        time.sleep(1) 
                        print("waiting info.cmd=", info.cmd)
                        if self.IsElementExist(info.cmd): 
                            item = self.driver.find_element(By.XPATH, info.cmd)
                            break

            else:
                item = info.item
                if item == None:
                    item = self.driver.find_element(By.XPATH, info.cmd) 
                    info.item = item

            
            if info.type == CmdType.CLICK:
                # self.driver.execute_script("arguments[0].click();", item)
                item.click()
            if info.type == CmdType.CLICK_SCRIPT:
                # 有些item.click() 会报InvalidArgumentException: Message: invalid argument
                self.driver.execute_script("arguments[0].click();", item) 
                
            if info.type == CmdType.CLICK_Action:
                # 有些item.click() 无响应 用这个鼠标模拟点击 
                action= ActionChains(self.driver)
                action.click(item).perform()

            if info.type == CmdType.INPUT:
                item.clear()
                item.send_keys(info.value)
                # item.clear()
                # item.send_keys(info.value)
                # item.text = info.value

            if info.type == CmdType.INPUT_CLEAR:
                item.clear()

            if info.type == CmdType.ENTER:
                item.send_keys(Keys.ENTER)
            if info.type == CmdType.CTR_V:
                item.send_keys(Keys.CONTROL,"v")
                 
            if info.type == CmdType.CLICK_List_ALL:
                list =  self.driver.find_elements(By.XPATH, info.cmd)
                for item in list:
                    item.click()

            if info.type == CmdType.CLICK_List_Item:
                list =  self.driver.find_elements(By.XPATH, info.cmd)
                item = list[info.index]
                if info.type2 == CmdType.CLICK:
                    item.click()
                if info.type2 == CmdType.CLICK_SCRIPT:
                    self.driver.execute_script("arguments[0].click();", item)
                if info.type2 == CmdType.CLICK_Action:
                    action= ActionChains(self.driver)
                    action.click(item).perform()

            time.sleep(info.delay)

        if isClear:
            self.Clear()