selenium 单浏览器模式

如果需要无限制的任意实例化，而不是把实例化后的对象传给另一个函数、方法处理。就需要控制单浏览器，不然每次实例化弹出来一个新浏览器，很占内存

#coding=utf8
from selenium.webdriver import  Chrome,Firefox
import unittest
class Browser():
    _browser_inst = {}
    def __init__(self,Klass = Firefox):
        if Klass in Browser._browser_inst :
            self.driver = Browser._browser_inst[Klass]
        else:
            self.driver = Klass()
            Browser._browser_inst[Klass] = self.driver

    def open(self,url):
        self.driver.get(url)

class BaiduTest(unittest.TestCase):
    def test1(self):
        b1 = Browser(Chrome)
        b1.open('https://www.baidu.com')

class SinaTest(unittest.TestCase):
    def test2(self):
        b2 = Browser(Firefox)
        b2.open('http://www.sina.com')

class SohuTest(unittest.TestCase):
    def test3(self):
        b3 = Browser(Chrome)
        b3.open('https://www.sohu.com')

if __name__ == '__main__':
   unittest.main()

这样不管实例化多少次，保证最多只能弹出一个火狐和谷歌浏览器。