上传文件至 'dgp/tests'

dgp/tests/get_order_dairly_wending.py

@@ -0,0 +1,263 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# 20201104
+## 20201106新添加 文鼎 平台接口
+########################## 文鼎订单接口，返回数据格式
+'''
+{
+  "code": 200,
+  "data": {
+    "totalPage": 1,
+    "rechargeList": [
+      {
+        "userId": 1067,
+        "nickName": "海洁测试!",
+        "userFollowTime": 1512717067449,
+        "wx_originalId": "gh_3f1fc031329d",
+        "wx_mpName": "测试公众号",
+        "wx_user_openId": "odjo61rGnt6Sgl6CeWhPZTfve7eA",
+        "rechargeUuid": "1fd47b51-1256-43ce-a72d-3266e23235ba",
+        "rechargeMethod": 1,
+        "money": 100,
+        "createTime": 1512717069449,
+        "payStatus": 0
+      },
+    ]
+'''
+########################## 
+
+import hashlib
+import time  
+import datetime
+
+from concurrent.futures import ProcessPoolExecutor
+import requests
+from util import date_util
+from util import platform_config_util      ## 账号配置
+from util import robust_util
+from apscheduler.schedulers.blocking import BlockingScheduler
+from util.MySQLConnection import MySQLConnection
+
+#import random
+#import math
+#from urllib import parse
+
+
+def md5(s):
+    md5 = hashlib.md5()
+    md5.update(s.encode("utf-8"))
+    return md5.hexdigest()
+
+
+def getSelfDateStr(times=time.time(),date_format='%Y%m%d'):
+    """
+    ## 20201028添加，阳光接口，文鼎接口，日期参数请求格式20201028，一日一拉api数据
+    description:  获取指定时间戳
+    time:         秒 默认当前时间
+    return:       返回指定时间戳的前一日日期 。 比如 ：接收20190512号的时间戳，返回 20190513 -> str
+    tips:         一天86400秒
+    """
+    timestamps = str(time.strftime(date_format,time.localtime(times)))
+    return timestamps
+
+
+def get_wending_account_list():
+    """
+    des cription:  文鼎账号列表
+    return:       [['consumerkey', 'secretkey', 'siteid', 'stage', 'account']] ->list
+    """
+    return platform_config_util.get_account_list('文鼎', 'wending_account_config.csv')
+
+
+## 5线程并发
+@robust_util.catch_exception
+def get_wending_order(st,et):
+    total_order_list = ()
+    start_exec_seconds = date_util.getCurrentSecondTime()
+    account_list = get_wending_account_list()
+
+    executor = ProcessPoolExecutor(max_workers=5)
+
+    futures = []
+    for account in account_list:
+        future = executor.submit(get_wending_order_task, st, et, account)
+        futures.append(future)
+
+    executor.shutdown(True)
+
+    for future in futures:
+        order_list = future.result()
+        if len(order_list) > 0:
+            total_order_list = order_list + total_order_list
+
+    print('文鼎订单数量：', len(total_order_list), '执行时长（秒）：', date_util.getCurrentSecondTime() - start_exec_seconds)
+    return total_order_list
+
+
+## 获取json对象
+def get_wending_json_object(url,params):
+  params['timestamp'] = int(time.time()*1000)
+
+  sorted_data = sorted(params.items(),reverse = False) 
+  s=""
+  for k,v in sorted_data:
+      s = s+str(k)+"="+str(v)
+  sign = md5(s).lower()
+  params['sign'] = sign
+
+  consumerkey = params['consumerkey']
+  secretkey = params['secretkey']
+  timestamp = params['timestamp']
+  siteid = params['siteid']
+  pageSize = params['pageSize'] 
+  starttime = params['starttime']
+  endtime = params['endtime']
+  page = params['page']
+  paystatus = params['paystatus']  
+  ## +'&secretkey='+str(secretkey)
+  parameter = 'consumerkey='+str(consumerkey)+'&timestamp='+str(timestamp)+'&siteid='+str(siteid)+'&pageSize='+str(pageSize)\
+              +'&starttime='+str(starttime)+'&endtime='+str(endtime)+'&page='+str(page)+'&paystatus='+str(paystatus)+'&sign='+str(sign)
+  global get_url
+  get_url = url+"?"+parameter
+  response_result_json = requests.get(url=get_url).json()
+  #response_result_json = requests.get(url=url, params=params).json()
+
+  del params['sign']
+  return response_result_json
+
+
+## 具体文鼎任务task
+def get_wending_order_task(st,et,account):
+    order_list = ()
+    url = 'https://bi.reading.163.com/dist-api/rechargeList'
+    ## 接口鉴权参数
+    consumerkey = account[0] 
+    secretkey = account[1]
+    ## 订单参数
+    siteid = account[2]
+    pageSize = 1000
+    page = 1
+    paystatus = 1
+    ## 期数自定义保存到记录中
+    stage = account[3]
+
+    while True:
+      if st >= et:
+          break
+      starttime = getSelfDateStr(st,'%Y%m%d%H%M')
+      endtime = getSelfDateStr(et,'%Y%m%d%H%M') 
+      params = {
+        'consumerkey': consumerkey,
+        'secretkey':secretkey,
+        'timestamp':int(1601481600),
+        'siteid':siteid,
+        'pageSize':pageSize, 
+        'starttime':starttime,
+        'endtime':endtime,
+        'page':page,
+        'paystatus':paystatus  
+      }
+      response_result_json = get_wending_json_object(url,params)
+
+      code = response_result_json['code']
+      if code != 200:
+        print('文鼎查询充值接口异常：',response_result_json,'传入参数', params,"请求url",get_url)
+        break
+      
+      totalPag = response_result_json['data']['totalPage']
+      if totalPag <= 0:
+        break
+
+      for page in range(1,totalPag+1):
+        params['page'] = page
+        response_result_json = get_wending_json_object(url,params)
+        order_item_list = response_result_json['data']['rechargeList']
+        '''
+        print(order_item_list)
+        [{
+          'userId': 48267585, 'userRegisterTime': 1568671618894, 'nickName': '\ue04a邓泽群\ue04a爱花园', 'ip': '124.13.64.179'
+          , 'userAgent': 'Mozilla/5.0 (Linux; Android 9; /arm64'
+          , 'userFollowTime': 1568671618894, 'wx_originalId': 'gh_0beeff4c0d70', 'wx_mpName': '美语阅读', 'wx_user_openId': 'oWL046E86PdO9wX34naL6IIwfaVc'
+          , 'rechargeUuid': '38fc210f-8e7d-4ae5-a395-d33c2a80e234', 'sourceUuid': 'ts_b4651ee782e94cce8fc6def301de1367_4', 'bookTitle': '绝代医相'
+          , 'ewTradeId': '4200000687202010012870758048', 'payTime': 1601565287000, 'rechargeMethod': 1, 'money': 2900, 'createTime': 1601565278043
+          , 'updateTime': 1601565288057, 'payStatus': 1
+        }]
+        '''
+        for x in order_item_list:
+            y={}
+            y['date'] = (int(x['payTime']//1000)+ 8 * 3600) // 86400 * 86400 - 8 * 3600    ## 网易的是13位时间戳
+            y['platform'] = '文鼎'
+            y['channel'] = x['wx_mpName']
+            y['channel_id'] = x['wx_originalId']
+            y['from_novel'] = x['bookTitle']
+            y['user_id'] = x['userId']
+            y['stage'] = stage
+            createTime = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(x['createTime']//1000))
+            y['order_time']= createTime
+            y['amount']=x['money']/100               ## 原数据单位：分 
+            uid_reg_time = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(x['userRegisterTime']//1000))
+            y['reg_time']= uid_reg_time
+            y['order_id']= x['rechargeUuid']
+
+            y = sorted(y.items(), key=lambda item:item[0])
+            y = dict(y)
+            y = tuple(y.values())
+            order_list = order_list+((y),)
+
+      if totalPag==params['page']:
+        break
+
+    print(f"文鼎数据日期-{starttime}到{endtime}-期数-{stage}-获取数据-{len(order_list)}条，例如》{order_list[0:1]}")
+    return order_list     
+
+
+def batch_save_order(data):
+    if data is None or len(data) == 0:
+        print('数据为空，不执行数据库操作！')
+    else:
+        sql = 'INSERT IGNORE INTO quchen_text.order_zwg(amount,channel,channel_id,date,from_novel,order_id,order_time,platform,reg_time,stage,user_id) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s);'
+        connect = MySQLConnection()
+        try:
+            num = connect.batch(sql, data)
+            # 提交
+            connect.commit()
+            print('订单数据最终入库【{num}】条'.format(num=num))
+        except Exception as e:
+            print('订单数据入库失败：', e)
+        finally:
+            connect.close()
+
+
+def start_order_job():
+    start_exec_seconds = date_util.getCurrentSecondTime()
+    #st_unix, et_unix = date_util.getPreviousHourAndCurrentHourSecondTime(start_exec_seconds)
+    # st_unix = 1602313200  # 2020/10/10 15:0:0
+    # et_unix = 1602316800  # 2020/10/10 16:0:0
+    #print('查询开始时间：', st_unix, date_util.getSecondsToDatetime(st_unix))
+    #print('查询结束时间：', et_unix, date_util.getSecondsToDatetime(et_unix))
+    ## 20201028添加阳光平台
+    st_unix = date_util.getYesterdayStartTime()
+    et_unix = date_util.getTodayStartTime()
+    batch_save_order(get_wending_order(st_unix, et_unix))
+    print('订单同步执行时间（秒）：', date_util.getCurrentSecondTime() - start_exec_seconds)
+
+
+start_order_job()
+'''
+start_job_time = '2020-11-06 10:04:00'
+if __name__ == '__main__':
+    scheduler = BlockingScheduler()
+    scheduler.add_job(start_order_job, 'interval', days =1 ,start_date=start_job_time)
+    #scheduler.add_job(start_order_job, 'interval',days =1,hours = 2,minutes = 0,seconds = 0)
+    #线上是24h执行一次
+    scheduler.start()     
+'''
+
+'''
+account_list = [ 
+  ['68442881','RFygHhX16LEYYe8i','1014108','趣程20期','qucheng20qi@163.com'],
+  ['77257999','86nPtJdYLe1k81gE','1021116','趣程21期','qucheng21qi@163.com']
+]
+for account in account_list:
+  get_wending_order_task(st=int(1601481600),et=int(1601568000),account=account)
+'''