mirror of https://github.com/6dylan6/jdpro.git
264 lines
9.7 KiB
Python
264 lines
9.7 KiB
Python
# !/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
# Modify : 2022/9/30
|
||
# 京豆近7天输出表格统计
|
||
# 用不着每天跑,定时自行设置吧，配合desi可指定账号
|
||
# https://raw.githubusercontent.com/6dylan6/jdpro/main/jd_beans_7days.py
|
||
'''
|
||
new Env('豆子7天统计');
|
||
8 8 30 9 * jd_beans_7days.py
|
||
'''
|
||
|
||
import requests
|
||
import datetime
|
||
import os,re,sys,json,time
|
||
from urllib.parse import unquote
|
||
from datetime import timedelta
|
||
from datetime import timezone
|
||
|
||
try:
|
||
from prettytable import PrettyTable
|
||
except:
|
||
os.system('pip3 install prettytable &> /dev/null')
|
||
from prettytable import PrettyTable
|
||
|
||
SHA_TZ = timezone(
|
||
timedelta(hours=8),
|
||
name='Asia/Shanghai',
|
||
)
|
||
requests.adapters.DEFAULT_RETRIES = 5
|
||
session = requests.session()
|
||
session.keep_alive = False
|
||
|
||
url = "https://api.m.jd.com/api"
|
||
|
||
def gen_body(page):
|
||
body = {
|
||
"beginDate": datetime.datetime.utcnow().replace(tzinfo=timezone.utc).astimezone(SHA_TZ).strftime("%Y-%m-%d %H:%M:%S"),
|
||
"endDate": datetime.datetime.utcnow().replace(tzinfo=timezone.utc).astimezone(SHA_TZ).strftime("%Y-%m-%d %H:%M:%S"),
|
||
"pageNo": page,
|
||
"pageSize": 20,
|
||
}
|
||
return body
|
||
|
||
def printf(text):
|
||
print(text)
|
||
sys.stdout.flush()
|
||
|
||
def column_pad(*columns):
|
||
max_len = max([len(x) for x in columns])
|
||
for y in columns:
|
||
y.extend(['NaN']*(max_len-len(y)))
|
||
|
||
class getJDCookie(object):
|
||
|
||
# 获取cookie
|
||
def getCookie(self):
|
||
global cookies
|
||
cookies = []
|
||
try:
|
||
if "JD_COOKIE" in os.environ:
|
||
if len(os.environ["JD_COOKIE"]) > 10:
|
||
cookies = os.environ["JD_COOKIE"]
|
||
printf("\n当前从环境变量获取CK\n")
|
||
return
|
||
except Exception as e:
|
||
printf(f"【getCookie Error】{e}")
|
||
|
||
# 检测cookie格式是否正确
|
||
|
||
def getUserInfo(self, ck, pinName, userNum):
|
||
url = 'https://me-api.jd.com/user_new/info/GetJDUserInfoUnion?orgFlag=JD_PinGou_New&callSource=mainorder&channel=4&isHomewhite=0&sceneval=2&sceneval=2&callback='
|
||
headers = {
|
||
'Cookie': ck,
|
||
'Accept': '*/*',
|
||
'Connection': 'close',
|
||
'Referer': 'https://home.m.jd.com/myJd/home.action',
|
||
'Accept-Encoding': 'gzip, deflate, br',
|
||
'Host': 'me-api.jd.com',
|
||
'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 14_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0.2 Mobile/15E148 Safari/604.1',
|
||
'Accept-Language': 'zh-cn'
|
||
}
|
||
try:
|
||
if sys.platform == 'ios':
|
||
resp = requests.get(url=url, verify=False, headers=headers, timeout=60).json()
|
||
else:
|
||
resp = requests.get(url=url, headers=headers, timeout=60).json()
|
||
if resp['retcode'] == "0":
|
||
nickname = resp['data']['userInfo']['baseInfo']['nickname']
|
||
if not nickname:
|
||
nickname = resp['data']['userInfo']['baseInfo']['curPin']
|
||
return ck, nickname
|
||
else:
|
||
context = f"账号{userNum}【{pinName}】Cookie 已失效！请重新获取\n。"
|
||
printf(context)
|
||
return ck, False
|
||
except Exception:
|
||
context = f"账号{userNum}【{pinName}】Cookie 已失效！请重新获取\n。"
|
||
printf(context)
|
||
return ck, False
|
||
|
||
def iscookie(self):
|
||
"""
|
||
:return: cookiesList,userNameList,pinNameList
|
||
"""
|
||
cookiesList = []
|
||
userNameList = []
|
||
pinNameList = []
|
||
if 'pt_key=' in cookies and 'pt_pin=' in cookies:
|
||
r = re.compile(r"pt_key=.*?pt_pin=.*?;", re.M | re.S | re.I)
|
||
result = r.findall(cookies)
|
||
if len(result) >= 1:
|
||
printf("您有{}个账号".format(len(result)))
|
||
u = 1
|
||
for i in result:
|
||
r = re.compile(r"pt_pin=(.*?);")
|
||
pinName = r.findall(i)
|
||
pinName = unquote(pinName[0])
|
||
# 获取账号名
|
||
ck, nickname = self.getUserInfo(i, pinName, u)
|
||
if nickname:
|
||
cookiesList.append(ck)
|
||
userNameList.append(nickname)
|
||
pinNameList.append(pinName)
|
||
else:
|
||
u += 1
|
||
continue
|
||
u += 1
|
||
if len(cookiesList) > 0 and len(userNameList) > 0:
|
||
return cookiesList, userNameList, pinNameList
|
||
else:
|
||
printf("没有可用CK，已退出\n")
|
||
exit(3)
|
||
else:
|
||
printf("CK格式错误！...本次运行退出\n")
|
||
exit(4)
|
||
else:
|
||
printf("CK格式错误或无CK！...请检查\n")
|
||
exit(4)
|
||
|
||
|
||
getCk = getJDCookie()
|
||
getCk.getCookie()
|
||
|
||
|
||
def gen_params(page):
|
||
body = gen_body(page)
|
||
params = {
|
||
"functionId": "jposTradeQuery",
|
||
"appid": "swat_miniprogram",
|
||
"client": "tjj_m",
|
||
"sdkName": "orderDetail",
|
||
"sdkVersion": "1.0.0",
|
||
"clientVersion": "3.1.3",
|
||
"timestamp": int(round(time.time() * 1000)),
|
||
"body": json.dumps(body)
|
||
}
|
||
return params
|
||
|
||
def creat_bean_count(date, beansin, beansout, beanstotal):
|
||
tb = PrettyTable()
|
||
tb.add_column('DATE', date)
|
||
tb.add_column('BEANSIN', beansin)
|
||
tb.add_column('BEANSOUT', beansout)
|
||
tb.add_column('TOTAL', beanstotal)
|
||
printf(tb)
|
||
|
||
|
||
def get_beans_7days(ck):
|
||
try:
|
||
day_7 = True
|
||
page = 0
|
||
headers = {
|
||
"Host": "api.m.jd.com",
|
||
"User-Agent": "Mozilla/5.0 (Linux; Android 10; MI 9 Build/QKQ1.190825.002; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/78.0.3904.62 XWEB/2797 MMWEBSDK/201201 Mobile Safari/537.36 MMWEBID/7986 MicroMessenger/8.0.1840(0x2800003B) Process/appbrand4 WeChat/arm64 Weixin NetType/4G Language/zh_CN ABI/arm64 MiniProgramEnv/android",
|
||
"Content-Type": "application/x-www-form-urlencoded;",
|
||
"Cookie": ck,
|
||
}
|
||
days = []
|
||
for i in range(0, 7):
|
||
days.append((datetime.date.today() - datetime.timedelta(days=i)).strftime("%Y-%m-%d"))
|
||
beans_in = {key: 0 for key in days}
|
||
beans_out = {key: 0 for key in days}
|
||
while day_7:
|
||
page = page + 1
|
||
url="https://api.m.jd.com/client.action?functionId=getJingBeanBalanceDetail&body=%7B%22pageSize%22%3A%2220%22%2C%22page%22%3A%22"+str(page)+"%22%7D&appid=ld"
|
||
resp = session.get(url, headers=headers, timeout=1000).text
|
||
res = json.loads(resp)
|
||
if res['code'] == '0' :
|
||
for i in res['detailList']:
|
||
for date in days:
|
||
if str(date) in i['date'] and int(i['amount']) > 0:
|
||
beans_in[str(date)] = beans_in[str(date)] + int(i['amount'])
|
||
break
|
||
elif str(date) in i['date'] and int(i['amount']) < 0:
|
||
beans_out[str(date)] = beans_out[str(date)] + int(i['amount'])
|
||
break
|
||
if i['date'].split(' ')[0] not in str(days):
|
||
day_7 = False
|
||
else:
|
||
print("未获取到数据，原因未知！！\n")
|
||
return {'code': 400, 'data': res}
|
||
return {'code': 200, 'data': [beans_in, beans_out, days]}
|
||
except Exception as e:
|
||
print(str(e))
|
||
return {'code': 400, 'data': str(e)}
|
||
|
||
|
||
def get_total_beans(ck):
|
||
try:
|
||
headers = {
|
||
"Host": "wq.jd.com",
|
||
"User-Agent": "jdapp;iPhone;9.4.4;14.3;network/4g;Mozilla/5.0 (iPhone; CPU iPhone OS 14_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148;supportJDSHWK/1",
|
||
"Content-Type": "application/x-www-form-urlencoded;",
|
||
"Cookie": ck,
|
||
"Referer": 'https://wqs.jd.com/my/jingdou/my.shtml?sceneval=2'
|
||
}
|
||
jurl = "https://wq.jd.com/user/info/QueryJDUserInfo?sceneval=2"
|
||
resp = requests.post(jurl, headers=headers).text
|
||
res = json.loads(resp)
|
||
return res['base']['jdNum']
|
||
except Exception as e:
|
||
printf(str(e))
|
||
|
||
|
||
def get_bean_data(i,ck):
|
||
try:
|
||
if ck:
|
||
#ck = cookies[i-1]
|
||
beans_res = get_beans_7days(ck)
|
||
beantotal = get_total_beans(ck)
|
||
if beans_res['code'] != 200:
|
||
return beans_res
|
||
else:
|
||
beans_in, beans_out = [], []
|
||
beanstotal = [int(beantotal), ]
|
||
for i in beans_res['data'][0]:
|
||
beantotal = int(beantotal) - int(beans_res['data'][0][i]) - int(beans_res['data'][1][i])
|
||
beans_in.append(int(beans_res['data'][0][i]))
|
||
beans_out.append(int(str(beans_res['data'][1][i]).replace('-', '')))
|
||
beanstotal.append(beantotal)
|
||
return {'code': 200, 'data': [beans_in[::-1], beans_out[::-1], beanstotal[::-1], beans_res['data'][2][::-1]]}
|
||
except Exception as e:
|
||
print(str(e))
|
||
|
||
def query():
|
||
try:
|
||
global cookiesList, userNameList, pinNameList, ckNum, beanCount, userCount
|
||
cookiesList, userNameList, pinNameList = getCk.iscookie()
|
||
for i,ck,user,pin in zip(range(1,len(cookiesList)+1),cookiesList,userNameList,pinNameList):
|
||
printf(f"\n****** [账号{i}]-{user} ******")
|
||
res=get_bean_data(i,ck)
|
||
if res['code'] != 200:
|
||
printf(res['data'])
|
||
continue
|
||
if res['data'][2][1:] != []:
|
||
creat_bean_count(res['data'][3], res['data'][0], res['data'][1], res['data'][2][1:])
|
||
time.sleep(2)
|
||
except Exception as e:
|
||
printf(str(e))
|
||
|
||
|
||
if __name__ == "__main__":
|
||
query()
|