#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
File: tieba.py
Author: WFRobert
Date: 2023/3/9 15:01
cron: 12 9 * * *

new Env('贴吧签到脚本');
Description: 贴吧签到脚本,实现每日自动签到所有贴吧
Update: 2023/9/1 更新cron
fix 20240625 ArcadiaScriptPublic  频道：https://t.me/ArcadiaScript 群组：https://t.me/ArcadiaScriptPublic
增加打印签到信息 更新cron
"""
import os
import requests
import hashlib
import time
import copy
import logging
import random
import initialize

# API_URL
LIKIE_URL = "http://c.tieba.baidu.com/c/f/forum/like"
TBS_URL = "http://tieba.baidu.com/dc/common/tbs"
SIGN_URL = "http://c.tieba.baidu.com/c/c/forum/sign"

ENV = os.environ

HEADERS = {
    'Host': 'tieba.baidu.com',
    'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.71 Safari/537.36',
}
SIGN_DATA = {
    '_client_type': '2',
    '_client_version': '9.7.8.0',
    '_phone_imei': '000000000000000',
    'model': 'MI+5',
    "net_type": "1",
}

# VARIABLE NAME
COOKIE = "Cookie"
BDUSS = "BDUSS"
EQUAL = r'='
EMPTY_STR = r''
TBS = 'tbs'
PAGE_NO = 'page_no'
ONE = '1'
TIMESTAMP = "timestamp"
DATA = 'data'
FID = 'fid'
SIGN_KEY = 'tiebaclient!!!'
UTF8 = "utf-8"
SIGN = "sign"
KW = "kw"

session = requests.Session()


def get_tbs(bduss):
    logging.info("🎈获取tbs开始")
    headers = copy.copy(HEADERS)
    headers.update({COOKIE: EMPTY_STR.join([BDUSS, EQUAL, bduss])})
    try:
        tbs = session.get(url=TBS_URL, headers=headers, timeout=5).json()[TBS]
    except Exception as e:
        logging.error("😢获取tbs出错" + e)
        logging.info("🎈重新获取tbs开始")
        tbs = session.get(url=TBS_URL, headers=headers, timeout=5).json()[TBS]
    logging.info("🎈获取tbs结束")
    return tbs


def get_favorite(bduss):
    logging.info("🎈获取关注的贴吧开始")
    # 客户端关注的贴吧
    returnData = {}
    i = 1
    data = {
        'BDUSS': bduss,
        '_client_type': '2',
        '_client_id': 'wappc_1534235498291_488',
        '_client_version': '9.7.8.0',
        '_phone_imei': '000000000000000',
        'from': '1008621y',
        'page_no': '1',
        'page_size': '200',
        'model': 'MI+5',
        'net_type': '1',
        'timestamp': str(int(time.time())),
        'vcode_tag': '11',
    }
    data = encodeData(data)
    try:
        res = session.post(url=LIKIE_URL, data=data, timeout=5).json()
    except Exception as e:
        logging.error("😢获取关注的贴吧出错" + e)
        return []
    returnData = res
    if 'forum_list' not in returnData:
        returnData['forum_list'] = []
    if res['forum_list'] == []:
        # return {'gconforum': [], 'non-gconforum': []}
        return []
    if 'non-gconforum' not in returnData['forum_list']:
        returnData['forum_list']['non-gconforum'] = []
    if 'gconforum' not in returnData['forum_list']:
        returnData['forum_list']['gconforum'] = []
    while 'has_more' in res and res['has_more'] == '1':
        i = i + 1
        data = {
            'BDUSS': bduss,
            '_client_type': '2',
            '_client_id': 'wappc_1534235498291_488',
            '_client_version': '9.7.8.0',
            '_phone_imei': '000000000000000',
            'from': '1008621y',
            'page_no': str(i),
            'page_size': '200',
            'model': 'MI+5',
            'net_type': '1',
            'timestamp': str(int(time.time())),
            'vcode_tag': '11',
        }
        data = encodeData(data)
        try:
            res = session.post(url=LIKIE_URL, data=data, timeout=5).json()
        except Exception as e:
            logging.error("😢获取关注的贴吧出错" + e)
            continue
        if 'forum_list' not in res:
            continue
        if 'non-gconforum' in res['forum_list']:
            returnData['forum_list']['non-gconforum'].append(res['forum_list']['non-gconforum'])
        if 'gconforum' in res['forum_list']:
            returnData['forum_list']['gconforum'].append(res['forum_list']['gconforum'])

    t = []
    for i in returnData['forum_list']['non-gconforum']:
        if isinstance(i, list):
            for j in i:
                if isinstance(j, list):
                    for k in j:
                        t.append(k)
                else:
                    t.append(j)
        else:
            t.append(i)
    for i in returnData['forum_list']['gconforum']:
        if isinstance(i, list):
            for j in i:
                if isinstance(j, list):
                    for k in j:
                        t.append(k)
                else:
                    t.append(j)
        else:
            t.append(i)
    logging.info("🎈获取关注的贴吧结束")
    return t


def encodeData(data):
    s = EMPTY_STR
    keys = data.keys()
    for i in sorted(keys):
        s += i + EQUAL + str(data[i])
    sign = hashlib.md5((s + SIGN_KEY).encode(UTF8)).hexdigest().upper()
    data.update({SIGN: str(sign)})
    return data


def client_sign(bduss, tbs, fid, kw):
    # 客户端签到
    logging.info("😎开始签到贴吧：" + kw)
    data = copy.copy(SIGN_DATA)
    data.update({BDUSS: bduss, FID: fid, KW: kw, TBS: tbs, TIMESTAMP: str(int(time.time()))})
    data = encodeData(data)
    res = session.post(url=SIGN_URL, data=data, timeout=5).json()
    return res


def main():
    global favorites
    if ('BDUSS' not in ENV):
        logging.error("😢未配置BDUSS")
        return
    b = ENV['BDUSS'].split(
        '&')
    for n, i in enumerate(b):
        logging.info("😊开始签到第" + str(n + 1) + "个用户")
        tbs = get_tbs(i)
        favorites = get_favorite(i)
        if favorites.__len__() > 0:
            for j in favorites:
                time.sleep(random.randint(1, 5))
                client_sign(i, tbs, j["id"], j["name"])
            initialize.info_message(f"完成第{str(n + 1)}个用户签到")
        else:
            initialize.info_message(f"第{str(n + 1)}个用户没有待签到的贴吧，请明天再来签到。")
        initialize.info_message(f"第{str(n + 1)}个用户签到{len(favorites)}个贴吧\n")
    initialize.info_message("所有用户签到结束")


if __name__ == '__main__':
    initialize.init()  # 日志格式化输出
    main()
    initialize.send_notify("百度贴吧签到")  # 发送通知