renyupeng 2 роки тому
батько
коміт
c722271e43
2 змінених файлів з 45 додано та 1 видалено
  1. 33 0
      spider/ReturnPromoterInfoSpider.py
  2. 12 1
      utils/PromoterInfoWebHook.py

+ 33 - 0
spider/ReturnPromoterInfoSpider.py

@@ -15,6 +15,7 @@ import requests
 from utils.cookie_update import cookie_update
 from utils.mysql_helper import insert
 from utils.mysql_utils import MysqlUtils
+from utils.mysql_utils_pro import MysqlProUtils
 from utils.send_feishu_msg import SendFeiShuMsg
 
 
@@ -32,6 +33,7 @@ class ReturnPromoterInfoSpider:
 
         try:
             rep = requests.get(url=url, headers=headers)
+            print(rep.text, '-----rep.txt---------')
             if json.loads(rep.text)["result"] != 1:
                 errItem = {"result": json.loads(rep.text)["result"]}
                 return json.dumps(errItem)
@@ -47,8 +49,39 @@ class ReturnPromoterInfoSpider:
                     "avgVideoSales": data["promoteBaseInfo"]["avgVideoSales"],
                     "videoSales": data["promoteBaseInfo"]["videoSales"]
                 }
+
+                sql = """update ruixuan.promoter_info set videoSales={videoSales} 
+                , avgVideoSales={avgVideoSales} where promoter_id={promoter_id}""".format(
+                    promoterId=promoterId, avgVideoSales=data["promoteBaseInfo"]["avgVideoSales"],
+                    videoSales=data["promoteBaseInfo"]["videoSales"])
+                MysqlProUtils().Operate(sql=sql)
                 returnItemJson = json.dumps(returnItem)
                 return returnItemJson
         except Exception as e:
             SendFeiShuMsg.send_robot_msg(
                 'ReturnPromoterInfoSpider {promoterId}请求错误请检查cookie{e}'.format(promoterId=promoterId, e=e))
+
+    @staticmethod
+    def PromoterUpdate(promoterId, cookie):
+        url = 'https://cps.kwaixiaodian.com/distribute/pc/seller/promoter/info?promoterId={promoterId}&type=1'.format(
+            promoterId=promoterId)
+        headers = {'User-Agent': 'Mozilla/5.0',
+                   'Cookie': cookie}
+
+        try:
+            rep = requests.get(url=url, headers=headers)
+            if json.loads(rep.text)["result"] != 1:
+                errItem = {"result": json.loads(rep.text)["result"]}
+                return json.dumps(errItem)
+            else:
+                data = json.loads(rep.text)["data"]
+                avgVideoSales = data["promoteBaseInfo"]["avgVideoSales"]
+                videoSales = data["promoteBaseInfo"]["videoSales"]
+                sql = "update ruixuan.promoter_info  set videoSales= '{videoSales}' , avgVideoSales='{avgVideoSales}' where promoter_id={promoter_id}".format(
+                    videoSales=videoSales,
+                    avgVideoSales=avgVideoSales, promoter_id=promoterId
+                )
+                MysqlProUtils().Operate(sql=sql)
+        except Exception as e:
+            SendFeiShuMsg.send_robot_msg(
+                'PromoterUpdate {promoterId}请求错误请检查cookie{e}'.format(promoterId=promoterId, e=e))

+ 12 - 1
utils/PromoterInfoWebHook.py

@@ -58,7 +58,9 @@ def webhook_get_promoter():
             return rollback
 
         else:
-            SendFeiShuMsg.send_cookie_robot_msg("cookie 临时失效 唯一电话为{phone_num},临时失效code{code}".format(phone_num=phone_num,code=json.loads(roll)["result"]))
+            SendFeiShuMsg.send_cookie_robot_msg(
+                "cookie 临时失效 唯一电话为{phone_num},临时失效code{code}".format(phone_num=phone_num,
+                                                                     code=json.loads(roll)["result"]))
             cookie_update.update_temporary_cookie_handler(phone_num)
             retry_cookie = cookie_update.get_cookie_handler()[0]
             rolls = ReturnPromoterInfoSpider().PromoterInfoSpiderHandler(promoterId, retry_cookie)
@@ -90,5 +92,14 @@ def webhook_get_promoter():
         return roll
 
 
+@app.route('/promoterInfo/updatePromoter', methods=['POST'])
+def webhook_update_promoter():
+    rep = json.loads(request.data)
+    promoterId = rep["promoterId"]
+    cookie = cookie_update.get_cookie_handler()[0]
+    ReturnPromoterInfoSpider().PromoterUpdate(promoterId, cookie)
+    return '0'
+
+
 if __name__ == '__main__':
     app.run(port=9999, host=ConfConstant.URL, debug=True)