python+selenium浏览器的二次封装

最新推荐文章于 2023-02-07 15:39:29 发布

Dream_flying@

最新推荐文章于 2023-02-07 15:39:29 发布

阅读量632

点赞数

本文链接：https://blog.csdn.net/qq_33940095/article/details/105301162

版权

#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)