學習目的:
掌握自動化框架中需要的一些基礎web操作
正式步驟:
使用title_contains檢查頁面是否正確
# -*- coding:utf-8 -*- import time from selenium import webdriver from selenium.webdriver.support import expected_conditions as EC dr = webdriver.Chrome() url = "https://www.baidu.com" dr.get(url) #校驗title是否包含校驗的內容 a = EC.title_contains("百度一下,你就知道") #如果包含括號內的內容,則會打印 if a: print("title正確") time.sleep(3) dr.close()
使用Expected_conditions判斷元素是否可見
# -*- coding:utf-8 -*- import time from selenium import webdriver from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.support.wait import WebDriverWait from selenium.webdriver.common.by import By url = "http://www.5itest.cn/register" dr = webdriver.Chrome() dr.get(url) time.sleep(3) EC.title_contains("注冊") #注冊輸入操作,如果用element直接傳參到WebDriverWait會報錯,提示需要傳入一個可迭代對象,而不是webelement,所以傳入locator element = dr.find_element_by_xpath('//*[@id="register_email"]') locator = (By.XPATH,'//*[@id="register_email"]') WebDriverWait(dr,10).until(EC.visibility_of_element_located(locator)) element.send_keys("111@163.com") time.sleep(3) dr.close()
獲取定位的標簽屬性或者輸入的內容
# -*- coding:utf-8 -*- import time from selenium import webdriver from selenium.webdriver.support import expected_conditions as EC url = "http://www.5itest.cn/register" dr = webdriver.Chrome() dr.get(url) time.sleep(3) EC.title_contains("注冊") email_element = dr.find_element_by_xpath('//*[@id="register_email"]') #打印定位了email的標簽其他元素屬性 print(email_element.get_attribute("placeholder")) email_element.send_keys("test@163.com") #獲取輸入的內容,查看是否輸入正確 text = email_element.get_attribute("value") assert text == "test@163.com" time.sleep(1) dr.quit()
隨機生成數據:用戶名、郵箱賬號
# -*- coding:utf-8 -*- import random import time from selenium import webdriver url = "http://www.5itest.cn/register" dr = webdriver.Chrome() dr.get(url) time.sleep(3) #Python join() 方法用於將序列中的元素以指定的字符連接生成一個新的字符串。 data_email = ''.join(random.sample("123456789abcdefg",6))+'@163.com' data_name = ''.join(random.sample("AaBbCcDdEeFf",4)) dr.find_element_by_xpath('//*[@id="register_email"]').send_keys(data_email) dr.find_element_by_xpath('//*[@id="register_nickname"]').send_keys(data_name) time.sleep(3) dr.close()
解決驗證碼--獲取驗證碼圖片
# -*- coding:utf-8 -*- import time import random from PIL import Image from selenium import webdriver from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.common.by import By from selenium.webdriver.support.wait import WebDriverWait url = "http://www.5itest.cn/register" dr = webdriver.Chrome() dr.get(url) dr.maximize_window() time.sleep(3) element_email = dr.find_element_by_xpath('//*[@id="register_email"]') elemant_name = dr.find_element_by_xpath('//*[@id="register_nickname"]') elemant_pwd = dr.find_element_by_xpath('//*[@id="register_password"]') data_email = ''.join(random.sample('asdzxc123456',6)) data_name = ''.join(random.sample('BNMGHJ',4)) locator = (By.XPATH,'//*[@id="register_email"]') WebDriverWait(dr,10).until(EC.visibility_of_element_located(locator)) element_email.send_keys(data_email+'@163.com') elemant_name.send_keys(data_name) elemant_pwd.send_keys('111111') #圖片處理 dr.save_screenshot(r"F:\python_stack\python_autotest\num.png") element_code = dr.find_element_by_xpath('//*[@id="getcode_num"]') print(element_code.location) code_left = element_code.location['x'] code_top = element_code.location['y'] code_right = element_code.size['width']+code_left code_bottom = element_code.size['height'] + code_top #打開頁面的截圖 im = Image.open("F:/python_stack/python_autotest/num.png") #根據上下左右的坐標,來截取驗證碼圖片 img = im.crop((code_left,code_top,code_right,code_bottom)) img.save("F:/python_stack/python_autotest/num_code.png") time.sleep(3) dr.close()
從驗證碼圖片獲取驗證碼
import pytesseract from PIL import Image img = Image.open("F:/python_stack/python_autotest/num_code.png") text = pytesseract.image_to_string(img) print(text)
PS:
如果報錯:tesseract is not installed or it's not in your path
請自己下載一個Tesseract-OCR
可以參考:https://blog.csdn.net/qq_42184699/article/details/92575404
重構封裝讀取配置文件方法
配置文件內容:
[RegisterElement]
element_email=id>register_email
elemant_name=id>register_nickname
elemant_pwd=id>register_password
[/RegisterElement]
讀取配置文件代碼:
# -*- coding:utf-8 -*- ''' pip install Configparser 安裝參數解析讀取的包 ''' import configparser class ReadIni(object): def __init__(self,filename = None,node= None): if filename == None: filename = "F:/python_stack/python_autotest/config/LocalElement.ini" if node == None: self.node = "RegisterElement" else: self.node = node self.cf = self.load_ini(filename) #加載文件 def load_ini(self,filename): cf = configparser.ConfigParser() cf.read(filename) return cf #獲取value值 def get_value(self,key): data = self.cf.get(self.node,key) return data if __name__ == "__main__": test = ReadIni() print(test.get_value("elemant_pwd"))
封裝定位元素類 : find_elemnet.py
# -*- coding:utf-8 -*- from util.ReadIni import ReadIni from selenium import webdriver class FindElement(object): def __init__(self, driver): self.driver = driver def get_element(self, key): readini = ReadIni() data = readini.get_value(key) by = data.split(">")[0] # print(by) value = data.split(">")[1] # print(value) try: if by == 'id': return self.driver.find_element_by_id(value) elif by == "name": return self.driver.find_element_by_name(value) elif by == "className": return self.driver.find_element_by_className(value) else: return self.driver.find_element_by_xpath(value) except: return None if __name__ == "__main__": driver = webdriver.Chrome() test = FindElement(driver) test.get_element("element_email")
頁面注冊方法封裝示例
# -*- coding:utf-8 -*- from util.find_elemnet import FindElement from selenium import webdriver class RegisterFunction(object): def __init__(self,url): self.driver = self.get_driver(url) #獲取driver,並打開網頁 def get_driver(self,url): driver = webdriver.Chrome() driver.get(url) driver.maximize_window() return driver #輸入用戶信息 def send_user_info(self,key,data): self.get_user_element(key).send_keys(data) #定位用戶的各個element,find_element為find_element.py的實例,然后把實例化的self.driver傳入 #然后find_element就可以正常定位 def get_user_element(self,key): find_element = FindElement(self.driver) user_element = find_element.get_element(key) return user_element def main(self): self.send_user_info("element_email","test@163.com") self.send_user_info("elemant_name","admin") self.send_user_info("elemant_pwd", "111111") if __name__ == "__main__": url = "http://www.5itest.cn/register" register = RegisterFunction(url) register.main()
總結:
頁面的常用方法建議本地建立一個方法庫表格,后續直接參考,這個是后續要寫的