twitterLogin.py

# twitterAccount login
# Copyright (C) 2021 yamahubuki <itiro.ishino@gmail.com>

import requests
import errorCodes

from bs4 import BeautifulSoup

# IDとpwを用いてログインし、セッションを返却


def login(id, pw):
    session = requests.Session()
    session.headers.update({"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36"})

    # STEP1: Referer対策のため、トップページへダミーアクセス
    ret = session.get("https://twitcasting.tv/", timeout=5)
    if ret.status_code != 200:
        return errorCodes.LOGIN_TWITTER_ERROR

    # STEP2: ログインページからCSRFトークンを取得
    page = session.get("https://twitcasting.tv/indexpasswordlogin.php")
    if page.status_code != 200 or len(page.history) != 0:
        return errorCodes.LOGIN_TWITTER_ERROR
    try:
        soup = BeautifulSoup(page.content, "lxml")
        form = soup.find("form", {"id": "login-form"})
        ret = form.find("input", {"name": "cs_session_id", "type": "hidden"})
        token = ret["value"]
    except:
        return errorCodes.LOGIN_TWITTER_ERROR

    # STEP3: ログイン用のリクエスト
    headers = {
        "Content-Type": "application/x-www-form-urlencoded",
        "Accept": "Accept: text/html, application/xhtml+xml, image/jxr, */*",
        "Accept-Language": "ja-JP",
    }
    body = {
        "username": id,
        "password": pw,
        "action": "login",
        "cs_session_id": token,
    }
    result = session.post("https://twitcasting.tv/indexpasswordlogin.php", body, headers=headers, timeout=5)

    # STEP4: 結果の判定と返却
    if result.status_code != 200 or len(result.history) != 1 or result.history[0].status_code != 302 or result.url != "https://twitcasting.tv/":
        if result.status_code == 200 and len(result.history) == 0 and result.url.startswith("https://twitcasting.tv/indexpasswordlogin.php"):
            return errorCodes.LOGIN_TWITTER_WRONG_ACCOUNT
        return errorCodes.LOGIN_TWITTER_ERROR
    return session