# !/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()