文件夹:common下
Base文件:
from selenium.webdriver import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium import webdriver
from common.log import Log
from common.config import *
import pyperclip
import pyautogui
import datetime
import time
import re
from datetime import timedelta
from datetime import datetime as dt
from common.data_process import RandomInfo
from common.data_process import FileOperate
class BaseMethod:
expectetext = None
def __init__(self):
self.driver = None
self.log = Log()
self.driver_file_path = ConfigReader().get_value("file", "driver_file_path")
self.file_download_path = ConfigReader().get_value("file", "file_download_path") # 自动下载文件
self.configreader = ConfigReader()
def open_url(self, url):
if "chromedrive" in self.driver_file_path:
profile = webdriver.ChromeOptions()
exploer = {
'profile.default_content_settings.popups': 0,
'download.default_directory': (self.file_download_path).replace("\\\\", "\\")}
profile.add_experimental_option('prefs', exploer)
profile.add_experimental_option("excludeSwitches", ["enable-automation"])
self.driver = webdriver.Chrome(self.driver_file_path, chrome_options=profile)
elif "geckodriver" in self.driver_file_path:
profile = webdriver.FirefoxProfile()
profile.set_preference('browser.download.folderList', 2)
profile.set_preference('browser.download.dir', self.file_download_path)
profile.set_preference("browser.helperApps.alwaysAsk.force", False)
profile.set_preference('browser.download.manager.showWhenStarting', False)
profile.set_preference('browser.download.manager.focusWhenStarting', False)
profile.set_preference('browser.download.manager.alertOnEXEOpen', False)
profile.set_preference('browser.helperApps.neverAsk.openFile', 'application/exe')
profile.set_preference('browser.helperApps.neverAsk.saveToDisk', 'application/octet-stream')
profile.set_preference('browser.download.manager.showAlertOnComplete', False)
self.driver = webdriver.Firefox(executable_path=self.driver_file_path, firefox_profile=profile)
self.driver.get(url)
self.driver.maximize_window()
def find_element_by_type(self, search_type, value):
try:
ele = self.driver.find_element(search_type, value)
return ele
except Exception as e:
self.log.logMsg(5, e)
return None
def find_ele(self, value):
try:
ele = None
ele = ele if ele is not None else self.find_element_by_type(By.ID, value)
ele = ele if ele is not None else self.find_element_by_type(By.NAME, value)
ele = ele if ele is not None else self.find_element_by_type(By.CLASS_NAME, value)
ele = ele if ele is not None else self.find_element_by_type(By.TAG_NAME, value)
ele = ele if ele is not None else self.find_element_by_type(By.LINK_TEXT, value)
ele = ele if ele is not None else self.find_element_by_type(By.PARTIAL_LINK_TEXT, value)
ele = ele if ele is not None else self.find_element_by_type(By.XPATH, value)
ele = ele if ele is not None else self.find_element_by_type(By.CSS_SELECTOR, value)
if ele is None:
self.log.logMsg(3, "NotFound:" + value)
return ele
except Exception as e:
self.log.logMsg(3, e)
return None
def ele_click(self, ele):
try:
ele.click()
return 0
except Exception as e:
self.log.logMsg(3, e)
return 1
def ele_clear(self, ele):
try:
ele.send_keys(Keys.CONTROL, 'a')
ele.send_keys(Keys.BACK_SPACE)
return 0
except Exception as e:
self.log.logMsg(3, e)
return 1
def ele_input(self, ele, value):
try:
self.ele_clear(ele)
ele.send_keys(value)
return 0
except Exception as e:
self.log.logMsg(3, e)
return 1
def ele_display(self, ele): # 判断对象是否可见,即css的display属性是否为none
try:
self.driver.find_element_by_xpath(ele).is_displayed()
return 0
except Exception as e:
self.log.logMsg(3, e)
return 1
def refresh(self):
self.driver.refresh()
def locateles(self): # 获取标签属性 每个元素都有tag(标签)属性,如搜索框的标签属性,就是最前面的input
count = self.driver.find_element_by_tag_name("table").find_elements_by_tag_name("tr")
return count
def quit(self):
self.driver.quit()
def close(self):
windows = self.driver.window_handles
if len(windows) > 1:
self.driver.switch_to.window(windows[1 - len(windows)]) # 回到最新的页面
self.driver.close()
else:
self.driver.close()
def back(self):
self.driver.back()
def js_screenshotsave(self, screenshot_path, file_name, js):
self.driver.execute_script(js)
time.sleep(2)
timestr = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
self.driver.get_screenshot_as_file(f'{
screenshot_path}\\{
timestr}_{
file_name}.png')
def getelements(self, positioningexpression, scroll):
js = None
pe_key, pe_value = str(positioningexpression).split("=")[0], str(positioningexpression).split("=")[1]
if "ClassName=" in positioningexpression:
js = (f'var q=document.getElementsByClassName("demo")[0].scrollTop=' + str(scroll)).replace("demo",
pe_value)
elif "Id=" in positioningexpression:
js = (f'var q=document.getElementsById("demo").scrollTop=' + str(scroll)).replace("demo", pe_value)
elif "Name=" in positioningexpression:
js = (f'var q=document.getElementsByName("demo").scrollTop=' + str(scroll)).replace("demo", pe_value)
return js
def movedown_posit(self, positioningexpression, x, y, screenshot_path, file_name):
pe_key, pe_value = str(positioningexpression).split("=")[0], str(positioningexpression).split("=")[1]
timestr = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
self.driver.get_screenshot_as_file(f'{
screenshot_path}\\{
timestr}_{
file_name}.png')
for i in range(1, int(pe_value) + 1):
if i == 1:
ActionChains(self.driver).move_by_offset(x, y).click().perform()
time.sleep(2)
timestr = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
self.driver.get_screenshot_as_file(f'{
screenshot_path}\\{
timestr}_{
file_name}.png')
else:
ActionChains(self.driver).move_by_offset(0, 0.1).click().perform()
time.sleep(2)
timestr = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
self.driver.get_screenshot_as_file(f'{
screenshot_path}\\{
timestr}_{
file_name}.png')
def screenshotsave(self, screenshot_path, file_name, positioningexpression): # 滚动切图
scrolllist = [scroll * 100 for scroll in range(2, 7, 2)]
size = self.driver.get_window_size()
height, width = size['height'], size['width']
x, y = (lambda x: x + 16 - 67)(int(width)), (lambda y: y - 129 - 30)(int(height))
windows = self.driver.window_handles
try:
if len(windows) > 1:
self.driver.switch_to.window(windows[1 - len(windows)])
timestr = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
self.driver.get_screenshot_as_file(f'{
screenshot_path}\\{
timestr}_{
file_name}.png')
time.sleep(2)
if "movedown=" in positioningexpression:
self.movedown_posit(positioningexpression, x, y, screenshot_path, file_name)
for scroll in scrolllist:
scroll = scroll if int(height) < 600 else scroll + 100
if positioningexpression == "":
js = "var q=document.documentElement.scrollTop=" + str(scroll)
self.js_screenshotsave(screenshot_path, file_name, js)
elif "ClassName" or "Id" or "Name" in positioningexpression:
self.js_screenshotsave(screenshot_path, file_name,
self.getelements(positioningexpression, scroll))
self.driver.switch_to.window(windows[0])
else:
timestr = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
self.driver.get_screenshot_as_file(f'{
screenshot_path}\\{
timestr}_{
file_name}.png')
time.sleep(2)
if "movedown=" in positioningexpression:
self.movedown_posit(positioningexpression, x, y, screenshot_path, file_name)
scrolllist = scrolllist if int(height) < 600 else [scroll + 100 for scroll in scrolllist]
for scroll in scrolllist:
if positioningexpression == "":
js = "var q=document.documentElement.scrollTop=" + str(scroll)
self.js_screenshotsave(screenshot_path, file_name, js)
elif "ClassName" or "Id" or "Name" in positioningexpression:
self.js_screenshotsave(screenshot_path, file_name, self.getelements(positioningexpression, scroll))
return 0
except Exception as e:
self.log.logMsg(3, e)
return 1
def point(self, step