faker2/jdCookie.py
2022-06-22 11:49:39 +08:00

64 lines
2.0 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/usr/bin/env python3
# -*- coding: utf-8 -*
'''
项目名称: JdScript / jdCookie
活动名称: 读取COOKIE / WSKEY
Author: SheYu09
'''
import requests
s = requests.session()
s.keep_alive = False
from os import environ
from inspect import stack
from USER_AGENTS import *
from re import compile, findall, split
def Name():
try:
if len(environ["Name"]):
Name = environ["Name"].split('&')
print("已获取并使用Env环境 Name:", Name, "\n")
return Name
except:
print("自行添加环境变量Name, 不同好友中间用&符号隔开\n")
exit()
def re_pin(r):
try: return compile(r'pin=(.*?);wskey=.*?;').findall(r)[0] if 'wskey=' in r else compile(r'pt_key=.*?;pt_pin=(.*?);').findall(r)[0]
except: print(r, '\nck格式不正确请检查\n')
def re_key(r, e):
try: return compile(rf'{r}').findall(e) if 'pin=' in r else compile(rf'{r}').findall(e)[0]
except: pass
def JD_API_HOST(C):
try:
if len(environ[f'JD_{C}']):
print(f" ****** 已获取并使用Env环境 {C} ******\n")
return environ[f'JD_{C}']
except:
print(f" ****** 获取Env环境 {C} 失败 ******")
print(f"自行添加环境变量JD_{C}\n")
return
def COOKIE():
global ptkeyList, ptpinList
ck = JD_API_HOST(stack()[0][3])
ptkeyList = re_key('pt_key=.*?;pt_pin=.*?;', ck)
ptpinList = re_key('pt_pin=(.*?);', ck)
def WSKEY():
global wskeyList, wspinList
ck = JD_API_HOST(stack()[0][3])
wskeyList = re_key('pin=.*?;wskey=.*?;', ck)
wspinList = re_key('pin=(.*?);', ck)
def jdCookie():
COOKIE(); WSKEY()
wskeyList and wspinList and [ptkeyList.remove(i) for i in [c for c in ptkeyList if re_pin(c) in wspinList]]
cookiesList = (ptkeyList and wskeyList and [wskeyList + ptkeyList] or ptkeyList and not wskeyList and [ptkeyList] or not ptkeyList and wskeyList and [wskeyList])[0]
print(f"====================共{len(cookiesList)}个京东账号Cookie=====================\n")
print(f"==================脚本执行- 北京时间(UTC+8){strftime('%Y-%m-%d %H:%M:%S', localtime())}==================\n")
cookiesList or exit()
return cookiesList