import time
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait


def login():
  acount_num = input('請輸入賬號:\n')
  passwd_str = input('請輸入密碼:\n')
  driver = webdriver.Edge()
  url = 'http://mail.163.com/'
  driver.get(url)

  # 等待頁面加載完成，出現(xiàn)可以點擊到密碼登錄的button
  wait = WebDriverWait(driver, 10)
  wait.until(EC.element_to_be_clickable((By.ID, 'lbNormal')))
  driver.find_element_by_id('lbNormal').click()


  # 使用CSSSelector正則匹配頭部
  elem = driver.find_element_by_css_selector("iframe[id^='x-URS-iframe']")
  # 163登陸框是使用iframe進行嵌套的，所以需要先切換到該iframe
  driver.switch_to.frame(elem)

  account_el = driver.find_element_by_xpath('//input[@name="email"]')
  account_el.clear()
  account_el.send_keys(acount_num)

  password_el = driver.find_element_by_xpath('//input[@name="password"]')
  password_el.clear()
  password_el.send_keys(passwd_str)

  login_el = driver.find_element_by_xpath('//a[@id="dologin"]')
  login_el.click()

  time.sleep(10)
  cur_cookies = driver.get_cookies()
  return cur_cookies


if __name__ == '__main__':
  login()