# -*- coding: utf-8 -*-
# @Time  : 2020/5/10 23:25
# @Author : 我就是任性-Amo
# @FileName: 77.通過爬蟲實現(xiàn)GitHub網(wǎng)頁的模擬登錄.py
# @Software: PyCharm
# @Blog  ：https://blog.csdn.net/xw1680

import requests # 導入網(wǎng)絡(luò)請求模塊
from lxml import etree # 導入數(shù)據(jù)解析模塊 都是第三方模塊需要安裝 
# pip install requests/lxml如果太慢 可以加上鏡像服務(wù)器 或者在Pycharm中使用圖形化界面進行安裝


class GitHubLogin(object):
  def __init__(self, username, password):
    # 構(gòu)造頭部信息
    self.headers = {
      "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_6) "
             "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36",
      "Host": "github.com",
      "Referer": "https://github.com/login"
    }
    self.login_url = "https://github.com/login" # 登錄頁面地址
    self.post_url = "https://github.com/session" # 實現(xiàn)登錄的請求地址
    self.session = requests.Session() # 創(chuàng)建Session會話對象
    self.user_name = username # 用戶名
    self.password = password # 密碼

  # 獲取authenticity_token信息
  def get_token(self):
    # 發(fā)送登錄頁面的網(wǎng)絡(luò)請求
    response = self.session.get(self.login_url, headers=self.headers)
    if response.status_code == 200: # 判斷請求是否成功
      html = etree.HTML(response.text) # 解析html
      # 提取authenticity_token信息
      token = html.xpath("http://div[@id='login']/form/input[1]/@value")[0]
      # print(token) 測試是否能夠獲取到token
    return token # 返回信息

  # 實現(xiàn)登錄
  def login(self):
    # 請求參數(shù)
    post_data = {
      "commit": "Sign in",
      "authenticity_token": self.get_token(),
      "login": self.user_name,
      "password": self.password,
      "webauthn - support": "supported"
    }
    # 發(fā)送登錄請求
    response = self.session.post(self.post_url, headers=self.headers, data=post_data)
    if response.status_code == 200: # 判斷請求是否成功
      html = etree.HTML(response.text) # 解析html
      # 獲取注冊號碼
      register_number = html.xpath("http://div[contains(@class,'Header-item')][last()]//strong")[0]
      print(f"注冊號碼為: {register_number.text}")
    else:
      print("登錄失敗")


if __name__ == '__main__':
  user_name = input("請輸入您的用戶名:") # 獲取輸入的用戶名
  password = input("請輸入您的密碼:") # 獲取輸入的密碼
  login = GitHubLogin(user_name, password) # 創(chuàng)建登錄類對象并傳遞輸入的用戶名與密碼
  login.login()