1. 程式人生 > >官網pyse的修改版

官網pyse的修改版

__name__ ddr 點擊 單選 style ren warning refresh www

from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support.select import Select


class Pyse(object):
def __init__(self, browser=‘chrome‘): # 初始化,生成driver
if browser == "firefox" or browser == "ff":
driver = webdriver.Firefox()
elif browser == "chrome":
option = webdriver.ChromeOptions()
option.add_argument("--start-maximized")
driver = webdriver.Chrome(chrome_options=option)

elif browser == "internet explorer" or browser == "ie":
driver = webdriver.Ie()
elif browser == "opera":
driver = webdriver.Opera()
elif browser == "phantomjs":
driver = webdriver.PhantomJS()
elif browser == ‘edge‘:
driver = webdriver.Edge()

try:
self.driver = driver
except Exception:
raise NameError(
"Not found %s browser,You can enter ‘ie‘, ‘ff‘, ‘opera‘, ‘phantomjs‘, ‘edge‘ or ‘chrome‘." % browser)

def element_wait(self, css, secs=5): # 元素等待
‘‘‘
Waiting for an element to display.

Usage:
driver.element_wait("css=>#el",10)
‘‘‘
if "=>" not in css:
raise NameError("Positioning syntax errors, lack of ‘=>‘.")

by = css.split("=>")[0]
value = css.split("=>")[1]

if by == "id":
WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.ID, value)))
elif by == "name":
WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.NAME, value)))
elif by == "class":
WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.CLASS_NAME, value)))
elif by == "link_text":
WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.LINK_TEXT, value)))
elif by == "xpath":
WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.XPATH, value)))
elif by == "css":
WebDriverWait(self.driver, secs, 1).until(EC.presence_of_element_located((By.CSS_SELECTOR, value)))
else:
raise NameError(
"Please enter the correct targeting elements,‘id‘,‘name‘,‘class‘,‘link_text‘,‘xpath‘,‘css‘.")

def get_element(self, css): # 獲取元素
‘‘‘
Judge element positioning way, and returns the element.
‘‘‘
if "=>" not in css:
raise NameError("Positioning syntax errors, lack of ‘=>‘.")

by = css.split("=>")[0]
value = css.split("=>")[1]
self.element_wait(css)

if by == "id":
element = self.driver.find_element_by_id(value)
elif by == "name":
element = self.driver.find_element_by_name(value)
elif by == "class":
element = self.driver.find_element_by_class_name(value)
elif by == "link_text":
element = self.driver.find_element_by_link_text(value)
elif by == "xpath":
element = self.driver.find_element_by_xpath(value)
elif by == "css":
element = self.driver.find_element_by_css_selector(value)
else:
raise NameError(
"Please enter the correct targeting elements,‘id‘,‘name‘,‘class‘,‘link_text‘,‘xpath‘,‘css‘.")
return element

def open(self, url): # 打開url
‘‘‘
open url.

Usage:
driver.open("https://www.baidu.com")
‘‘‘
self.driver.get(url)

def max_window(self): # 瀏覽器窗口最大化
‘‘‘
Set browser window maximized.

Usage:
driver.max_window()
‘‘‘
self.driver.maximize_window()

def set_window(self, wide, high): # 設置瀏覽器窗口的寬和高
‘‘‘
Set browser window wide and high.

Usage:
driver.set_window(wide,high)
‘‘‘
self.driver.set_window_size(wide, high)

def type(self, css, text): # 向文本框發送字符串
‘‘‘
Operation input box.

Usage:
driver.type("css=>#el","selenium")
‘‘‘
el = self.get_element(css)
el.send_keys(text)

def clear(self, css): # 清除輸入框裏的內容
‘‘‘
Clear the contents of the input box.

Usage:
driver.clear("css=>#el")
‘‘‘
el = self.get_element(css)
el.clear()

def click(self, css): # 鼠標左鍵單擊
‘‘‘
It can click any text / image can be clicked
Connection, check box, radio buttons, and even drop-down box etc..

Usage:
driver.click("css=>#el")
‘‘‘
el = self.get_element(css)
el.click()

def right_click(self, css): # 鼠標右鍵單擊
‘‘‘
Right click element.

Usage:
driver.right_click("css=>#el")
‘‘‘
el = self.get_element(css)
ActionChains(self.driver).context_click(el).perform()

def move_to_element(self, css): # 鼠標懸浮到目標元素上再點擊
‘‘‘
Mouse over the element.

Usage:
driver.move_to_element("css=>#el")
‘‘‘
el = self.get_element(css)
ActionChains(self.driver).move_to_element(el).perform()

def double_click(self, css): # 雙擊操作
‘‘‘
Double click element.

Usage:
driver.double_click("css=>#el")
‘‘‘
el = self.get_element(css)
ActionChains(self.driver).double_click(el).perform()

