python+selenium登陆网站

python代码模拟登陆网站，第一次需要手动输入

#-*-coding:utf-8-*-
from selenium import webdriver
import time,traceback
from selenium.webdriver.common.action_chains import ActionChains
import json,os

driver = webdriver.Chrome()     # 创建Chrome对象.
driver.maximize_window()
# 操作这个对象.
driver.get('https://login.test.com/')     # get方式访问
time.sleep(5)

if os.path.exists('ttcookie.txt'):
    fr=open('ttcookie.txt','r')
    cookielist=json.load(fr)
    fr.close()
    for cookie in cookielist:
        driver.add_cookie(cookie)
    driver.get('https://login.test.com/')   
else:
    #获取网站cookie
    diccookie=driver.get_cookies()
    fw=open('ttcookie.txt','w')
    json.dump(diccookie,fw)
    fw.close()
time.sleep(5)
#滚动到浏览器顶部
js_top = "var q=document.documentElement.scrollTop=0"
#滚动到浏览器底部
js_bottom = "var q=document.documentElement.scrollTop=10000"
js_bottom2 = "window.scrollTo(0,document.body.scrollHeight)"

driver.execute_script(js_top)

el = driver.find_element_by_link_text("菜单名")
el.click()
time.sleep(2)

el = driver.find_elements_by_class_name('byte-tabs-header-title')[1]
el.click()
time.sleep(2)

driver.quit()   # 使用完, 记得关闭浏览器, 不然chromedriver.exe进程为一直在内存中.