option.add_experimental_option("debuggerAddress", "127.0.0.1:9200")

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.options import Options
import time,os
import datetime
def fuzzy_find(x,_dict):
    # 對(duì)x通過鍵名模糊查找后返回鍵名，鍵值
    for key,value in _dict.items(): 
        if x in key:
            return key,value
    return x,None
# 定義一個(gè)字典，設(shè)置網(wǎng)頁標(biāo)題和url，后面操作會(huì)使用到    
url_dict = {
            '通達(dá)信問小達(dá)':'https://wenda.tdx.com.cn/site/wenda/index.html',
            '雪球':'https://xueqiu.com/',
            '百度':'http://www.baidu.com',
            }
class Chrome_browser():
    def __init__(self):
        print('start Chrome_browser')
        options = Options()
        options.add_experimental_option("debuggerAddress", "127.0.0.1:9200")
        self.browser = webdriver.Chrome(options=options)
        self.browser.implicitly_wait(8)  # 設(shè)置隱式時(shí)間等待
        self._max()
    def _max(self):
        self.browser.maximize_window()  # 最大化瀏覽器
    def _get_tab_dict(self):
        tab_dict = {}
        handles = self.browser.window_handles
        for handle in handles:  # 切換窗口
            # print ('switch to second window', handle)
            # self.browser.close() # 關(guān)閉第一個(gè)窗口
            self.browser.switch_to.window(handle)  # 切換到第二個(gè)窗口
            # print(self.browser.title)
            title = self.browser.title.replace(u'\xa0', '').split('-')[0] # 剔除標(biāo)題中的'-'字符
            tab_dict[title] = handle
        return tab_dict
    def open_tab(self,url_dict=url_dict):
        for k,v in url_dict.items():
            print('進(jìn)入'+k,datetime.datetime.now())
            self.browser.switch_to.new_window('tab')
            self.browser.get(v)
            self.browser.refresh()
            time.sleep(2)
    def _switch(self, name='', act=''):
        # 包含同時(shí)關(guān)閉的功能
        tab_dict = self._get_tab_dict()
        print('tab_dict',tab_dict)
        if name != '':
            key,value = fuzzy_find(name,tab_dict)
            # print('key,value',key,value)
            if value != None:
                self.browser.switch_to.window(value)
                if act == 'close':
                    self.browser.close()
                    return False
                return True
        else:
            return False
    def _close(self):
        # 關(guān)閉所有窗口，關(guān)閉單個(gè)由switch完成。
        tab_dict = self._get_tab_dict()
        for k,v in tab_dict.items():  # 切換窗口
            self.browser.switch_to.window(v)
            self.browser.close()
if __name__=='__main__':
    cb = Chrome_browser()
    cb.open_tab()
    cb._switch(name='通達(dá)信',act='') # 切換到title為name的窗口，act='close'則切換完同時(shí)關(guān)閉。
    cb._close() # 關(guān)閉所有窗口，關(guān)閉單個(gè)由switch完成。