MOD:添加重试机制,截图功能

web_module/tornado_api.py

@@ -116,8 +116,6 @@ class create_ad_plan(BaseHandler):
         log_ad = None
         try:
             request_dict = json.loads(self.request.body, encoding='utf-8')
-            print(self.request.body)
-            print(request_dict)
             ad_plan_list = request_dict['planList']
             user_id = request_dict['userId']
             # 2.2存行为记录

web_module/user_action.py

@@ -37,6 +37,7 @@ def carry_plan(user_id, ad_plan_list, log_ad, db, cookie_canuse, task_name):
                 # 现在默认layout_name在30个字符以内
                 layout_name = _['idea']['jump_type_page_type']['layout_name']
                 try:
+
                     # 1.检查1.落地页是否创建过了
                     log_ad.select_ad_master(service_name, wechat_name, sql_session)
                     if CreateAd.check_sucess_api(layout_name=layout_name, log_ad=log_ad):

wechat_action/create_ad_layout.py

@@ -2,6 +2,7 @@ from selenium.webdriver.support.wait import WebDriverWait
 from selenium.webdriver.common.keys import Keys
 from selenium.webdriver import ActionChains
 from logging import handlers
+from datetime import datetime
 import logging
 import requests
 import random
@@ -775,6 +776,11 @@ class CreateAd:
                     return self.create_layout(layout, sql_session, err_num=err_num + 1)
         except Exception as e:
             logging.error(e)
+            self.log_ad.driver.save_screenshot(
+                'user_id:{}_time_{}_layout_name:{}_layout_error.png'.format(self.log_ad.user_id,
+                                                                            datetime.now().strftime(
+                                                                                "%Y-%m-%d, %H:%M:%S"),
+                                                                            layout['layoutName']))
             # TODO:有空时讲 e 内容设置为原始内容
             if err_num > 3:
                 return {'sucess': False, 'result_info': str(e)}

wechat_action/create_ad_plan.py

@@ -3,6 +3,7 @@ from selenium.webdriver.support.wait import WebDriverWait
 from wechat_action.create_ad_plan_idea import IdeaAction
 from wechat_action.human_ad import HumanAd
 from selenium.webdriver.common.keys import Keys
+from datetime import datetime
 import re
 import time
 import logging
@@ -415,7 +416,9 @@ class CreateAdPlan():
                                                    sql_session=sql_session)
                     return self.run(sql_session, err_num=err_num + 1)
         except Exception as e:
-            # self.driver.save_screenshot('liuyi{}.png'.format(time.time()))
+            self.driver.save_screenshot(
+                'user_id:{}_time_{}_plan_name:{}_plan_error.png'.format(self.login_ad.user_id, datetime.now().strftime(
+                    "%Y-%m-%d, %H:%M:%S"), self.task['title']))
             time.sleep(5)
 
             # TODO:有空时讲 e 内容设置为原始内容