【Python Selenium】简单数据生成脚本
最近因工作需要，写了一个简单的自动化脚本，纯属学习，顺便学习下selenium模块。
废话不多说，直接上代码！！
这里一位大神重写了元素定位、send_keys等方法，咱们直接进行调用。
适用Python3.6
# coding=utf-8 """
basePageUtil.py
基础类basePage，封装所有页面都公用的方法，
定义open函数，重定义find_element,switch_frame,send_keys等函数。
在初始化方法中定义驱动driver,url,pagetitle
""" from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver import ActionChains class basePage(object):
"""
basePage封装所有页面都公用的方法，例如driver, url ,FindElement等
"""
# 初始化driver,url,pagetitle等
# 实例化basePage类时，最先执行的就是__init__方法，该方法的入参，其实就是basePage类的入参。
#__init__方法不能有返回值，只能返回None
# self只实例本身，相较于类Page而言。 def __init__(self, selenium_driver, url, pagetitle='',downloadir='',pageurl=''):
self.driver = selenium_driver
self.url = url
self.pagetitle = pagetitle
self.downloadir = downloadir
self.pageurl=pageurl # 通过title断言进入的页面是否正确。
# 使用title获取当前窗口title，检查输入的title是否在当前title中，返回比较结果（True 或 False）
def on_page(self, pagetitle):
return pagetitle in self.driver.title # 打开页面，并校验页面链接是否加载正确
# 以单下划线_开头的方法，在使用import *时，该方法不会被导入，保证该方法为类私有的。
def _open(self, url, pagetitle='',pageurl=''):
# 使用get打开访问链接地址
self.driver.get(url)
self.driver.maximize_window()
print(self.driver.title,self.driver.current_url)
if pagetitle:
# 使用assert进行校验，打开的窗口title是否与配置的title一致。调用on_page()方法
assert self.on_page(pagetitle), "Check Page Error:\t%s" % url
if pageurl:
# 校验打开后的url与传入url是否一致
assert pageurl==self.driver.current_url,'{0}!={1}'.format(pageurl,self.driver.current_url) # 定义open方法，调用_open()进行打开链接
def open(self):
self._open(self.url, self.pagetitle,self.pageurl) # 重写元素定位方法
def find_element(self, loc):
try:
# 等待元素可见
return WebDriverWait(self.driver, 5).until(EC.visibility_of_element_located(loc))
except Exception as e:
elements = WebDriverWait(self.driver, 5).until(EC.visibility_of_any_elements_located(loc))
return elements[0] if elements else False #重写元素定位方法
def find_elements(self, loc):
try:
# 等待元素可见
return WebDriverWait(self.driver, 5).until(EC.visibility_of_any_elements_located(loc))
except BaseException:
print('page {0} does not have locator {1}'.format(self, loc)) # 重写switch_frame方法
def switch_frame(self, loc):
return self.driver.switch_to_frame(loc) # 重写switch_frame方法
def switch_window(self, loc):
return self.driver.switch_to_window(loc) # 定义script方法，用于执行js脚本
def script(self, src):
self.driver.execute_script(src) # 重写定义send_keys方法
def send_keys(self, loc: object, vaule: object, clear_first: object = True, click_first: object = True) -> object:
try:
if click_first:
self.find_element(loc).click()
if clear_first:
self.find_element(loc).clear() self.find_element(loc).send_keys(vaule)
except AttributeError:
print('%s page does not have "%s" locator' % (self, loc)) # 重写鼠标悬停方法
def move_to_element(self, element='', loc=''):
if loc:
element = self.find_element(loc)
elif not element:
assert False,'Not Found Element'
ActionChains(self.driver).move_to_element(element).perform()
自己写的脚本如下，脚本有些简单，没有进行类的封装：
"""
selenium驱动Firefox实现人工操作进行数据生成，输出日志到文本文件
"""
from basePageUtil import *
from selenium import webdriver
from selenium.webdriver.common.by import By
import xlrd
import re
import time
from datetime import datetime,timedelta
from selenium.webdriver.common.keys import Keys #读取Excel表格数据
def excle_table(file,table,colnameindex,file_log,fb,by_index=0):
# 某一行数据
colvalues = table.row_values(colnameindex)
#定义一个空列表，用来存放Excel的每行数据
# list = []
#表头数据
row = table.row_values(0)
if row:
list = {}
for i in range(len(colvalues)):
if i ==10:
try:
#格式化Excel表日期
excel_date = xlrd.xldate.xldate_as_datetime(colvalues[i], 0)
#日期格式化为str类型
dt = datetime.strptime(str(excel_date), "%Y-%m-%d %H:%M:%S").strftime('%Y-%m-%d')
#将日期数据传入到相应的表头下
list[row[i]] = dt
except Exception:
# e_bd = '第{0}条数据,初登日期格式错误'.format(colnameindex) + '\n'
# fb.write(e_bd)
continue
elif i == 11 or i ==12:
try:
# 格式化Excel表日期
excel_date = xlrd.xldate.xldate_as_datetime(colvalues[i], 0)
# 日期格式化为datetime类型
dt = datetime.strptime(str(excel_date), "%Y-%m-%d %H:%M:%S").strftime("%Y-%m-%d %H:%M")
# 将日期数据传入到相应的表头下
list[row[i]] = dt
except Exception:
# e_edate = '第{0}条数据，起保日期、终保日期格式错误'.format(colnameindex) + '\n'
# fb.write(e_edate)
continue
else:
list[row[i]] = colvalues[i]
return list
def flag(year):
if year % 400 == 0 or (year % 100 != 0 and year % 4 == 0):
return True
else:
return False def login():
driver = webdriver.Firefox()
url = 'http://172.20.xx.xx/xxxx/login.do'
driver.get(url)
driver.maximize_window()
UserName = "username"
PassWord = "password" RawData = basePage(driver, url)
# 登录
RawData.find_element((By.ID, 'j_username')).send_keys(UserName)
RawData.find_element((By.ID, 'j_password')).send_keys(PassWord)
RawData.find_element((By.ID, 'login_ok')).click()
# 点击数据生成
RawData.find_element((By.LINK_TEXT, '数据生成')).click() # 点击商业险数据生成
RawData.find_element((By.ID, 'main_treeDemo_34_9_span')).click()
#数据文件路径
file = r"C:\Users\zg\Desktop\data.xlsx"
# 打开excle文件
data = xlrd.open_workbook(file)
# 读取第一个sheet页
table = data.sheet_by_index(0)
# 行数
nrows = table.nrows
#错误日志文件
file_out = r'C:\Users\zg\Desktop\log.txt'
DateNow = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
fb = open(file_out,'a')
fb.write(DateNow + '\n')
# 数据条数
succ_num = 0
fail_num = 0
NUM = nrows
for data_index in range(1,NUM): listdata = excle_table(file,table,data_index,file_out,fb)
if (len(listdata) <= 0):
assert 0, u'Excle数据异常'
try:
#省份选择
RawData.find_element((By.XPATH,'//a[@lookupgroup=\'cityLookup\']/i')).click()
# RawData.find_element((By.LINK_TEXT, '河北')).click()
province = listdata['省份']
time.sleep(0.5)
#js设置搜索框大小
scri1 = "var doc = document.getElementById('city');doc.style.height='685px';"
scri2 = "var doc = document.getElementsByTagName('div');" \
"for(var i=0;i < doc.length;i++){" \
" if (doc[i].className == 'dialog dialog-box'){" \
"doc[i].style.top='20px';}}"
scri3 = "var doc = document.getElementsByTagName('div');" \
"for(var i=0;i < doc.length;i++){" \
" if (doc[i].className == 'dialogContent layoutBox unitBox'){" \
"doc[i].style.height='690px';}}"
#执行js脚本
driver.execute_script(scri1)
driver.execute_script(scri2)
driver.execute_script(scri3)
try:
RawData.find_element((By.LINK_TEXT,province)).click()
except:
e_pro = '第{0}条数据省份有误,'.format(data_index)
fb.write(e_pro)
raise #公司名称
RawData.find_element((By.XPATH,'//a[@id=\'companySelect\']/i')).click()
#获取公司名称
company = re.search('[^a-zA-Z\s]+', listdata['需要生成哪家公司的数据']).group()
try:
# 输入公司名称
RawData.find_element((By.ID, 'compname')).send_keys(company)
#点击检索
RawData.find_element((By.XPATH, "//button[text()='检索']")).click()
#点击选择
RawData.find_element((By.LINK_TEXT, '选择')).click()
except:
RawData.find_element((By.XPATH, '/html/body/div[12]/div[1]/div/div/a[1]/i')).click()
e_ComCode = '第{0}条数据公司名称有误,'.format(data_index)
fb.write(e_ComCode)
raise
try:
#数据库选择
RawData.find_element((By.XPATH,"//button[@data-id='commerAreaVsUrl']")).click()
loc_db = "/html/body/div/div/ul/li/a/span[text()='{0} 联调']".format(province)
RawData.find_element((By.XPATH, loc_db)).click()
except:
e_db = '第{0}条数据数据库名称有误,'.format(data_index)
fb.write(e_db)
raise try:
#业务种类选择
RawData.find_element((By.XPATH,"//button[@data-id='biz_bjautotypes']")).click()
YWZL = listdata['业务种类']
RawData.find_element((By.XPATH, "/html/body/div[13]/div/ul/li/a/span[text()='" + YWZL + "']")).click()
except:
e_YWZL = '第{0}条数据业务种类有误,'.format(data_index)
fb.write(e_YWZL)
raise
# 日期
# 起保日期
eds = driver.find_elements_by_xpath("//input[@name='effectiveDate']")
# 终保日期
exds = driver.find_elements_by_xpath("//input[@name='expireDate']")
# 签单日期
bds = driver.find_elements_by_xpath("//input[@name='billDate']")
# 投保查询日期
qds = driver.find_elements_by_xpath("//input[@name='queryDate']")
# 起保日期
effdt_data = listdata['起保日期']
# 终保日期
exdt_data = listdata['终保日期']
# 起保日期格式化
effdt_data_tra = datetime.strptime(effdt_data, "%Y-%m-%d %H:%M")
# 终保日期格式化
exdt_data_tra = datetime.strptime(exdt_data, "%Y-%m-%d %H:%M")
#业务种类列表
YWZL1 = ['连续5年没有发生赔款','连续4年没有发生赔款','连续3年没有发生赔款','连续2年没有发生赔款','连续1年没有发生赔款']
years = re.search('\d', YWZL).group()
#对业务种类进行判断
if YWZL in YWZL1: if flag(effdt_data_tra.year) == True and effdt_data_tra.month == 2 and effdt_data_tra.day == 29:
flagYear = True
else:
flagYear = False
for i in range(0, int(years)):
# 传入日期前数据清空
eds[i].clear()
exds[i].clear()
bds[i].clear()
qds[i].clear()
# 起保日期年份
effdt_year = effdt_data_tra.year
#起保日期减一天
effdt_now = effdt_data_tra - timedelta(days=1)
# 传入起保日期
eds[i].send_keys(str(effdt_data_tra.strftime("%Y-%m-%d %H:%M")))
# 传入终保日期
exds[i].send_keys(str(exdt_data_tra.strftime("%Y-%m-%d %H:%M")))
# 传入签单日期
bds[i].send_keys(effdt_now.strftime("%Y-%m-%d"))
# 传入投保查询日期
qds[i].send_keys(effdt_now.strftime("%Y-%m-%d"))
# 终保日期
exdt_data_tra = effdt_data_tra
if flagYear == True:
if flag(effdt_year - 1) == True:
effdt_data_tra_str = "{0}-02-29 00:00".format(effdt_year - 1)
if flag(effdt_year - 1) == False:
effdt_data_tra_str = "{0}-02-28 00:00".format(effdt_year - 1)
effdt_data_tra = datetime.strptime(effdt_data_tra_str, "%Y-%m-%d %H:%M")
else:
if flag(effdt_year) == True and (
effdt_data_tra.month > 2 or effdt_data_tra.month == 2 and effdt_data_tra.day == 29):
# 起保日期减366天
effdt_data_tra = effdt_data_tra - timedelta(days=366)
elif flag(effdt_year) == False and flag(effdt_year - 1) == True and effdt_data_tra.month < 3:
# 起保日期减366天
effdt_data_tra = effdt_data_tra - timedelta(days=366)
else:
# 起保日期减365天
effdt_data_tra = effdt_data_tra - timedelta(days=365)
else:
# 结案时间
CTs = RawData.find_elements((By.XPATH, "//input[@name='ClaimCloseTime']")) for i in range(0,int(years)):
#传入日期前数据清空
eds[0].clear()
exds[0].clear()
bds[0].clear()
qds[0].clear()
CTs[i].clear()
# 起保日期年份
effdt_year = effdt_data_tra.year
# 起保日期减一天
effdt_now = effdt_data_tra - timedelta(days=1)
# 起保日期加一天,结案时间
CT_now = effdt_data_tra + timedelta(days=i+1)
# 传入起保日期
eds[0].send_keys(str(effdt_data_tra))
# 传入终保日期
exds[0].send_keys(str(exdt_data_tra))
# 传入签单日期
bds[0].send_keys(effdt_now.strftime("%Y-%m-%d"))
# 传入投保查询日期
qds[0].send_keys(effdt_now.strftime("%Y-%m-%d"))
# 传入结案时间
CTs[i].send_keys(CT_now.strftime("%Y-%m-%d %H:%M"))
#向上翻页
RawData.find_element((By.ID, 'bjautolayout_table_css')).send_keys(Keys.PAGE_UP)
try:
#号牌选择
RawData.find_element((By.XPATH,"//button[@data-id='autolicenseType']")).click()
license = listdata['号牌种类'].strip()
RawData.find_element((By.XPATH, "/html/body/div[14]/div/ul/li/a/span[text()='"+license+"']")).click()
except:
e_license = '第{0}条数据号牌种类有误,'.format(data_index)
fb.write(e_license)
raise try:
#车辆类型
RawData.find_element((By.XPATH,"//button[@data-id='car_type']")).click()
CLLX = listdata['车辆种类']
RawData.find_element((By.XPATH, "/html/body/div[15]/div/ul/li/a/span[text()='"+ CLLX +"']")).click()
except:
e_CLLX = '第{0}条数据车辆类型有误,'.format(data_index)
fb.write(e_CLLX)
raise try:
#使用性质
RawData.find_element((By.XPATH,"//button[@data-id='car_type2s']")).click()
SYXZ = listdata['使用性质'].strip()
RawData.find_element((By.XPATH, "/html/body/div[16]/div/ul/li/a/span[text()='"+ SYXZ +"']")).click()
except:
e_SYXZ = '第{0}条数据使用性质有误,'.format(data_index)
fb.write(e_SYXZ)
raise #车主姓名
# RawData.find_element((By.ID,"roleName")).send_keys('张三') #保额
# RawData.find_element((By.ID,"sumLimit")).send_keys('1000') try:
#车三项
RawData.find_element((By.XPATH,"//button[@data-id='iscarform']")).click()
RawData.find_element((By.XPATH,"/html/body/div[17]/div/ul/li/a/span[text()='否']")).click()
except:
e_CSX = '第{0}条数据车三项选择有误,'.format(data_index)
fb.write(e_CSX)
raise try:
#初登日期
Registerdate = str(listdata['初登日期'])
RawData.find_element((By.ID, "registerdate")).send_keys(Registerdate)
except:
e_registerdate = '第{0}条数据初登日期有误,'.format(data_index)
fb.write(e_registerdate)
raise try:
#是否过户车
RawData.find_element((By.XPATH,"//button[@data-id='transferid']")).click()
GHC = listdata['是否过户车'].strip()
RawData.find_element((By.XPATH,"/html/body/div[18]/div/ul/li/a/span[text()='"+GHC+"']")).click()
except:
e_GHC = '第{0}条数据过户车选择有误,'.format(data_index)
fb.write(e_GHC)
raise try:
#数据条数
DataNum = str(int(listdata['数据条数']))
# DataNum = re.search('\d+',listdata['数据条数']).group()
RawData.find_element((By.ID, "commertshu")).send_keys(DataNum)
except:
e_DataNum = '第{0}条数据数据条数有误,'.format(data_index)
fb.write(e_DataNum)
raise try:
#是否是异地车
RawData.find_element((By.XPATH,"/html/body/div[1]/div[3]/div/div[2]/div[2]/div[2]/form/div[1]/table/tbody/tr[39]/td[2]/div/button")).click()
YDC = listdata['是否异地车'].strip()
RawData.find_element((By.XPATH,"/html/body/div[19]/div/ul/li/a/span[text()='"+YDC+"']")).click()
except:
e_YDC = '第{0}条数据异地车选择有误,'.format(data_index)
fb.write(e_YDC)
raise # 保单系数算法
RawData.find_element((By.XPATH,"//button[@data-id='case_type2']")).click()
RawData.find_element((By.XPATH,"/html/body/div[20]/div/ul/li/a/span[text()='新']")).click() # 多年车贷
RawData.find_element((By.XPATH,"/html/body/div[1]/div[3]/div/div[2]/div[2]/div[2]/form/div[1]/table/tbody/tr[40]/td[4]/div/button")).click()
RawData.find_element((By.XPATH,"/html/body/div[21]/div/ul/li/a/span[text()='否']")).click() try:
#提交公司代码
RawData.find_element((By.XPATH,"/html/body/div[1]/div[3]/div/div[2]/div[2]/div[2]/form/div[1]/table/tbody/tr[41]/td[2]/div/button")).click()
ComCode = listdata['本公司名称'].replace(' ','-').strip()
RawData.find_element((By.XPATH, "/html/body/div[22]/div/ul/li/a/span[text()='"+ ComCode +"']")).click()
except:
e_ComCode = '第{0}条数据公司代码有误,'.format(data_index)
fb.write(e_ComCode)
raise
#点击提交
RawData.find_element((By.XPATH,"//button[text()='提交']")).click()
# time.sleep(3)
RawData.find_element((By.XPATH,"//button[text()='确定']")).click()
#刷新商业数据生成页面
RawData.find_element((By.ID, 'main_treeDemo_34_9_span')).click()
# 成功条数d
succ_num += 1
time.sleep(0.5)
e_succ = '第{0}条数据生成成功'.format(data_index) + '\n'
fb.write(e_succ)
except:
e_fail = '数据生成失败,请校验数据。\n’
fb.write(e_fail)
# 刷新商业数据生成页面
RawData.find_element((By.ID, 'main_treeDemo_34_9_span')).click()
fail_num +=1
time.sleep(0.5)
fb.write('\n本次数据条数共{0}条，成功{1}条，失败{2}条。\n'.format(NUM - 1, succ_num, fail_num))
fb.close()
if __name__== '__main__':
login()
代码比较简单，不喜勿喷，本人正在学习阶段！
Python学习交流QQ群：25452556，欢迎大家加入（群主就是重写代码那位）！
【Python Selenium】简单数据生成脚本的更多相关文章
- sql server数据库备份单个表的结构和数据生成脚本【转】
1.使用场景:sql server数据库备份单个表的结构和数据,在我们要修改正式系统的数据的一天或者多条某些数据时候,要执行update语句操作,安全稳健考虑,最好先做好所修改的表的结构和数据备份! ...
- python测试框架&&数据生成&&工具最全资源汇总
xUnit frameworks 单元测试框架frameworks 框架unittest - python自带的单元测试库,开箱即用unittest2 - 加强版的单元测试框架,适用于Python 2 ...
- Python selenium巧用Javascript脚本注入解决按钮点选问题
前段时间,笔者忙于应付公司组织的雅思考试,白天.晚上但凡有空,笔者都是埋头伏案,啃剑桥雅思(剑4~剑12)的官方模拟题或者做着与雅思考试相关的准备工作,这个过程持续了40余天.最近总算鼓起勇气走进考场 ...
- sql server数据库备份单个表的结构和数据生成脚本
1.使用场景:sql server数据库备份单个表的结构和数据,在我们要修改正式系统的数据的一天或者多条某些数据时候,要执行update语句操作,安全稳健考虑,最好先做好所修改的表的结构和数据备份! ...
- Python+Selenium学习--自动生成HTML测试报告
前言 在脚本运行完成之后,除了在log.txt 文件看到运行日志外,我们更希望能生一张漂亮的测试报告来展示用例执行的结果. HTMLTestRunner 是Python 标准库的unit ...
- python+selenium 简单尝试
前言 selenium是一种自动化测试工具,简单来说浏览器会根据写好的测试脚本自动做一些操作. 关于自动化测试,一开始接触的是splinter,但是安装的时候发现它是基于selenium的,于是打算直 ...
- Python+Selenium+Unittest+HTMLTestRunner生成测试报告+发送至邮箱，记一次完整的cnblog登录测试示例，
测试思路:单个测试集.单个测试汇成多个测试集.运行测试集.生成测试报告.发送至邮箱. 第一步:建立单个测试集,以cnblog登录为例. 测试用例: cnblog的登录测试,简单分下面几种情况:(1)用 ...
- Python+selenium第一个自动化脚本
第一个自动化脚本(用Python写的) from selenium import webdriver #从selenium导入webdriber driver=webdriber.Firefox() ...
- Python&Selenium借助HTMLTestRunner生成自动化测试报告
一.摘要 本篇博文介绍Python和Selenium进行自动化测试时,借助著名的HTMLTestRunner生成自动化测试报告 HTMLTestRunner.py百度很多,版本也很多,自行搜索下载放到 ...
随机推荐
- jmeter之吞吐量、吞吐率、TPS、带宽及压力测试和负载测试及其区别
一般使用单位时间内服务器处理的请求数来描述其并发处理能力.称之为吞吐率(Throughput),单位是 “req/s”.吞吐率特指Web服务器单位时间内处理的请求数另一种描述,吞吐率是,单位时间内网络 ...
- 巧用netsh命令实现端口转发(端口映射)不求人
好处:即时生效,随意修改删除,不影响其他ip映射 记事本保存为bat格式批量添加后,并查看映射: netsh interface portproxy add v4tov4 listenport=701 ...
- Api容器在应用架构演化中的用途
单层架构 在最开始编程的时候相信大家都写过下面这种架构,界面代码,业务代码,数据库连接全部在工程面完成.当然这种架构在处理很小的程序的时候依然有生命力 两层架构 后来我们发现数据访问的代码大量重复,应 ...
- World Wind Java开发之一（转）
http://blog.csdn.net/giser_whu/article/details/40477235 参照<World wind Java三维地理信息系统开发指南随书光盘>以及官 ...
- Android（java）学习笔记102：Dalivk虚拟机的初始化过程
1. 初始化下面系统函数(调用dvmStartup函数初始化所有相关的函数) 开始学习虚拟机的初始化过程,先从dvmStartup函数开始,这个函数实现所有开始虚拟机的准备工作: dvmAllo ...
- python实现剑指offer删除链表中重复的节点
题目描述 在一个排序的链表中,存在重复的结点,请删除该链表中重复的结点,重复的结点不保留,返回链表头指针. 例如,链表1->2->3->3->4->4->5 处理后 ...
- Redis学习记录(三)
1.Redis集群的搭建 1.1redis-cluster架构图 架构细节: (1)所有的redis节点批次互联(PING-PONG机制),内部使用二进制协议优化传输速度和带宽. (2)节点的fail ...
- WebAppBuilder独立于portal之arcgis for js应用框架研究之二
WAB采用ArcGIS JavaScript for API作为地图开发底层,采用Web AppBuilder作为开发框架,利用该框架即拿即用的Widget来构建应用,比如制图.查询.地理处理.编辑. ...
- MySQL DBA从小白到大神实战
MySQL5.6 For CentOS 6.6 源码编译安装 o1.关闭防火墙o2.配置sysctl.confo3.检查操作系统上是否安装了MySQLo4.下载mysql源码包o5.添加用户和组o6. ...
- react 的虚拟dom
前端优化的主要方面就是减少页面的DOM操作,减少重排和重绘,React在这方面做了优化,采用了所谓的虚拟DOM,其实我们平时也会遇到虚拟DOM,只是你没有注意罢了,请听我娓娓道来. 所谓的虚拟DOM ...