初识selenium(2)——浏览器的常见操作

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()
                #打开ie驱动
            elif browser=='ie':
                driver=webdriver.Ie()
            else:
            #打开微软驱动
                driver=webdriver.Edge()
             #休眠5s，给到打开驱动的一个缓冲时间，时间可自己评估，若无该时间容易导致后续运行报错
            time.sleep(5)
            return driver
          #打开失败
        except:
            print("打开浏览器失败")
            return None
  #打开对应的网址
    def ger_url(self,url):
      #判断浏览器驱动是否启动
        if self.driver!=None:
          #判断网址是否正确
            if 'https://' in url:
               #get函数打开网址
                self.driver.get(url)
                #同样睡眠5s，给到打开网址的一个缓冲时间
                time.sleep(5)
            else:
                print("地址输入错误")
        else:
            print("浏览器打开失败，无对应驱动")
   #浏览器的常见操
    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.close()
#调用封装的函数
seleniu_driver=seleniumDriver('chrome')
seleniu_driver.handle_windows('max')