#!/usr/bin/python3
# -*- conding:utf-8 -*-
__author__ = 'mayi'

"""
模拟登录豆瓣：https://www.douban.com/
"""
from selenium import webdriver

# 调用环境变量指定的PhantomJS浏览器创建浏览器对象，executable_path：指定PhantomJS位置
driver = webdriver.PhantomJS(executable_path = r"D:\Program Files\phantomjs\bin\phantomjs")

# get()方法会一直等到页面被完全加载，然后才会继续程序
driver.get("https://www.douban.com/")
# 等待3秒
driver.implicitly_wait(3)

# 通过电脑登陆前的快照
driver.save_screenshot("DoubanHome.jpg")

email = input("请输入您的邮箱/手机：")
password = input("请输入您的密码：")

# 输入账号密码：
driver.find_element_by_id("form_email").send_keys(email)
driver.find_element_by_id("form_password").send_keys(password)

# 模拟点击登录
driver.find_element_by_xpath("//input[@class='bn-submit']").click()

# 等待3秒
driver.implicitly_wait(3)

# 生成登陆后的快照
driver.save_screenshot("DoubanLanding.jpg")

# 保存登陆后的源代码
with open("douban.html", "w", encoding = "utf-8") as f:
    f.write(driver.page_source)

# 关闭当前页面，如果只有一个页面，会关闭浏览器
driver.close()

# 关闭浏览器
driver.quit()