selenium 搭配python多線程

from selenium import webdriver


from time import sleep, ctime
import threading


def test_baidu(browser, search):
     print('start:%s' % ctime())
     print('browser:%s,' % browser)

     if browser == "firefox":
         driver = webdriver.Firefox()
     if browser == 'chrome':
         driver = webdriver.Chrome()
     else:
         #因為沒有ie驅動,所以和其他的瀏覽器一樣,都讓他默認為chrome
         print("browser 參數有誤,默認轉為谷歌")
         driver = webdriver.Chrome()


     driver.get("http://www.baidu.com")
     driver.find_element_by_id("kw").send_keys(search)
     driver.find_element_by_id("su").click()
     sleep(2)
     driver.quit()

if __name__ == '__main__':
     # 啟動參數（指定瀏覽器與百度收縮內容）
     lists = {'chrome': 'threading',"ie":"ssjs" ,'firefox': 'python'}
     threads = []
     files = range(len(lists))
     print(files)

     # 創建線程
     print(lists.items())
     for browser, search in lists.items():
         t = threading.Thread(target=test_baidu, args=(browser, search))
         threads.append(t)
     print(threads)

     # 啟動線程
     for t in threads:
          t.start()
     for t in files:
         threads[t].join()
     print('end:%s' % ctime())