mirror of
https://github.com/Sitoi/dailycheckin.git
synced 2024-11-17 21:58:03 +08:00
104 lines
4.1 KiB
Python
104 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
import hashlib
|
|
import json
|
|
import os
|
|
import re
|
|
|
|
import requests
|
|
|
|
|
|
class TiebaCheckIn:
|
|
def __init__(self, tieba_cookie_list):
|
|
self.tieba_cookie_list = tieba_cookie_list
|
|
|
|
@staticmethod
|
|
def login_info(session):
|
|
return session.get(url="https://zhidao.baidu.com/api/loginInfo", ).json()
|
|
|
|
def valid(self, session):
|
|
try:
|
|
content = session.get(url="http://tieba.baidu.com/dc/common/tbs")
|
|
except Exception as e:
|
|
raise Exception(f"登录验证异常,错误信息: {e}")
|
|
data = json.loads(content.text)
|
|
if data["is_login"] == 0:
|
|
raise Exception("登录失败,cookie 异常")
|
|
tbs = data["tbs"]
|
|
user_name = self.login_info(session=session)["userName"]
|
|
return tbs, user_name
|
|
|
|
@staticmethod
|
|
def tieba_list(session):
|
|
data = session.get(url="https://tieba.baidu.com/mo/q/newmoindex").json()
|
|
if data["no"] == 0:
|
|
return [x["forum_name"] for x in data["data"]["like_forum"]]
|
|
else:
|
|
return []
|
|
|
|
@staticmethod
|
|
def tieba_list_more(session):
|
|
content = session.get(url="http://tieba.baidu.com/f/like/mylike?&pn=1", timeout=(5, 20))
|
|
pn = int(re.match(r".*/f/like/mylike\?&pn=(.*?)\">尾页.*", content.text, re.S | re.I).group(1))
|
|
next_page = 1
|
|
pattern = re.compile(r".*?<a href=\"/f\?kw=.*?title=\"(.*?)\">")
|
|
while next_page <= pn:
|
|
tbname = pattern.findall(content.text)
|
|
for x in tbname:
|
|
yield x
|
|
next_page += 1
|
|
content = session.get(url=f"http://tieba.baidu.com/f/like/mylike?&pn={next_page}", timeout=(5, 20))
|
|
|
|
def get_tieba_list(self, session):
|
|
try:
|
|
session.get(url="https://tieba.baidu.com/f/user/json_userinfo", allow_redirects=False).json()
|
|
except Exception as e:
|
|
print(e)
|
|
tieba_list = self.tieba_list(session=session)
|
|
else:
|
|
tieba_list = self.tieba_list_more(session=session)
|
|
return tieba_list
|
|
|
|
@staticmethod
|
|
def sign(session, tb_name_list, tbs):
|
|
success_count, error_count, exist_count = 0, 0, 0
|
|
for tb_name in tb_name_list:
|
|
md5 = hashlib.md5(f"kw={tb_name}tbs={tbs}tiebaclient!!!".encode("utf-8")).hexdigest()
|
|
data = {"kw": tb_name, "tbs": tbs, "sign": md5}
|
|
try:
|
|
response = session.post(url="http://c.tieba.baidu.com/c/c/forum/sign", data=data).json()
|
|
print(tb_name, response)
|
|
if response["error_code"] == "0":
|
|
success_count += 1
|
|
elif response["error_code"] == "160002":
|
|
exist_count += 1
|
|
else:
|
|
error_count += 1
|
|
except Exception as e:
|
|
print(f"贴吧 {tb_name} 签到异常,原因{str(e)}")
|
|
msg = f"贴吧总数: {len(tb_name_list)}\n签到成功: {success_count}\n已经签到: {exist_count}\n签到失败: {error_count}"
|
|
return msg
|
|
|
|
def main(self):
|
|
msg_list = []
|
|
for tieba_cookie in self.tieba_cookie_list:
|
|
tieba_cookie = {
|
|
item.split("=")[0]: item.split("=")[1] for item in tieba_cookie.get("tieba_cookie").split("; ")
|
|
}
|
|
session = requests.session()
|
|
requests.utils.add_dict_to_cookiejar(session.cookies, tieba_cookie)
|
|
session.headers.update({"Referer": "https://www.baidu.com/"})
|
|
tbs, user_name = self.valid(session=session)
|
|
tb_name_list = self.get_tieba_list(session=session)
|
|
msg = self.sign(session=session, tb_name_list=tb_name_list, tbs=tbs)
|
|
msg = f"【百度贴吧签到】\n帐号信息: {user_name}\n{msg}"
|
|
print(msg)
|
|
msg_list.append(msg)
|
|
return msg_list
|
|
|
|
|
|
if __name__ == "__main__":
|
|
with open(os.path.join(os.path.dirname(os.path.dirname(__file__)), "config.json"), "r", encoding="utf-8") as f:
|
|
datas = json.loads(f.read())
|
|
_tieba_cookie_list = datas.get("TIEBA_COOKIE_LIST", [])
|
|
TiebaCheckIn(tieba_cookie_list=_tieba_cookie_list).main()
|