最近有时间学了学python 写了些自动化脚本来提升工作效率(zhuang bi)
不过作为一名功能测试,学习python还是有一些困难的,踩了一些坑,东拼西凑终于写好了一个能用来跑回归的脚本,还算蛮有成就感的~
我主要用了pytesser3来实现对图片验证码的识别功能;
使用Python selenium 的八种定位方式、配合Katalon Recorder(Selenium IDE for FF55+)#火狐插件 来抓取元素来实现自动化;
由于代码不规范,有些元素异常难抓 我偷懒用了 外部库PyMouse 直接点击屏幕座标点,简单粗暴
下面是源码 一些相关资源我放到最后了
#coding = utf -8
from selenium import webdriver
import time ,unittest
import yzm #这个是我自己写的图片验证码函数 源码在下面
from pymouse import PyMouse # 外部库PyMouse
#shpt类继承unittest.TestCase类,从TestCase继承是告诉unittest,这是一个测试案例
class TEXT(unittest.TestCase):
#setUp用户设置初始化,在测试用例执行前这个方法中的函数先被调用,这里将浏览器的调用URL的访问初始化
def setUp(self):
self.browser = webdriver.Firefox()
self.base_url = "https://www.baidu.com"
#脚本运行时,错误信息将被打印到这个空列表中
self.verificationErrors = []
#是否继续接受下一个警告
self.accpt_next_alert = True
def test_shpt_login(self):
browser = self.browser
browser.get("http://********/busiclient/") #公司网址我就不发了。。。
browser.maximize_window() #将浏览器最大化显示
#7输入正确用户名和密码
browser.refresh() # 刷新方法 refresh
browser.find_element_by_name("loginname").click()
browser.find_element_by_name("loginname").send_keys("ad11")
browser.find_element_by_name("password").click()
browser.find_element_by_name("password").send_keys("111111")
y = yzm.yzm_1(855,600,930,624)#这是座标,用来截图的,交给pytesser3来识别
browser.find_element_by_name("code").send_keys(y)
browser.find_element_by_id("to-recover").click()
time.sleep(5)
while browser.current_url != "http://********"):#我们服务器有点慢,验证码错误提示又抓不到。。所以我是通过主页url来判断是否登录成功的。。
#print("登录成功")
print("重新输入验证码")
browser.get("http://********/busiclient/")
#browser.refresh() # 刷新方法 refresh
browser.find_element_by_name("loginname").click()
browser.find_element_by_name("loginname").send_keys("ad11")
browser.find_element_by_name("password").click()
browser.find_element_by_name("password").send_keys("111111")
y = yzm.yzm_1(855,600,930,624)
browser.find_element_by_name("code").send_keys(y)
browser.find_element_by_id("to-recover").click()
time.sleep(5)
print("商户登录成功")
#我的停车券
try:
browser.find_element_by_link_text(u"我的停车券").click()
self.assertEqual(u"我的停车券", browser.title)
print("我的停车券连接 pass")
except AssertionError as e:
print ("我的停车券连接 gg" )
browser.implicitly_wait(5)
#设置审批停车券
name_1 = browser.find_element_by_xpath("//div[3]/div[2]/table/tbody/tr/td/div").text#名称
type_1 = browser.find_element_by_xpath("//td[2]/div").text#类型
much_1 = browser.find_element_by_xpath("//td[3]/div").text#额度
way_1 = browser.find_element_by_xpath("//td[5]/div").text#使用方式
# key_1 = browser.find_element_by_xpath("//td[6]/div").text#使用密码
m = PyMouse()
#print(m.position())#获取当前座标的位置
m.click(1334,285)#鼠标移动到xy位置
#print(m.position())#获取当前座标的位置
browser.implicitly_wait(5)
name_2 = browser.find_element_by_id("couponName").text#名称
name_2 = browser.find_element_by_id("couponType").text#类型
type_2 =browser.find_element_by_id("couponCount").text#额度
much_2 =browser.find_element_by_xpath("//form[@id='couponForm']/div/div/div/i").click()#审批
browser.find_element_by_link_text(u"确认").click()
browser.implicitly_wait(5)
name_3 = browser.find_element_by_xpath("//div[3]/div[2]/table/tbody/tr/td/div").text#名称
type_3 = browser.find_element_by_xpath("//td[2]/div").text#类型
much_3 = browser.find_element_by_xpath("//td[3]/div").text#额度
way_3 = browser.find_element_by_xpath("//td[5]/div").text#使用方式
try:
name_1 == name_2 == name_3
type_1 == type_2 == type_3
much_1 == much_2 == much_3
way_3 =="审批"
print("设置审批停车券 pass")
except AssertionError as e:
print ("设置审批停车券 gg" )
browser.implicitly_wait(5)
#设置密码停车券
name_1 = browser.find_element_by_xpath("//div[3]/div[2]/table/tbody/tr/td/div").text#名称
type_1 = browser.find_element_by_xpath("//td[2]/div").text#类型
much_1 = browser.find_element_by_xpath("//td[3]/div").text#额度
way_1 = browser.find_element_by_xpath("//td[5]/div").text#使用方式
m = PyMouse()
m.click(1334,285)#鼠标移动到xy位置
browser.implicitly_wait(5)
name_2 = browser.find_element_by_id("couponName").text#名称
name_2 = browser.find_element_by_id("couponType").text#类型
type_2 =browser.find_element_by_id("couponCount").text#额度
browser.find_element_by_xpath("//form[@id='couponForm']/div/div/div[2]/i").click()#密码
browser.find_element_by_name("USAGE_PWD").click()
browser.find_element_by_name("USAGE_PWD").send_keys("2333")
browser.find_element_by_link_text(u"确认").click()
browser.implicitly_wait(5)
name_3 = browser.find_element_by_xpath("//div[3]/div[2]/table/tbody/tr/td/div").text#名称
type_3 = browser.find_element_by_xpath("//td[2]/div").text#类型
much_3 = browser.find_element_by_xpath("//td[3]/div").text#额度
way_3 = browser.find_element_by_xpath("//td[5]/div").text#使用方式
key_3 = browser.find_element_by_xpath("//td[6]/div").text#使用密码
try:
name_1 == name_2 == name_3
type_1 == type_2 == type_3
much_1 == much_2 == much_3
way_3 =="审批"
key_3 =="2333"
print("设置密码停车券 pass")
except AssertionError as e:
print ("设置密码停车券 gg" )
browser.implicitly_wait(5)
#设置无限制停车券
name_1 = browser.find_element_by_xpath("//div[3]/div[2]/table/tbody/tr/td/div").text#名称
type_1 = browser.find_element_by_xpath("//td[2]/div").text#类型
much_1 = browser.find_element_by_xpath("//td[3]/div").text#额度
way_1 = browser.find_element_by_xpath("//td[5]/div").text#使用方式
# key_1 = browser.find_element_by_xpath("//td[6]/div").text#使用密码
m = PyMouse()
#print(m.position())#获取当前座标的位置
m.click(1334,285)#鼠标移动到xy位置
#print(m.position())#获取当前座标的位置
browser.implicitly_wait(5)
name_2 = browser.find_element_by_id("couponName").text#名称
name_2 = browser.find_element_by_id("couponType").text#类型
type_2 =browser.find_element_by_id("couponCount").text#额度
much_2 =browser.find_element_by_xpath("//form[@id='couponForm']/div/div/div[3]/i").click()#无限制
browser.find_element_by_link_text(u"确认").click()
browser.implicitly_wait(5)
name_3 = browser.find_element_by_xpath("//div[3]/div[2]/table/tbody/tr/td/div").text#名称
type_3 = browser.find_element_by_xpath("//td[2]/div").text#类型
much_3 = browser.find_element_by_xpath("//td[3]/div").text#额度
way_3 = browser.find_element_by_xpath("//td[5]/div").text#使用方式
try:
name_1 == name_2 == name_3
type_1 == type_2 == type_3
much_1 == much_2 == much_3
way_3 =="无限制"
print("设置无限制停车券 pass")
except AssertionError as e:
print ("设置无限制停车券 gg" )
# =============================================================================
# m.position()#获取当前座标的位置
# m.move(x,y)#鼠标移动到xy位置
# m.click(x,y)#移动并且在xy位置点击
# m.click(x,y,1|2)#移动并且在xy位置点击,左右键点击
# =============================================================================
#抛异常
def is_element_present(self, how, what):
try:
self.browser.find_element(by = how , value = what)
except NoSuchElementException as e :
return False
return True
#对弹窗异常的处理
def is_alert_present(self):
try:
self.browser()
except NoAlerPresentException as e:
return False
return True
#关闭警告以及对得到问文本的处理
def close_alert_and_get_its_text(self):
try:
alert = self.browser()
alert_text = alert.text
if self.accept_next_alert:
alert.accept()
else:
alert.dismiss()
return alert_text
finally:
self.accpt_next_alert = True
#tearDown方法在每个测试方法实行后调用,做所有用例执行完成的清理工作,如退出浏览器等
def tearDown(self):
self.browser.quit()
#断言 对前面verificationErrors方法获得列表进行比对,如果列表不为空,就输出列表中的报错信息
self.assertEqual([], self.verificationErrors)
if __name__ == "__main__":
#unittset.mian方法来测试类中以test开头的用例
unittest.main()
下面是图片验证码函数
#coding = utf -8
from PIL import Image
from PIL import ImageEnhance
from PIL import ImageGrab
import pytesser3
def yzm_1(x1,y1,x2,y2):
im = ImageGrab.grab((x1,y1,x2,y2))
im.save('文件路径\\Python36\\Lib\\site-packages\\\\pytesser3\\!.png')
img=Image.open('文件路径\\Python36\\Lib\\site-packages\\pytesser3\\!.png')
imgry = img.convert('L')#图像加强,二值化
sharpness =ImageEnhance.Contrast(imgry)#对比度增强
sharp_img = sharpness.enhance(2.0)
sharp_img.save("文件路径\\Python36\\Lib\\site-packages\\pytesser3\\2.png")
return pytesser3.image_file_to_string('2.png')
验证码识别.py 文件要放在pytesser3 文件夹下运行,不然会报错
https://pan.baidu.com/s/1bIzo1hVzjg4EYSnxLYMc2w 这里有一些资源 不过安装教程、报错处理什么的我不管了,请自行百度
里面的座标识别工具不是病毒。。虽然这个图标看起来怪怪的;同时这个识别到的座标和我的分辨率不一样但也能用,或许座标和分辨率就是不一样??反正取到的座标能用。。
提取码 1fmc
注:识别验证码的图片源要尽量清楚点。。要是太模糊或者线条颜色干扰太多,识别率感人