python selenium切换窗口

from    selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait  
from selenium.webdriver.support import expected_conditions as EC  
from selenium.webdriver.common.by import By 
import  time,itertools
def     main():
        driver=webdriver.Firefox()
        driver.maximize_window()
        time.sleep(1)
        driver.get("https://edu.csdn.net/")
        WebDriverWait(driver,20,0.5).until(EC.presence_of_element_located((By.CLASS_NAME,'train_list.size3.clearfix.csdn-tracking-statistics')))
        print(driver.execute_script("return document.title"))
        now_window=driver.current_window_handle
        driver.switch_to.active_element.click()
        handles = driver.window_handles
        for i in itertools.cycle(range(len(handles))):
                driver.switch_to.window(handles[i])
                time.sleep(2)
        input("enter to end")
        driver.close()
if      __name__=="__main__":
        main()