python3+selenium框架设计05-配置文件和浏览器引擎类

时间:2021-12-23 23:31:05

python3配置文件的增删改查等操作可以使用内置的ConfigParser模块,可以自行百度学习,也可以看Python3学习笔记27-ConfigParser模块

配置文件一般存放着环境信息,比如url和一些测试过程中后续需要用到的字段。还有测试中需要导入,导出的一些文件。在项目下新建Config文件夹,文件夹下新建config.ini文件。项目结构如下

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

实际项目测试中,比如测试环境和生产的地址的切换,用户信息这些都是放在配置文件中,所以会频繁的使用对配置文件进行操作的方法,那我们就需要对配置文件的增删改查进行封装

先看下config.ini文件

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

因为ConfigParser模块在写入时会把注释给清空掉。所以如果要实现配置文件去控制想要测试哪个环境信息,只能配置一个开关。没办法通过去掉注释这种方法。

environment是配置一些开关,比如browser用来配置浏览器,可以设定0是谷歌浏览器,1是火狐浏览器。switch是环境开关,0是测试环境,1是生产环境

test放一些关于测试环境用到的信息

prod放一些关于生产环境用到的信息

接下来进行具体的实现。在Common文件夹下,对Base_Page.py文件添加全局变量

path = getcwd.get_cwd()
config_path = os.path.join(path, 'Config/config.ini')
config = configparser.ConfigParser()
config.read(config_path,encoding="utf-8-sig")

上面代码是为了提供配置文件的路径,接下来是封装配置的文件的增删改查

    def config_get(self,key,section=None):
'''读取配置文件字段的值并返回'''
switch = config.get('environment', 'switch')
if section==None and switch == str(0):
section = 'test'
elif section==None and switch == str(1):
section = 'prod'
config_get = config.get(section,key)
return config_get

这是将读取配置文件的方法进行封装。代码中key是配置文件中section某个键,section是具体哪个部分的配置文件,默认是None。如果需要读的信息不是test或者prod下,才需要进行传入。来看下测试代码

from framework.Base_Page import BasePage

s = BasePage(driver=1)
bro =s.config_get('browser','environment')
print('browser是:%s' %bro) url = s.config_get('url')
print('url是:%s' % url)

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

可以看到第一个如果要读取不是test或prod下的配置内容,需要把section和key都传进去,注意顺序先key再section。第二个因为switch是1,所以读的是prod下的,也就是生产环境的url。

接下来看写入配置文件

   def config_write(self,key = None,value = None,section = None):
'''往配置文件写入键值'''
switch = config.get('environment', 'switch')
if section == None and switch == str(0):
section = 'test'
elif section == None and switch == str(1):
section = 'prod'
if key is not None and value is not None:
config.set(section,key,value)
log1.info('在section:%s下写入%s=%s' %(section,key,value))
with open(config_path,'w',encoding='utf-8')as f :
config.write(f)
else:
config.add_section(section)
log1.info('新增section:%s' % section)
with open(config_path,'w',encoding='utf-8')as f :
config.write(f)

和上面读取的思路差不多懒得解释了。来看测试代码。

s.config_write('用户名','username','environment')
s.config_write('yonghuming','username')
s.config_write(section='yonghu')

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

可以看到新增成功了。

再看下删除

 def config_delete(self,key = None,section = None):
'''删除配置文件字段'''
switch = config.get('environment', 'switch')
if section == None and switch == str(0):
section = 'test'
elif section == None and switch == str(1):
section = 'prod'
if key is not None :
config.remove_option(section,key)
log1.info('删除section:%s下key为:%s的记录' % (section,key))
with open(config_path,'w',encoding='utf-8')as f :
config.write(f)
else:
config.remove_section(section)
log1.info('删除section:%s' % section)
with open(config_path,'w',encoding='utf-8')as f :
config.write(f)

和读取也是差不多的思路,看下测试代码

s.config_delete('用户名','environment')
s.config_delete('yonghuming')
s.config_delete(section='yonghu')

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

在对配置文件操作封装文成之后,封装浏览器引擎类,根据配置文件的不对,启动不同的浏览器。

def open_browser(self):
browser = self.config_get('browser','environment')
log1.info('读取浏览器配置')
url = self.config_get('url')
log1.info('读取url:%s' % url)
try:
if browser == str(0) :
self.driver = webdriver.Chrome()
log1.info('打开的浏览器为chrome')
elif browser == str(1) :
self.driver = webdriver.Firefox()
log1.info('打开的浏览器为chrome')
self.driver.get(url)
self.driver.maximize_window()
log1.info('浏览器最大化')
self.driver.implicitly_wait(10)
log1.info('设置静态等待时间10秒')
return self.driver
except BaseException:
log1.error('浏览器打开报错')

看下测试代码

from framework.Base_Page import BasePage
import unittest
class baidu(unittest.TestCase): def test_nos(self):
s = BasePage(self)
self.driver = s.open_browser() if __name__ == '__main__':
unittest.main()

python3+selenium框架设计05-配置文件和浏览器引擎类python3+selenium框架设计05-配置文件和浏览器引擎类

我这边执行成功