331 lines
13 KiB
Python
331 lines
13 KiB
Python
# Source: https://github.com/LinYuanovo/AutoTaskScripts/blob/main/miniapp/%E7%89%9B%E6%B2%B9%E8%B0%B7.py
|
||
# Raw: https://raw.githubusercontent.com/LinYuanovo/AutoTaskScripts/main/miniapp/%E7%89%9B%E6%B2%B9%E8%B0%B7.py
|
||
# Repo: LinYuanovo/AutoTaskScripts
|
||
# Path: miniapp/牛油谷.py
|
||
# UploadedAt: 2025-08-27T15:31:08+08:00
|
||
# SHA256: 44b982444bf4e7a54c443f2d50fe7db9abb2e1ef38ed16932d994949db0ebfea
|
||
# Category: 微信小程序/抓包版
|
||
# Evidence: appid/openid/session_key/encryptedData等小程序字段
|
||
|
||
"""
|
||
作者: 临渊
|
||
日期: 2025/6/30
|
||
name: 牛油谷
|
||
入口: 微信小程序 (https://s.c1ns.cn/zfe6A)
|
||
功能: 签到、查积分
|
||
变量: nyg_token = 'token' (app.niuyougu.com.cn域名下请求中的token值)
|
||
定时: 一天两次
|
||
cron: 10 8,9 * * *
|
||
------------更新日志------------
|
||
2025/6/30 V1.0 初始化脚本
|
||
2025/7/28 V1.1 修改头部注释，以便拉库
|
||
"""
|
||
|
||
MULTI_ACCOUNT_SPLIT = ["\n", "@"] # 分隔符列表
|
||
MULTI_ACCOUNT_PROXY = False # 是否使用多账号代理，默认不使用，True则使用多账号代理
|
||
NOTIFY = False # 是否推送日志，默认不推送，True则推送
|
||
|
||
import json
|
||
import random
|
||
import time
|
||
import requests
|
||
import os
|
||
import logging
|
||
import traceback
|
||
import ssl
|
||
from datetime import datetime
|
||
|
||
class TLSAdapter(requests.adapters.HTTPAdapter):
|
||
"""
|
||
自定义TLS
|
||
解决unsafe legacy renegotiation disabled
|
||
貌似python太高版本依然会报错
|
||
"""
|
||
def init_poolmanager(self, *args, **kwargs):
|
||
ctx = ssl.create_default_context()
|
||
ctx.set_ciphers("DEFAULT@SECLEVEL=1")
|
||
ctx.options |= 0x4 # <-- the key part here, OP_LEGACY_SERVER_CONNECT
|
||
kwargs["ssl_context"] = ctx
|
||
return super(TLSAdapter, self).init_poolmanager(*args, **kwargs)
|
||
|
||
class AutoTask:
|
||
def __init__(self, script_name):
|
||
"""
|
||
初始化自动任务类
|
||
:param script_name: 脚本名称，用于日志显示
|
||
"""
|
||
self.script_name = script_name
|
||
self.log_msgs = [] # 日志收集
|
||
self.proxy_url = os.getenv("PROXY_API_URL") # 代理api，返回一条txt文本，内容为代理ip:端口
|
||
self.wx_appid = "wxfa2a1ee65e3c9122" # 微信小程序id
|
||
self.wx_code_url = os.getenv("soy_codeurl_data")
|
||
self.wx_code_token = os.getenv("soy_codetoken_data")
|
||
self.host = "app.niuyougu.com.cn"
|
||
self.nickname = ""
|
||
self.token = ""
|
||
self.user_agent = "Mozilla/5.0 (Linux; Android 12; M2012K11AC Build/SKQ1.220303.001; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/134.0.6998.136 Mobile Safari/537.36 XWEB/1340129 MMWEBSDK/20240301 MMWEBID/9871 MicroMessenger/8.0.48.2580(0x28003036) WeChat/arm64 Weixin NetType/WIFI Language/zh_CN ABI/arm64 MiniProgramEnv/android"
|
||
self.setup_logging()
|
||
|
||
def setup_logging(self):
|
||
"""
|
||
配置日志系统
|
||
"""
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format='%(asctime)s - %(levelname)s\t- %(message)s',
|
||
datefmt='%Y-%m-%d %H:%M:%S',
|
||
handlers=[
|
||
# logging.FileHandler(f'{self.script_name}_{datetime.now().strftime("%Y%m%d")}.log', encoding='utf-8'), # 保存日志
|
||
logging.StreamHandler()
|
||
]
|
||
)
|
||
|
||
def log(self, msg, level="info"):
|
||
if level == "info":
|
||
logging.info(msg)
|
||
elif level == "error":
|
||
logging.error(msg)
|
||
elif level == "warning":
|
||
logging.warning(msg)
|
||
self.log_msgs.append(msg)
|
||
|
||
def get_proxy(self):
|
||
"""
|
||
获取代理
|
||
:return: 代理
|
||
"""
|
||
if not self.proxy_url:
|
||
self.log("[获取代理] 没有找到环境变量PROXY_API_URL，不使用代理", level="warning")
|
||
return None
|
||
url = self.proxy_url
|
||
response = requests.get(url)
|
||
proxy = response.text
|
||
self.log(f"[获取代理] {proxy}")
|
||
return proxy
|
||
|
||
def check_proxy(self, proxy, session):
|
||
"""
|
||
检查代理
|
||
:param proxy: 代理
|
||
:param session: session
|
||
:return: 是否可用
|
||
"""
|
||
try:
|
||
url = f"https://{self.host}/api/user/sign"
|
||
response = session.post(url, timeout=5)
|
||
if response.status_code == 200:
|
||
self.log(f"[检查代理] {proxy} 应该可用")
|
||
return True
|
||
else:
|
||
self.log(f"[检查代理] {response.text}")
|
||
return False
|
||
except Exception as e:
|
||
return False
|
||
|
||
|
||
def check_env(self):
|
||
"""
|
||
检查环境变量
|
||
:return: 环境变量字符串
|
||
"""
|
||
try:
|
||
# 从环境变量获取cookie
|
||
nyg_token = os.getenv(f"nyg_token")
|
||
if not nyg_token:
|
||
self.log("[检查环境变量] 没有找到环境变量nyg_token，请检查环境变量", level="error")
|
||
return None
|
||
|
||
# 自动检测分隔符
|
||
split_char = None
|
||
for sep in MULTI_ACCOUNT_SPLIT:
|
||
if sep in nyg_token:
|
||
split_char = sep
|
||
break
|
||
if not split_char:
|
||
# 如果都没有分隔符，默认当作单账号
|
||
nyg_tokens = [nyg_token]
|
||
else:
|
||
nyg_tokens = nyg_token.split(split_char)
|
||
|
||
for nyg_token in nyg_tokens:
|
||
if "=" in nyg_token:
|
||
nyg_token = nyg_token.split("=")[1]
|
||
yield nyg_token
|
||
else:
|
||
yield nyg_token
|
||
except Exception as e:
|
||
self.log(f"[检查环境变量] 发生错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
raise
|
||
|
||
def wx_code_auth(self, wx_id):
|
||
"""
|
||
微信授权取code
|
||
:param wx_id: 微信id
|
||
:return: 微信code
|
||
"""
|
||
try:
|
||
url = self.wx_code_url
|
||
headers = {
|
||
"Authorization": self.wx_code_token,
|
||
"Content-Type": "application/json"
|
||
}
|
||
payload = {
|
||
"wxid": wx_id,
|
||
"appid": self.wx_appid
|
||
}
|
||
response = requests.post(url, headers=headers, json=payload, timeout=5)
|
||
response.raise_for_status()
|
||
response_json = response.json()
|
||
if response_json['code'] == 200:
|
||
code = response_json['data']['code']
|
||
return code
|
||
else:
|
||
self.log(f"[微信授权]{response_json['msg']}", level="error")
|
||
return False
|
||
except requests.RequestException as e:
|
||
self.log(f"[微信授权]发生网络错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
return False
|
||
except Exception as e:
|
||
self.log(f"[微信授权]发生未知错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
return False
|
||
|
||
def wxlogin(self, session, code):
|
||
"""
|
||
登录
|
||
:param session: session
|
||
:param code: 微信code
|
||
:return: 登录结果
|
||
"""
|
||
try:
|
||
url = f"https://{self.host}/api/user/wxlogin"
|
||
payload = f"token=&code={code}&channelId=0"
|
||
response = session.post(url, data=payload, timeout=5)
|
||
response_json = response.json()
|
||
if int(response_json['code']) == 0:
|
||
self.token = response_json['data']['token']
|
||
self.nickname = response_json['data']['nickname']
|
||
self.log(f"[{self.nickname}] 登录: 成功")
|
||
return True
|
||
else:
|
||
self.log(f"[登录] 失败，错误信息: {response_json['msg']}", level="error")
|
||
return False
|
||
except requests.RequestException as e:
|
||
self.log(f"[登录] 发生网络错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
return False
|
||
except Exception as e:
|
||
self.log(f"[登录] 发生错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
return False
|
||
|
||
def sign_in(self, session):
|
||
"""
|
||
签到
|
||
:param session: session
|
||
:return: 签到结果
|
||
"""
|
||
try:
|
||
url = f"https://{self.host}/api/user/sign"
|
||
payload = f"token={self.token}&eid=1&_t={int(time.time()*1000)}"
|
||
response = session.post(url, data=payload, timeout=5)
|
||
response_json = response.json()
|
||
if int(response_json['code']) == 0:
|
||
self.log(f"[{self.nickname}] 签到: 成功")
|
||
return True
|
||
elif int(response_json['code']) == -2:
|
||
self.log(f"[{self.nickname}] 签到: 今日已签到")
|
||
return True
|
||
else:
|
||
self.log(f"[{self.nickname}] 签到: 发生错误: {response_json['msg']}", level="error")
|
||
return False
|
||
except Exception as e:
|
||
self.log(f"[{self.nickname}] 签到: 发生错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
return False
|
||
|
||
def get_user_info(self, session, log=False):
|
||
"""
|
||
获取用户信息
|
||
:param session: session
|
||
:param log: 是否打印日志 (可选)
|
||
:return: 用户信息
|
||
"""
|
||
try:
|
||
url = f"https://{self.host}/api/user/getUserInfo"
|
||
payload = f"token={self.token}&eid=1&_t={int(time.time()*1000)}"
|
||
response = session.post(url, data=payload, timeout=5)
|
||
response_json = response.json()
|
||
if int(response_json['code']) == 0:
|
||
self.nickname = response_json['data']['nickname']
|
||
if log:
|
||
self.log(f"[{self.nickname}] 积分余额: {response_json['data']['card_remaining']}")
|
||
return True
|
||
else:
|
||
self.log(f"[{self.nickname}] 获取用户信息: 发生错误: {response_json['msg']}", level="error")
|
||
return False
|
||
except Exception as e:
|
||
self.log(f"[{self.nickname}] 获取用户信息: 发生错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
return False
|
||
|
||
|
||
def run(self):
|
||
"""
|
||
运行任务
|
||
"""
|
||
try:
|
||
self.log(f"【{self.script_name}】开始执行任务")
|
||
# 检查环境变量
|
||
for index, token in enumerate(self.check_env(), 1):
|
||
self.log("")
|
||
self.log(f"------ 【账号{index}】开始执行任务 ------")
|
||
|
||
if MULTI_ACCOUNT_PROXY:
|
||
proxy = self.get_proxy()
|
||
if proxy:
|
||
session = requests.Session()
|
||
session.proxies.update({"http": f"http://{proxy}", "https": f"http://{proxy}"})
|
||
# 检查代理，不可用重新获取
|
||
while not self.check_proxy(proxy, session):
|
||
proxy = self.get_proxy()
|
||
session.proxies.update({"http": f"http://{proxy}", "https": f"http://{proxy}"})
|
||
else:
|
||
session = requests.Session()
|
||
else:
|
||
session = requests.Session()
|
||
|
||
self.token = token
|
||
headers = {
|
||
"User-Agent": self.user_agent,
|
||
"Content-Type": "application/x-www-form-urlencoded"
|
||
}
|
||
session.headers.update(headers)
|
||
|
||
# 查询用户信息
|
||
user_info = self.get_user_info(session)
|
||
if user_info:
|
||
# 签到
|
||
self.sign_in(session)
|
||
# 查询用户信息
|
||
self.get_user_info(session, log=True)
|
||
|
||
self.log(f"------ 【账号{index}】执行任务完成 ------")
|
||
except Exception as e:
|
||
self.log(f"【{self.script_name}】执行过程中发生错误: {str(e)}\n{traceback.format_exc()}", level="error")
|
||
finally:
|
||
if NOTIFY:
|
||
# 如果notify模块不存在，从远程下载至本地
|
||
if not os.path.exists("notify.py"):
|
||
url = "https://raw.githubusercontent.com/whyour/qinglong/refs/heads/develop/sample/notify.py"
|
||
response = requests.get(url)
|
||
with open("notify.py", "w", encoding="utf-8") as f:
|
||
f.write(response.text)
|
||
import notify
|
||
else:
|
||
import notify
|
||
# 任务结束后推送日志
|
||
title = f"{self.script_name} 运行日志"
|
||
header = "作者：临渊\n"
|
||
content = header + "\n" +"\n".join(self.log_msgs)
|
||
notify.send(title, content)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
auto_task = AutoTask("牛油谷")
|
||
auto_task.run() |