# -*- coding: utf-8 -*-

""" 知乎登入分為兩種登入
    一是手機登入 API : https://www.zhihu.com/login/phone_num
    二是郵箱登入 API : https://www.zhihu.com/login/email

    第一步、開啟首頁獲取_xref值，驗證圖片
    第二步、輸入賬號密碼
    第三步、看是否需要驗證、要則下載驗證碼圖片，手動輸入
    第四步、判斷是否登入成功、登入成功後獲取頁面值。

    requests 與 http.cookiejar 相結合使用
    session = requests.session()
    session.cookies = http.cookiejar.LWPCookies(filename='abc')
    ...
    請求網址後
    ...
    session.cookies.save()  儲存cookies

    載入cookies
    try:
        session.cookies.load(ignore_discard=True)
    except:
        print('沒有cookies')
"""
 


import requests
from bs4 import BeautifulSoup as BS
import time
from subprocess import Popen  # 開啟圖片
import http.cookiejar
import re

# 模擬瀏覽器訪問
headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/48.0.2564.109 Safari/537.36'
}
home_url = "https://www.zhihu.com"
 

base_login = "https://www.zhihu.com/#signin/"  # 一定不能寫成http,否則無法登入

session = requests.session()
session.cookies = http.cookiejar.LWPCookieJar(filename='ZhiHuCookies')
try:
    # 載入Cookies檔案
    session.cookies.load(ignore_discard=True)
except:
    print("cookie未儲存或cookie已過期")
    # 第一步 獲取_xsrf
    _xsrf = BS(session.get(home_url, headers=headers).text, "lxml"
 
).find("input", {"name": "_xsrf"})["value"]

    # 第二步 根據賬號判斷登入方式
    account = input("請輸入您的賬號：")
    password = input("請輸入您的密碼：")

    # 第三步 獲取驗證碼圖片
    gifUrl = "http://www.zhihu.com/captcha.gif?r=" + str(int(time.time() * 1000)) + "&type=login"
    gif = session.get(gifUrl, headers=headers)
    # 儲存圖片
    with open('code.gif', 'wb') as f:
        f.write(gif.content)
    # 開啟圖片
    Popen('code.gif', shell=True)
    # 輸入驗證碼
    captcha = input('captcha: ')

    data = {
        "captcha": captcha,
        "password": password,
        "_xsrf": _xsrf,
    }

    # 第四步 判斷account型別是手機號還是郵箱
    if re.match("^.+\@(\[?)[a-zA-Z0-9\-\.]+\.([a-zA-Z]{2,3}|[0-9]{1,3})(\]?)$", account):
        # 郵箱
        data["email"] = account
        base_login = base_login + "email"
    else:
        # 手機號
        data["phone_num"] = account
        base_login = base_login + "phone_num"

    print(data)

    # 第五步 登入
    response = session.post(base_login, data=data, headers=headers)
    print(response.content.decode("utf-8"))

    # 第六步 儲存cookie
    session.cookies.save()

# 獲取首頁資訊
resp = session.get(home_url, headers=headers, allow_redirects=False)
print(resp.content.decode("utf-8"))