python+selenium浏览器的二次封装

#coding=utf-8
#项目中封装浏览器的方法
#类使用
from selenium import webdriver
import time
class SeleniumDriver:
def __init__(self,browser):
self.driver = self.open_browser(browser)
def open_browser(self,browser):    #封装浏览器
try:
if browser == 'chrome':
driver = webdriver.Chrome()
elif browser == 'firefox':
driver = webdriver.Firefox()
elif browser == 'ie':
driver = webdriver.Ie()
else:
driver = webdriver.Edge()
time.sleep(5)
return driver
except:
print('打开浏览器失败')
return None
def get_url(self,url):    #封装get打开url

if self.driver !=None:
if 'http://' in url:
self.driver.get(url)
else:
print('你的url有问题')
else:
print('case 失败')
self.driver.quit()

def handle_windows(self,*args):      #封装浏览器各种操作

value = len(args)
if value == 1:
if args[0] =='max':
self.driver.maximize_window()
elif args[0] == 'min':
self.driver.minimize_window()
elif args[0] == 'back':
self.driver.back()
elif args[0] == 'go':
self.driver.forward()
else:
self.driver.refresh()
elif value == 2:
self.driver.set_window_size(args[0],args[1])
else:
print('你传递的参数有问题')
time.sleep(5)
self.driver.quit()
seleniumdriver =SeleniumDriver('chrome')
seleniumdriver.handle_windows(100,30)

• 点赞
• 收藏
• 分享
• 文章举报