def drag_and_drop(self, el_css, ta_css): # 拖拽操作,拼拼圖
‘‘‘
Drags an element a certain distance and then drops it.

Usage:
driver.drag_and_drop("css=>#el","css=>#ta")
‘‘‘
element = self.get_element(el_css)
target = self.get_element(ta_css)
ActionChains(driver).drag_and_drop(element, target).perform()

def click_text(self, text): # 包含link的點擊操作
‘‘‘
Click the element by the link text

Usage:
driver.click_text("新聞")
‘‘‘
self.driver.find_element_by_partial_link_text(text).click()

def close(self): # 關閉當前瀏覽器
‘‘‘
Simulates the user clicking the "close" button in the titlebar of a popup
window or tab.

Usage:
driver.close()
‘‘‘
self.driver.close()

def quit(self): # 關閉所有打開的瀏覽器
‘‘‘
Quit the driver and close all the windows.

Usage:
driver.quit()
‘‘‘
self.driver.quit()

def submit(self, css): # 提交操作
‘‘‘
Submit the specified form.

Usage:
driver.submit("css=>#el")
‘‘‘
el = self.get_element(css)
el.submit()

def F5(self): # 瀏覽器按鈕刷新操作
‘‘‘
Refresh the current page.

Usage:
driver.F5()
‘‘‘
self.driver.refresh()

def js(self, script): # 執行js
‘‘‘
Execute JavaScript scripts.

Usage:
driver.js("window.scrollTo(200,1000);")
‘‘‘
self.driver.execute_script(script)

def get_attribute(self, css, attribute): # 獲取屬性
‘‘‘
Gets the value of an element attribute.

Usage:
driver.get_attribute("css=>#el","type")
‘‘‘
el = self.get_element(css)
return el.get_attribute(attribute)

def get_text(self, css): # 獲取文本
‘‘‘
Get element text information.

Usage:
driver.get_text("css=>#el")
‘‘‘
el = self.get_element(css)
return el.text

def get_display(self, css): # 讓元素展示
‘‘‘
Gets the element to display,The return result is true or false.

Usage:
driver.get_display("css=>#el")
‘‘‘
el = self.get_element(css)
return el.is_displayed()

def get_title(self): # 獲取標題
‘‘‘
Get window title.

Usage:
driver.get_title()
‘‘‘
return self.driver.title

def get_url(self): # 獲取url
‘‘‘
Get the URL address of the current page.

Usage:
driver.get_url()
‘‘‘
return self.driver.current_url

def get_windows_img(self, file_path): # 截圖
‘‘‘
Get the current window screenshot.

Usage:
driver.get_windows_img()
‘‘‘
self.driver.get_screenshot_as_file(file_path)

def wait(self, secs): # 隱式等待
‘‘‘
Implicitly wait.All elements on the page.

Usage:
driver.wait(10)
‘‘‘
self.driver.implicitly_wait(secs)

def accept_alert(self): # 關閉提示框,確定
‘‘‘
Accept warning box.

Usage:
driver.accept_alert()
‘‘‘
self.driver.switch_to.alert.accept()

def dismiss_alert(self): # 關閉提示框,取消
‘‘‘
Dismisses the alert available.

Usage:
driver.dismiss_alert()
‘‘‘
self.driver.switch_to.alert.dismiss()

def switch_to_frame(self, css): # 向下跳一層
‘‘‘
Switch to the specified frame.

Usage:
driver.switch_to_frame("css=>#el")
‘‘‘
self.element_wait(css)
iframe_el = self.get_element(css)
self.driver.switch_to.frame(iframe_el)

def switch_to_frame_out(self): # 跳到最外層
‘‘‘
Returns the current form machine form at the next higher level.
Corresponding relationship with switch_to_frame () method.

Usage:
driver.switch_to_frame_out()
‘‘‘
self.driver.switch_to.default_content()

def open_new_window(self, css): # 打開新窗口並切換到新打開的窗口
‘‘‘
Open the new window and switch the handle to the newly opened window.

Usage:
driver.open_new_window()
‘‘‘
original_windows = self.driver.current_window_handle
el = self.get_element(css)
el.click()
all_handles = self.driver.window_handles
for handle in all_handles:
if handle != original_windows:
self.driver.switch_to.window(handle)

# def _save_png(self, name):
# self.get_windows_img(name)

def wait_and_save_exception(self, css, name): # 等待並保存異常
try:
self.element_wait(css, secs=5)
print(name)
return True
except Exception as e:
from lib.core.path import PICTUREPATH
name = PICTUREPATH + name + ‘.jpg‘
print(name)
self.get_windows_img(name)
return False

def wait_and_exception(self, css):
try:
self.element_wait(css, secs=10)
return True
except Exception as e:
return False

def select_by_value(self, css, value): # 通過value修改單選項
el = self.get_element(css)
Select(el).select_by_value(value)

if __name__ == ‘__main__‘:
driver = Pyse("chrome")

官網pyse的修改版