shawenguan hace 1 año
padre
commit
1c104dc8f4
Se han modificado 1 ficheros con 10 adiciones y 9 borrados
  1. 10 9
      addons/elife/OfpayHelper.py

+ 10 - 9
addons/elife/OfpayHelper.py

@@ -7,7 +7,7 @@ import pymysql
 import logging
 # import jwt
 import json
-from datetime import datetime
+from datetime import datetime, timezone, timedelta
 
 from mitmproxy import flowfilter
 from mitmproxy import http
@@ -54,6 +54,11 @@ flow.response.timestamp_start #响应开始的时间戳
 flow.response.timestamp_end #响应结束的时间戳
 
 """
+def seconds_to_beijing_time(seconds):
+    tmp_dt = datetime.utcfromtimestamp(seconds);
+    tmp_dt= tmp_dt.astimezone(timezone(timedelta(hours=8)));
+    time_str = tmp_dt.strftime('%Y-%m-%d %H:%M:%S');
+    return time_str;
 
 class OfpayHelper:
     order_simple_data = {
@@ -265,10 +270,8 @@ class OfpayHelper:
                 payload['customerInfo'] = customer_info;
             account = payload['customerInfo']['phone'];
 
-            tmp_dt = datetime.utcfromtimestamp(payload['iat']);
-            sign_time = tmp_dt.strftime('%Y-%m-%d %H:%M:%S');
-            tmp_dt = datetime.utcfromtimestamp(payload['exp']);
-            expire_time = tmp_dt.strftime('%Y-%m-%d %H:%M:%S');
+            sign_time = seconds_to_beijing_time(payload['iat']);
+            expire_time = seconds_to_beijing_time(payload['exp']);
 
         except Exception as e:
             print(e);
@@ -416,10 +419,8 @@ class OfpayHelper:
             rsp_data['list'].insert(0, simple_data);
 
         update_time = create_time;
-        tmp_dt = datetime.utcfromtimestamp(payload['iat']);
-        sign_time = tmp_dt.strftime('%Y-%m-%d %H:%M:%S');
-        tmp_dt = datetime.utcfromtimestamp(payload['exp']);
-        expire_time = tmp_dt.strftime('%Y-%m-%d %H:%M:%S');
+        sign_time = seconds_to_beijing_time(payload['iat']);
+        expire_time = seconds_to_beijing_time(payload['exp']);
         simple_data['createTime'] = expire_time;
 
         sql_query = f'''