mirror of
https://github.com/wd210010/only_for_happly.git
synced 2026-11-05 20:26:45 +08:00
130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
#!/usr/bin/python3
|
|
# -- coding: utf-8 --
|
|
# -------------------------------
|
|
# @Author : github@wd210010 https://github.com/wd210010/only_for_happly
|
|
# @Time : 2024/05/14 9:23
|
|
# -------------------------------
|
|
# cron "15 20 6,15 * * *" script-path=xxx.py,tag=匹配cron用
|
|
# const $ = new Env('百度贴吧')
|
|
|
|
#变量为百度贴吧登录后的ck全部 变量名为tieback
|
|
|
|
import hashlib
|
|
import re,os
|
|
|
|
import requests
|
|
|
|
|
|
class Tieba:
|
|
def __init__(self, check_items):
|
|
self.check_items = check_items
|
|
|
|
@staticmethod
|
|
def login_info(session):
|
|
return session.get("https://zhidao.baidu.com/api/loginInfo").json()
|
|
|
|
def valid(self, session):
|
|
try:
|
|
res = session.get("http://tieba.baidu.com/dc/common/tbs").json()
|
|
except Exception as e:
|
|
return False, f"登录验证异常，错误信息: {e}"
|
|
if res["is_login"] == 0:
|
|
return False, "登录失败，cookie 异常"
|
|
tbs = res["tbs"]
|
|
user_name = self.login_info(session=session)["userName"]
|
|
return tbs, user_name
|
|
|
|
@staticmethod
|
|
def tieba_list_more(session):
|
|
res = session.get(
|
|
"https://tieba.baidu.com/f/like/mylike?&pn=1",
|
|
timeout=(5, 20),
|
|
allow_redirects=False,
|
|
).text
|
|
try:
|
|
pn = int(
|
|
re.match(r".*/f/like/mylike\?&pn=(.*?)\">尾页.*", res, re.S | re.I)[1]
|
|
)
|
|
|
|
except Exception:
|
|
pn = 1
|
|
pattern = re.compile(r".*?<a href=\"/f\?kw=.*?title=\"(.*?)\">")
|
|
for next_page in range(2, pn + 2):
|
|
yield from pattern.findall(res)
|
|
res = session.get(
|
|
f"https://tieba.baidu.com/f/like/mylike?&pn={next_page}",
|
|
timeout=(5, 20),
|
|
allow_redirects=False,
|
|
).text
|
|
|
|
def get_tieba_list(self, session):
|
|
return list(self.tieba_list_more(session))
|
|
|
|
@staticmethod
|
|
def sign(session, tb_name_list, tbs):
|
|
success_count, error_count, exist_count, shield_count = 0, 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:
|
|
res = session.post(
|
|
url="http://c.tieba.baidu.com/c/c/forum/sign", data=data
|
|
).json()
|
|
if res["error_code"] == "0":
|
|
success_count += 1
|
|
elif res["error_code"] == "160002":
|
|
exist_count += 1
|
|
elif res["error_code"] == "340006":
|
|
shield_count += 1
|
|
else:
|
|
error_count += 1
|
|
except Exception as e:
|
|
print(f"贴吧 {tb_name} 签到异常，原因{str(e)}")
|
|
return (
|
|
f"贴吧总数: {len(tb_name_list)}\n"
|
|
f"签到成功: {success_count}\n"
|
|
f"已经签到: {exist_count}\n"
|
|
f"被屏蔽的: {shield_count}\n"
|
|
f"签到失败: {error_count}"
|
|
)
|
|
|
|
def main(self):
|
|
msg_all = ""
|
|
for check_item in self.check_items:
|
|
cookie = {
|
|
item.split("=")[0]: item.split("=")[1]
|
|
for item in check_item.get("cookie").split("; ")
|
|
}
|
|
session = requests.session()
|
|
session.cookies.update(cookie)
|
|
session.headers.update({"Referer": "https://www.baidu.com/"})
|
|
tbs, user_name = self.valid(session)
|
|
if tbs:
|
|
tb_name_list = self.get_tieba_list(session)
|
|
msg = f"帐号信息: {user_name}\n{self.sign(session, tb_name_list, tbs)}"
|
|
else:
|
|
msg = f"帐号信息: {user_name}\n签到状态: Cookie 可能过期"
|
|
msg_all += msg + "\n\n"
|
|
return msg_all
|
|
|
|
|
|
def string_to_dict(s):
|
|
parts = s.split('#')
|
|
result_dict = {}
|
|
result_dict['cookie'] = parts[0]
|
|
return result_dict
|
|
|
|
def start():
|
|
s = os.getenv("tieback").split('#')
|
|
print(f'共{len(s)}个账号')
|
|
for i in range(len(s)):
|
|
item =s[i]
|
|
_check_items = [string_to_dict(item)]
|
|
result = Tieba(check_items=_check_items).main()
|
|
print(result)
|
|
|
|
if __name__ == "__main__":
|
|
start()
|