mirror of https://github.com/6dylan6/jdpro.git
694 lines
32 KiB
Python
694 lines
32 KiB
Python
# -*- coding: utf-8 -*
|
||
'''
|
||
new Env('wskey转换');
|
||
'''
|
||
import base64
|
||
import hashlib
|
||
import hmac
|
||
import json
|
||
import logging
|
||
import os
|
||
import random
|
||
import re
|
||
import socket
|
||
import struct
|
||
import sys
|
||
import time
|
||
import uuid
|
||
|
||
WSKEY_MODE = 0
|
||
# 0 = Default / 1 = Debug!
|
||
|
||
if "WSKEY_DEBUG" in os.environ or WSKEY_MODE: # 判断调试模式变量
|
||
logging.basicConfig(level=logging.DEBUG, format='%(message)s') # 设置日志为 Debug等级输出
|
||
logger = logging.getLogger(__name__) # 主模块
|
||
logger.debug("\nDEBUG模式开启!\n") # 消息输出
|
||
else:
|
||
logging.basicConfig(level=logging.INFO, format='%(message)s') # Info级日志
|
||
logger = logging.getLogger(__name__) # 主模块
|
||
|
||
try:
|
||
import requests # 导入HTTP模块
|
||
except Exception as e:
|
||
logger.info(str(e) + "\n缺少requests模块, 请执行命令：pip3 install requests\n") # 日志输出
|
||
sys.exit(1) # 退出脚本
|
||
os.environ['no_proxy'] = '*' # 禁用代理
|
||
requests.packages.urllib3.disable_warnings() # 抑制错误
|
||
try:
|
||
from notify import send # 导入青龙消息通知模块
|
||
except Exception as err:
|
||
logger.debug(str(err)) # 调试日志输出
|
||
logger.info("无推送文件") # 标准日志输出
|
||
|
||
ver = 40904 # 版本号
|
||
|
||
|
||
def ttotp(key):
|
||
key = base64.b32decode(key.upper() + '=' * ((8 - len(key)) % 8))
|
||
counter = struct.pack('>Q', int(time.time() / 30))
|
||
mac = hmac.new(key, counter, 'sha1').digest()
|
||
offset = mac[-1] & 0x0f
|
||
binary = struct.unpack('>L', mac[offset:offset + 4])[0] & 0x7fffffff
|
||
return str(binary)[-6:].zfill(6)
|
||
|
||
|
||
def sign_core(par):
|
||
arr = [0x37, 0x92, 0x44, 0x68, 0xA5, 0x3D, 0xCC, 0x7F, 0xBB, 0xF, 0xD9, 0x88, 0xEE, 0x9A, 0xE9, 0x5A]
|
||
key2 = b"80306f4370b39fd5630ad0529f77adb6"
|
||
arr1 = [0 for _ in range(len(par))]
|
||
for i in range(len(par)):
|
||
r0 = int(par[i])
|
||
r2 = arr[i & 0xf]
|
||
r4 = int(key2[i & 7])
|
||
r0 = r2 ^ r0
|
||
r0 = r0 ^ r4
|
||
r0 = r0 + r2
|
||
r2 = r2 ^ r0
|
||
r1 = int(key2[i & 7])
|
||
r2 = r2 ^ r1
|
||
arr1[i] = r2 & 0xff
|
||
return bytes(arr1)
|
||
|
||
def get_sign(functionId, body, uuid, client, clientVersion, st, sv):
|
||
all_arg = "functionId=%s&body=%s&uuid=%s&client=%s&clientVersion=%s&st=%s&sv=%s" % (
|
||
functionId, body, uuid, client, clientVersion, st, sv)
|
||
ret_bytes = sign_core(str.encode(all_arg))
|
||
info = hashlib.md5(base64.b64encode(ret_bytes)).hexdigest()
|
||
return info
|
||
|
||
def base64Encode(string):
|
||
string1 = "KLMNOPQRSTABCDEFGHIJUVWXYZabcdopqrstuvwxefghijklmnyz0123456789+/"
|
||
string2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
||
return base64.b64encode(string.encode("utf-8")).decode('utf-8').translate(str.maketrans(string1, string2))
|
||
|
||
def base64Decode(string):
|
||
string1 = "KLMNOPQRSTABCDEFGHIJUVWXYZabcdopqrstuvwxefghijklmnyz0123456789+/"
|
||
string2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
||
stringbase = base64.b64decode(string.translate(str.maketrans(string1, string2))).decode('utf-8')
|
||
return stringbase
|
||
|
||
def genJDUA():
|
||
st = round(time.time() * 1000)
|
||
aid = base64Encode(''.join(str(uuid.uuid4()).split('-'))[16:])
|
||
oaid = base64Encode(''.join(str(uuid.uuid4()).split('-'))[16:])
|
||
ua = 'jdapp;android;11.1.4;;;appBuild/98176;ef/1;ep/{"hdid":"JM9F1ywUPwflvMIpYPok0tt5k9kW4ArJEU3lfLhxBqw=","ts":%s,"ridx":-1,"cipher":{"sv":"CJS=","ad":"%s","od":"%s","ov":"CzO=","ud":"%s"},"ciphertype":5,"version":"1.2.0","appname":"com.jingdong.app.mall"};Mozilla/5.0 (Linux; Android 12; M2102K1C Build/SKQ1.220303.001; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/97.0.4692.98 Mobile Safari/537.36' % (st, aid, oaid, aid)
|
||
return ua
|
||
|
||
def genParams():
|
||
suid = ''.join(str(uuid.uuid4()).split('-'))[16:]
|
||
buid = base64Encode(suid)
|
||
st = round(time.time() * 1000)
|
||
sv = random.choice(["102", "111", "120"])
|
||
ep = json.dumps({
|
||
"hdid": "JM9F1ywUPwflvMIpYPok0tt5k9kW4ArJEU3lfLhxBqw=",
|
||
"ts": st,
|
||
"ridx": -1,
|
||
"cipher": {
|
||
"area": "CV8yEJUzXzU0CNG0XzK=",
|
||
"d_model": "JWunCVVidRTr",
|
||
"wifiBssid": "dW5hbw93bq==",
|
||
"osVersion": "CJS=",
|
||
"d_brand": "WQvrb21f",
|
||
"screen": "CJuyCMenCNq=",
|
||
"uuid": buid,
|
||
"aid": buid,
|
||
"openudid": buid
|
||
},
|
||
"ciphertype": 5,
|
||
"version": "1.2.0",
|
||
"appname": "com.jingdong.app.mall"
|
||
}).replace(" ", "")
|
||
body = '{"to":"https%3a%2f%2fplogin.m.jd.com%2fjd-mlogin%2fstatic%2fhtml%2fappjmp_blank.html"}'
|
||
sign = get_sign("genToken", body, suid, "android", "11.1.4", st, sv)
|
||
params = {
|
||
'functionId': 'genToken',
|
||
'clientVersion': '11.1.4',
|
||
'build': '98176',
|
||
'client': 'android',
|
||
'partner': 'google',
|
||
'oaid': suid,
|
||
'sdkVersion': '31',
|
||
'lang': 'zh_CN',
|
||
'harmonyOs': '0',
|
||
'networkType': 'UNKNOWN',
|
||
'uemps': '0-2',
|
||
'ext': '{"prstate": "0", "pvcStu": "1"}',
|
||
'eid': 'eidAcef08121fds9MoeSDdMRQ1aUTyb1TyPr2zKHk5Asiauw+K/WvS1Ben1cH6N0UnBd7lNM50XEa2kfCcA2wwThkxZc1MuCNtfU/oAMGBqadgres4BU',
|
||
'ef': '1',
|
||
'ep': ep,
|
||
'st': st,
|
||
'sign': sign,
|
||
'sv': sv
|
||
}
|
||
return params
|
||
|
||
|
||
def ql_send(text):
|
||
if "WSKEY_SEND" in os.environ and os.environ["WSKEY_SEND"] == 'disable':
|
||
return True
|
||
else:
|
||
try:
|
||
send('WSKEY转换', text) # 消息发送
|
||
except Exception as err:
|
||
logger.debug(str(err)) # Debug日志输出
|
||
logger.info("通知发送失败") # 标准日志输出
|
||
|
||
|
||
# 登录青龙 返回值 token
|
||
def get_qltoken(username, password, twoFactorSecret): # 方法 用于获取青龙 Token
|
||
logger.info("Token失效, 新登陆\n") # 日志输出
|
||
if twoFactorSecret:
|
||
try:
|
||
twoCode = ttotp(twoFactorSecret)
|
||
except Exception as err:
|
||
logger.debug(str(err)) # Debug日志输出
|
||
logger.info("TOTP异常")
|
||
sys.exit(1)
|
||
url = ql_url + "api/user/login" # 设置青龙地址 使用 format格式化自定义端口
|
||
body = {
|
||
'username': username,
|
||
'password': password
|
||
} # HTTP请求载荷
|
||
headers = {
|
||
'Accept': 'application/json',
|
||
'Content-Type': 'application/json'
|
||
} # HTTP请求头 设置为 Json格式
|
||
try:
|
||
res = requests.post(url=url, headers=headers, json=body) # 使用 requests模块进行 HTTP POST请求
|
||
if res.status_code == 200 and res.json()["code"] == 420:
|
||
url = ql_url + 'api/user/two-factor/login'
|
||
body = {
|
||
'username': username,
|
||
'password': password,
|
||
'code': twoCode
|
||
}
|
||
res = requests.put(url=url, headers=headers, json=body)
|
||
if res.status_code == 200 and res.json()["code"] == 200:
|
||
token = res.json()["data"]['token'] # 从 res.text 返回值中 取出 Token值
|
||
return token
|
||
else:
|
||
logger.info("两步校验失败\n") # 日志输出
|
||
sys.exit(1)
|
||
elif res.status_code == 200 and res.json()["code"] == 200:
|
||
token = res.json()["data"]['token'] # 从 res.text 返回值中 取出 Token值
|
||
return token
|
||
except Exception as err:
|
||
logger.debug(str(err)) # Debug日志输出
|
||
sys.exit(1)
|
||
else:
|
||
url = ql_url + 'api/user/login'
|
||
body = {
|
||
'username': username,
|
||
'password': password
|
||
} # HTTP请求载荷
|
||
headers = {
|
||
'Accept': 'application/json',
|
||
'Content-Type': 'application/json'
|
||
} # HTTP请求头 设置为 Json格式
|
||
try:
|
||
res = requests.post(url=url, headers=headers, json=body) # 使用 requests模块进行 HTTP POST请求
|
||
if res.status_code == 200 and res.json()["code"] == 200:
|
||
token = res.json()["data"]['token'] # 从 res.text 返回值中 取出 Token值
|
||
return token
|
||
else:
|
||
ql_send("青龙登录失败!")
|
||
sys.exit(1) # 脚本退出
|
||
except Exception as err:
|
||
logger.debug(str(err)) # Debug日志输出
|
||
logger.info("使用旧版青龙登录接口")
|
||
url = ql_url + 'api/login' # 设置青龙地址 使用 format格式化自定义端口
|
||
try:
|
||
res = requests.post(url=url, headers=headers, json=body) # 使用 requests模块进行 HTTP POST请求
|
||
token = json.loads(res.text)["data"]['token'] # 从 res.text 返回值中 取出 Token值
|
||
except Exception as err:
|
||
logger.debug(str(err)) # Debug日志输出
|
||
logger.info("青龙登录失败, 请检查面板状态!") # 标准日志输出
|
||
ql_send('青龙登陆失败, 请检查面板状态.')
|
||
sys.exit(1) # 脚本退出
|
||
else: # 无异常执行分支
|
||
return token # 返回 token值
|
||
# else: # 无异常执行分支
|
||
# return token # 返回 token值
|
||
|
||
|
||
# 返回值 Token
|
||
def ql_login() -> str: # 方法 青龙登录(获取Token 功能同上)
|
||
path = '/ql/config/auth.json' # 设置青龙 auth文件地址
|
||
if not os.path.isfile(path):
|
||
path = '/ql/data/config/auth.json' # 尝试设置青龙 auth 新版文件地址
|
||
if os.path.isfile(path): # 进行文件真值判断
|
||
with open(path, "r") as file: # 上下文管理
|
||
auth = file.read() # 读取文件
|
||
file.close() # 关闭文件
|
||
auth = json.loads(auth) # 使用 json模块读取
|
||
username = auth["username"] # 提取 username
|
||
password = auth["password"] # 提取 password
|
||
token = auth["token"] # 提取 authkey
|
||
try:
|
||
twoFactorSecret = auth["twoFactorSecret"]
|
||
except Exception as err:
|
||
logger.debug(str(err)) # Debug日志输出
|
||
twoFactorSecret = ''
|
||
if token == '': # 判断 Token是否为空
|
||
return get_qltoken(username, password, twoFactorSecret) # 调用方法 get_qltoken 传递 username & password
|
||
else:
|
||
url = ql_url + 'api/user' # 设置URL请求地址 使用 Format格式化端口
|
||
headers = {
|
||
'Authorization': 'Bearer {0}'.format(token),
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.71 Safari/537.36 Edg/94.0.992.38'
|
||
} # 设置用于 HTTP头
|
||
res = requests.get(url=url, headers=headers) # 调用 request模块发送 get请求
|
||
if res.status_code == 200: # 判断 HTTP返回状态码
|
||
return token # 有效 返回 token
|
||
else:
|
||
return get_qltoken(username, password, twoFactorSecret) # 调用方法 get_qltoken 传递 username & password
|
||
else:
|
||
logger.info("没有发现auth文件, 你这是青龙吗???") # 输出标准日志
|
||
sys.exit(0) # 脚本退出
|
||
|
||
|
||
# 返回值 list[wskey]
|
||
def get_wskey() -> list: # 方法 获取 wskey值 [系统变量传递]
|
||
if "JD_WSCK" in os.environ: # 判断 JD_WSCK是否存在于环境变量
|
||
wskey_list = os.environ['JD_WSCK'].split('&') # 读取系统变量 以 & 分割变量
|
||
if len(wskey_list) > 0: # 判断 WSKEY 数量 大于 0 个
|
||
return wskey_list # 返回 WSKEY [LIST]
|
||
else:
|
||
logger.info("JD_WSCK变量未启用") # 标准日志输出
|
||
sys.exit(1) # 脚本退出
|
||
else:
|
||
logger.info("未添加JD_WSCK变量") # 标准日志输出
|
||
sys.exit(0) # 脚本退出
|
||
|
||
|
||
# 返回值 list[jd_cookie]
|
||
def get_ck() -> list: # 方法 获取 JD_COOKIE值 [系统变量传递] <! 此方法未使用 !>
|
||
if "JD_COOKIE" in os.environ: # 判断 JD_COOKIE是否存在于环境变量
|
||
ck_list = os.environ['JD_COOKIE'].split('&') # 读取系统变量 以 & 分割变量
|
||
if len(ck_list) > 0: # 判断 WSKEY 数量 大于 0 个
|
||
return ck_list # 返回 JD_COOKIE [LIST]
|
||
else:
|
||
logger.info("JD_COOKIE变量未启用") # 标准日志输出
|
||
sys.exit(1) # 脚本退出
|
||
else:
|
||
logger.info("未添加JD_COOKIE变量") # 标准日志输出
|
||
sys.exit(0) # 脚本退出
|
||
|
||
|
||
# 返回值 bool
|
||
def check_ck(ck) -> bool: # 方法 检查 Cookie有效性 使用变量传递 单次调用
|
||
searchObj = re.search(r'pt_pin=([^;\s]+)', ck, re.M | re.I) # 正则检索 pt_pin
|
||
if searchObj: # 真值判断
|
||
pin = searchObj.group(1) # 取值
|
||
else:
|
||
pin = ck.split(";")[1] # 取值 使用 ; 分割
|
||
if "WSKEY_UPDATE_HOUR" in os.environ: # 判断 WSKEY_UPDATE_HOUR是否存在于环境变量
|
||
updateHour = 23 # 更新间隔23小时
|
||
if os.environ["WSKEY_UPDATE_HOUR"].isdigit(): # 检查是否为 DEC值
|
||
updateHour = int(os.environ["WSKEY_UPDATE_HOUR"]) # 使用 int化数字
|
||
nowTime = time.time() # 获取时间戳 赋值
|
||
updatedAt = 0.0 # 赋值
|
||
searchObj = re.search(r'__time=([^;\s]+)', ck, re.M | re.I) # 正则检索 [__time=]
|
||
if searchObj: # 真值判断
|
||
updatedAt = float(searchObj.group(1)) # 取值 [float]类型
|
||
if nowTime - updatedAt >= (updateHour * 60 * 60) - (10 * 60): # 判断时间操作
|
||
logger.info(str(pin) + ";即将到期或已过期\n") # 标准日志输出
|
||
return False # 返回 Bool类型 False
|
||
else:
|
||
remainingTime = (updateHour * 60 * 60) - (nowTime - updatedAt) # 时间运算操作
|
||
hour = int(remainingTime / 60 / 60) # 时间运算操作 [int]
|
||
minute = int((remainingTime % 3600) / 60) # 时间运算操作 [int]
|
||
logger.info(str(pin) + ";未到期，{0}时{1}分后更新\n".format(hour, minute)) # 标准日志输出
|
||
return True # 返回 Bool类型 True
|
||
elif "WSKEY_DISCHECK" in os.environ:
|
||
logger.info("不检查账号有效性\n--------------------\n") # 标准日志输出
|
||
return False # 返回 Bool类型 False
|
||
else:
|
||
url = 'https://me-api.jd.com/user_new/info/GetJDUserInfoUnion' # 设置JD_API接口地址
|
||
headers = {
|
||
'Cookie': ck,
|
||
'Referer': 'https://home.m.jd.com/myJd/home.action',
|
||
'user-agent': genJDUA()
|
||
} # 设置 HTTP头
|
||
try:
|
||
res = requests.get(url=url, headers=headers, verify=False, timeout=10,
|
||
allow_redirects=False) # 进行 HTTP请求[GET] 超时 10秒
|
||
except Exception as err:
|
||
logger.debug(str(err)) # 调试日志输出
|
||
logger.info("JD接口错误 请重试或者更换IP") # 标准日志输出
|
||
return False # 返回 Bool类型 False
|
||
else:
|
||
if res.status_code == 200: # 判断 JD_API 接口是否为 200 [HTTP_OK]
|
||
try:
|
||
code = int(json.loads(res.text)['retcode']) # 使用 Json模块对返回数据取值 int([retcode])
|
||
except Exception as err:
|
||
logger.debug(str(err))
|
||
logger.info("JD接口风控, 建议更换IP或增加间隔时间")
|
||
return False
|
||
if code == 0: # 判断 code值
|
||
logger.info(str(pin) + ";状态正常\n") # 标准日志输出
|
||
return True # 返回 Bool类型 True
|
||
else:
|
||
logger.info(str(pin) + ";状态失效\n")
|
||
return False # 返回 Bool类型 False
|
||
else:
|
||
logger.info("JD接口错误码: " + str(res.status_code)) # 标注日志输出
|
||
return False # 返回 Bool类型 False
|
||
|
||
|
||
# 返回值 bool jd_ck
|
||
def getToken(wskey): # 方法 获取 Wskey转换使用的 Token 由 JD_API 返回 这里传递 wskey
|
||
# try:
|
||
# url = str(base64.b64decode(url_t).decode()) + 'api/genToken' # 设置云端服务器地址 路由为 genToken
|
||
# header = {"User-Agent": ua} # 设置 HTTP头
|
||
# params = requests.get(url=url, headers=header, verify=False, timeout=20).json() # 设置 HTTP请求参数 超时 20秒 Json解析
|
||
# except Exception as err:
|
||
# logger.info("Params参数获取失败") # 标准日志输出
|
||
# logger.debug(str(err)) # 调试日志输出
|
||
# # return False, wskey # 返回 -> False[Bool], Wskey
|
||
# return False # 返回 -> False[Bool], Wskey
|
||
params = genParams()
|
||
headers = {
|
||
'cookie': wskey,
|
||
'content-type': 'application/x-www-form-urlencoded; charset=UTF-8',
|
||
'charset': 'UTF-8',
|
||
'accept-encoding': 'br,gzip,deflate',
|
||
'user-agent': genJDUA()
|
||
} # 设置 HTTP头
|
||
url = 'https://api.m.jd.com/client.action' # 设置 URL地址
|
||
data = 'body=%7B%22to%22%3A%22https%253a%252f%252fplogin.m.jd.com%252fjd-mlogin%252fstatic%252fhtml%252fappjmp_blank.html%22%7D&' # 设置 POST 载荷
|
||
try:
|
||
res = requests.post(url=url, params=params, headers=headers, data=data, verify=False,
|
||
timeout=10) # HTTP请求 [POST] 超时 10秒
|
||
res_json = json.loads(res.text) # Json模块 取值
|
||
tokenKey = res_json['tokenKey'] # 取出TokenKey
|
||
except Exception as err:
|
||
logger.info("JD_WSKEY接口抛出错误 尝试重试 更换IP") # 标准日志输出
|
||
logger.info(str(err)) # 标注日志输出
|
||
# return False, wskey # 返回 -> False[Bool], Wskey
|
||
return False # 返回 -> False[Bool], Wskey
|
||
else:
|
||
return appjmp(wskey, tokenKey) # 传递 wskey, Tokenkey 执行方法 [appjmp]
|
||
|
||
|
||
# 返回值 bool jd_ck
|
||
def appjmp(wskey, tokenKey): # 方法 传递 wskey & tokenKey
|
||
wskey = "pt_" + str(wskey.split(";")[0]) # 变量组合 使用 ; 分割变量 拼接 pt_
|
||
if tokenKey == 'xxx': # 判断 tokenKey返回值
|
||
logger.info(str(wskey) + ";疑似IP风控等问题 默认为失效\n--------------------\n") # 标准日志输出
|
||
# return False, wskey # 返回 -> False[Bool], Wskey
|
||
return False # 返回 -> False[Bool], Wskey
|
||
headers = {
|
||
'User-Agent': genJDUA(),
|
||
'accept': 'accept:text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
|
||
'x-requested-with': 'com.jingdong.app.mall'
|
||
} # 设置 HTTP头
|
||
params = {
|
||
'tokenKey': tokenKey,
|
||
'to': 'https://plogin.m.jd.com/jd-mlogin/static/html/appjmp_blank.html'
|
||
} # 设置 HTTP_URL 参数
|
||
url = 'https://un.m.jd.com/cgi-bin/app/appjmp' # 设置 URL地址
|
||
try:
|
||
res = requests.get(url=url, headers=headers, params=params, verify=False, allow_redirects=False,
|
||
timeout=20) # HTTP请求 [GET] 阻止跳转 超时 20秒
|
||
except Exception as err:
|
||
logger.info("JD_appjmp 接口错误 请重试或者更换IP\n") # 标准日志输出
|
||
logger.info(str(err)) # 标准日志输出
|
||
# return False, wskey # 返回 -> False[Bool], Wskey
|
||
return False # 返回 -> False[Bool], Wskey
|
||
else:
|
||
try:
|
||
res_set = res.cookies.get_dict() # 从res cookie取出
|
||
pt_key = 'pt_key=' + res_set['pt_key'] # 取值 [pt_key]
|
||
pt_pin = 'pt_pin=' + res_set['pt_pin'] # 取值 [pt_pin]
|
||
# if "WSKEY_UPDATE_HOUR" in os.environ: # 判断是否在系统变量中启用 WSKEY_UPDATE_HOUR
|
||
if WSKEY_UPDATE_BOOL:
|
||
jd_ck = str(pt_key) + ';' + str(pt_pin) + ';__time=' + str(time.time()) + ';' # 拼接变量
|
||
else:
|
||
jd_ck = str(pt_key) + ';' + str(pt_pin) + ';' # 拼接变量
|
||
except Exception as err:
|
||
logger.info("JD_appjmp提取Cookie错误 请重试或者更换IP\n") # 标准日志输出
|
||
logger.info(str(err)) # 标准日志输出
|
||
# return False, wskey # 返回 -> False[Bool], Wskey
|
||
return False # 返回 -> False[Bool], Wskey
|
||
else:
|
||
if 'fake' in pt_key: # 判断 pt_key中 是否存在fake
|
||
logger.info(str(wskey) + ";WsKey状态失效\n") # 标准日志输出
|
||
# return False, wskey # 返回 -> False[Bool], Wskey
|
||
return False # 返回 -> False[Bool], Wskey
|
||
else:
|
||
logger.info(str(wskey) + ";WsKey状态正常\n") # 标准日志输出
|
||
# return True, jd_ck # 返回 -> True[Bool], jd_ck
|
||
return jd_ck
|
||
|
||
|
||
def update(): # 方法 脚本更新模块
|
||
up_ver = int(cloud_arg['update']) # 云端参数取值 [int]
|
||
if ver >= up_ver: # 判断版本号大小
|
||
logger.info("当前脚本版本: " + str(ver)) # 标准日志输出
|
||
logger.info("--------------------\n") # 标准日志输出
|
||
else:
|
||
logger.info("当前脚本版本: " + str(ver) + "新版本: " + str(up_ver)) # 标准日志输出
|
||
logger.info("存在新版本, 请更新脚本后执行") # 标准日志输出
|
||
logger.info("--------------------\n") # 标准日志输出
|
||
text = '当前脚本版本: {0}新版本: {1}, 请更新脚本~!'.format(ver, up_ver) # 设置发送内容
|
||
ql_send(text)
|
||
|
||
|
||
def ql_api(method, api, body=None) -> dict:
|
||
headers = {
|
||
"Content-Type": "application/json",
|
||
"Authorization": f"Bearer {token}"
|
||
}
|
||
url = ql_url + api
|
||
for retry_count in range(3):
|
||
try:
|
||
if type(body) == dict:
|
||
res = ql_session.request(method, url=url, headers=headers, json=body).json()
|
||
else:
|
||
res = ql_session.request(method, url=url, headers=headers, data=body).json()
|
||
except Exception as err:
|
||
logger.debug(str(err))
|
||
logger.info(f"\n青龙{api}接口错误，重试次数：{retry_count + 1}")
|
||
continue
|
||
else:
|
||
return res
|
||
logger.info(f"\n青龙{api}接口多次重试仍然失败")
|
||
sys.exit(1)
|
||
|
||
|
||
def ql_check(port) -> bool: # 方法 检查青龙端口
|
||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # Socket模块初始化
|
||
sock.settimeout(2) # 设置端口超时
|
||
try:
|
||
sock.connect(('127.0.0.1', port)) # 请求端口
|
||
except Exception as err: # 捕捉异常
|
||
logger.debug(str(err)) # 调试日志输出
|
||
sock.close() # 端口关闭
|
||
return False # 返回 -> False[Bool]
|
||
else: # 分支判断
|
||
sock.close() # 关闭端口
|
||
return True # 返回 -> True[Bool]
|
||
|
||
|
||
def serch_ck(pin): # 方法 搜索 Pin
|
||
for i in range(len(envlist)): # For循环 变量[envlist]的数量
|
||
if "name" not in envlist[i] or envlist[i]["name"] != "JD_COOKIE": # 判断 envlist内容
|
||
continue # 继续循环
|
||
if pin in envlist[i]['value']: # 判断envlist取值['value']
|
||
value = envlist[i]['value'] # 取值['value']
|
||
id = envlist[i][ql_id] # 取值 [ql_id](变量)
|
||
logger.info(str(pin) + "检索成功\n") # 标准日志输出
|
||
# return True, value, id # 返回 -> True[Bool], value, id
|
||
return value, id # 返回 -> value, id
|
||
else:
|
||
continue # 继续循环
|
||
logger.info(str(pin) + "检索失败\n") # 标准日志输出
|
||
return False # 返回 -> False[Bool], 1
|
||
|
||
|
||
def get_env(): # 方法 读取变量
|
||
api = 'api/envs'
|
||
res = ql_api("GET", api)
|
||
data = res['data']
|
||
return data
|
||
|
||
|
||
def check_id() -> str: # 方法 兼容青龙老版本与新版本 id & _id的问题
|
||
api = 'api/envs'
|
||
res = ql_api("GET", api)
|
||
if '_id' in res['data'][0]: # 判断 [_id]
|
||
logger.info("使用 _id 键值") # 标准日志输出
|
||
return '_id' # 返回 -> '_id'
|
||
else:
|
||
logger.info("使用 id 键值") # 标准日志输出
|
||
return 'id' # 返回 -> 'id'
|
||
|
||
|
||
def ql_update(eid, newck): # 方法 青龙更新变量 传递 id cookie
|
||
api = 'api/envs'
|
||
body = {
|
||
'name': 'JD_COOKIE',
|
||
'value': newck,
|
||
ql_id: eid
|
||
}
|
||
ql_api("PUT", api, body)
|
||
ql_enable(eid)
|
||
|
||
|
||
def ql_enable(eid): # 方法 青龙变量启用 传递值 eid
|
||
api = 'api/envs/enable'
|
||
body = f'[{eid}]'
|
||
res = ql_api("PUT", api, body)
|
||
if res['code'] == 200: # 判断返回值为 200
|
||
logger.info("\n账号启用\n--------------------\n") # 标准日志输出
|
||
return True
|
||
else:
|
||
logger.info("\n账号启用失败\n--------------------\n") # 标准日志输出
|
||
return False
|
||
|
||
|
||
def ql_disable(eid): # 方法 青龙变量禁用 传递 eid
|
||
api = 'api/envs/disable'
|
||
body = f'[{eid}]'
|
||
res = ql_api("PUT", api, body)
|
||
if res['code'] == 200: # 判断返回值为 200
|
||
logger.info("\n账号禁用成功\n--------------------\n") # 标准日志输出
|
||
else:
|
||
logger.info("\n账号禁用失败\n--------------------\n") # 标准日志输出
|
||
|
||
|
||
def ql_insert(i_ck): # 方法 插入新变量
|
||
api = 'api/envs'
|
||
body = json.dumps([{"value": i_ck, "name": "JD_COOKIE"}])
|
||
res = ql_api("POST", api, body)
|
||
if res['code'] == 200: # 判断返回值为 200
|
||
logger.info("\n账号添加完成\n--------------------\n") # 标准日志输出
|
||
else:
|
||
logger.info("\n账号添加失败\n--------------------\n") # 标准日志输出
|
||
|
||
|
||
# def cloud_info(): # 方法 云端信息
|
||
# url = str(base64.b64decode(url_t).decode()) + 'api/check_api' # 设置 URL地址 路由 [check_api]
|
||
# for i in range(3): # For循环 3次
|
||
# try:
|
||
# headers = {"authorization": "Bearer Shizuku"} # 设置 HTTP头
|
||
# res = requests.get(url=url, verify=False, headers=headers, timeout=20).text # HTTP[GET] 请求 超时 20秒
|
||
# except requests.exceptions.ConnectTimeout:
|
||
# logger.info("\n获取云端参数超时, 正在重试!" + str(i)) # 标准日志输出
|
||
# time.sleep(1) # 休眠 1秒
|
||
# continue # 循环继续
|
||
# except requests.exceptions.ReadTimeout:
|
||
# logger.info("\n获取云端参数超时, 正在重试!" + str(i)) # 标准日志输出
|
||
# time.sleep(1) # 休眠 1秒
|
||
# continue # 循环继续
|
||
# except Exception as err:
|
||
# logger.info("\n未知错误云端, 退出脚本!") # 标准日志输出
|
||
# logger.debug(str(err)) # 调试日志输出
|
||
# sys.exit(1) # 脚本退出
|
||
# else:
|
||
# try:
|
||
# c_info = json.loads(res) # json读取参数
|
||
# except Exception as err:
|
||
# logger.info("云端参数解析失败") # 标准日志输出
|
||
# logger.debug(str(err)) # 调试日志输出
|
||
# sys.exit(1) # 脚本退出
|
||
# else:
|
||
# return c_info # 返回 -> c_info
|
||
|
||
|
||
# def check_cloud():
|
||
# url_list = ['aHR0cHM6Ly9hcGkubW9tb2UubGluay8=', 'aHR0cHM6Ly9hcGkubGltb2UuZXUub3JnLw==',
|
||
# 'aHR0cHM6Ly9hcGkuaWxpeWEuY2Yv']
|
||
# for i in url_list:
|
||
# url = str(base64.b64decode(i).decode()) # 设置 url地址 [str]
|
||
# try:
|
||
# requests.get(url=url, verify=False, timeout=10) # HTTP[GET]请求 超时 10秒
|
||
# except Exception as err:
|
||
# logger.debug(str(err)) # 调试日志输出
|
||
# continue # 循环继续
|
||
# else: # 分支判断
|
||
# info = ['HTTPS', 'Eu_HTTPS', 'CloudFlare'] # 输出信息[List]
|
||
# logger.info(str(info[url_list.index(i)]) + " Server Check OK\n--------------------\n") # 标准日志输出
|
||
# return i # 返回 ->i
|
||
# logger.info("\n云端地址全部失效, 请检查网络!") # 标准日志输出
|
||
# ql_send('云端地址失效. 请联系作者或者检查网络.') # 推送消息
|
||
# sys.exit(1) # 脚本退出
|
||
|
||
|
||
def check_port(): # 方法 检查变量传递端口
|
||
logger.info("\n--------------------\n") # 标准日志输出
|
||
port = int(os.environ.get("QL_PORT", "5700") if str(os.environ.get("QL_PORT")).isdigit() else "5700")
|
||
if ql_check(port): # 调用方法 [ql_check] 传递 [port]
|
||
logger.info(str(port) + "端口检查通过") # 标准日志输出
|
||
return port # 返回->port
|
||
else:
|
||
logger.info(
|
||
str(port) + "端口检查失败, 如果改过端口, 请在变量中声明端口 \n在config.sh中加入 export QL_PORT=\"端口号\"") # 标准日志输出
|
||
logger.info("\n如果你很确定端口没错, 还是无法执行, 在GitHub给我发issus\n--------------------\n") # 标准日志输出
|
||
sys.exit(1) # 脚本退出
|
||
|
||
|
||
if __name__ == '__main__': # Python主函数执行入口
|
||
port = check_port() # 调用方法 [check_port] 并赋值 [port]
|
||
ql_url = f'http://127.0.0.1:{port}/'
|
||
ql_session = requests.session()
|
||
token = ql_login() # 调用方法 [ql_login] 并赋值 [token]
|
||
ql_id = check_id()
|
||
# url_t = check_cloud()
|
||
# cloud_arg = cloud_info()
|
||
# update()
|
||
# ua = cloud_arg['User-Agent']
|
||
wslist = get_wskey()
|
||
envlist = get_env()
|
||
sleepTime = int(os.environ.get("WSKEY_SLEEP", "10") if str(os.environ.get("WSKEY_SLEEP")).isdigit() else "10")
|
||
tryCount = int(os.environ.get("WSKEY_TRY_COUNT", "1") if str(os.environ.get("WSKEY_TRY_COUNT")).isdigit() else "1")
|
||
WSKEY_UPDATE_BOOL = bool(os.environ.get("WSKEY_UPDATE_HOUR"))
|
||
WSKEY_AUTO_DISABLE = bool(os.environ.get("WSKEY_AUTO_DISABLE"))
|
||
for ws in wslist: # wslist变量 for循环 [wslist -> ws]
|
||
wspin = ws.split(";")[0] # 变量分割 ;
|
||
if "pin" in wspin: # 判断 pin 是否存在于 [wspin]
|
||
wspin = "pt_" + wspin + ";" # 封闭变量
|
||
return_serch = serch_ck(wspin) # 变量 pt_pin 搜索获取 key eid
|
||
if return_serch: # bool: True 搜索到账号
|
||
jck, eid = return_serch # 拿到 JD_COOKIE
|
||
if not check_ck(jck): # bool: False 判定 JD_COOKIE 有效性
|
||
for count in range(tryCount): # for循环 [tryCount]
|
||
count += 1 # 自增
|
||
return_ws = getToken(ws) # 使用 WSKEY 请求获取 JD_COOKIE bool jd_ck
|
||
if return_ws:
|
||
break # 中断循环
|
||
if count < tryCount: # 判断循环次
|
||
logger.info("{0} 秒后重试，剩余次数：{1}\n".format(sleepTime, tryCount - count)) # 标准日志输出
|
||
time.sleep(sleepTime) # 脚本休眠 使用变量 [sleepTime]
|
||
if return_ws: # 判断 [return_ws]返回值 Bool类型
|
||
# logger.info("wskey转pt_key成功", nt_key) # 标准日志输出 [未启用]
|
||
logger.info("wskey转换成功") # 标准日志输出
|
||
ql_update(eid, return_ws) # 函数 ql_update 参数 eid JD_COOKIE
|
||
else:
|
||
if WSKEY_AUTO_DISABLE:
|
||
logger.info(str(wspin) + "账号失效") # 标准日志输出
|
||
text = f"账号: {wspin} WsKey疑似失效" # 设置推送内容
|
||
else:
|
||
logger.info(str(wspin) + "账号禁用") # 标准日志输出
|
||
ql_disable(eid) # 执行方法[ql_disable] 传递 eid
|
||
text = f"账号: {wspin} WsKey疑似失效, 已禁用Cookie"
|
||
ql_send(text)
|
||
else:
|
||
logger.info(str(wspin) + "账号有效") # 标准日志输出
|
||
ql_enable(eid) # 执行方法[ql_enable] 传递 eid
|
||
logger.info("--------------------\n") # 标准日志输出
|
||
else:
|
||
logger.info("\n新wskey\n") # 标准日志分支
|
||
return_ws = getToken(ws) # 使用 WSKEY 请求获取 JD_COOKIE bool jd_ck
|
||
if return_ws: # 判断 (return_ws[0]) 类型: [Bool]
|
||
logger.info("wskey转换成功\n") # 标准日志输出
|
||
ql_insert(return_ws) # 调用方法 [ql_insert]
|
||
logger.info(f"暂停{sleepTime}秒\n") # 标准日志输出
|
||
time.sleep(sleepTime) # 脚本休眠
|
||
else:
|
||
logger.info("WSKEY格式错误\n--------------------\n") # 标准日志输出
|
||
logger.info("执行完成\n--------------------") # 标准日志输出
|
||
sys.exit(0) # 脚本退出
|
||
# Enjoy
|