增加字典‘DISPATCHER_TRIGGER_TYPE’表示‘调度触发类型’

This commit is contained in:
azhengzz 2021-03-10 17:55:43 +08:00
parent 44ee8f39b3
commit 8c76f100a3
2 changed files with 10 additions and 3 deletions

View File

@ -107,3 +107,9 @@ class CONTENT_TYPE:
"""HTTP请求体数据类型""" """HTTP请求体数据类型"""
X_WWW_FORM_URLENCODED = 'application/x-www-form-urlencoded' X_WWW_FORM_URLENCODED = 'application/x-www-form-urlencoded'
FORM_DATA = 'multipart/form-data' FORM_DATA = 'multipart/form-data'
class DISPATCHER_TRIGGER_TYPE:
"""调度触发类型"""
BY_HAND = '手工'
BY_SCHEDULE = '定时任务'

View File

@ -13,7 +13,8 @@ import contextlib
from app.template_global import sort_by_order_in_module, sort_by_order_in_project, sort_by_order_in_logic_controller from app.template_global import sort_by_order_in_module, sort_by_order_in_project, sort_by_order_in_logic_controller
from app.extensions import http_cookie_manager from app.extensions import http_cookie_manager
from app.cores.dictionaries import (ELEMENT_TYPE, STATUS, CASE_TYPE, DISPATCHER_STATUS, DISPATCHER_TYPE, from app.cores.dictionaries import (ELEMENT_TYPE, STATUS, CASE_TYPE, DISPATCHER_STATUS, DISPATCHER_TYPE,
DISPATCHER_END_TYPE, REPORT_RESULT, TOOL_TYPE, LOGIC_CONTROLLER_TYPE) DISPATCHER_END_TYPE, REPORT_RESULT, TOOL_TYPE, LOGIC_CONTROLLER_TYPE,
DISPATCHER_TRIGGER_TYPE)
from app.cores.logger import DispatcherLogger from app.cores.logger import DispatcherLogger
from app.models import (Case, Scene, Module, Project, Dispatcher, DispatcherDetail, Report, ReportCaseData, from app.models import (Case, Scene, Module, Project, Dispatcher, DispatcherDetail, Report, ReportCaseData,
ReportCaseExpectationData, LogicController, DingTalkRobotSetting, Tool, ReportToolData, ReportCaseExpectationData, LogicController, DingTalkRobotSetting, Tool, ReportToolData,
@ -994,7 +995,7 @@ def async_project_run(project_id):
with app.test_request_context(): # 在线程中创建请求上下文,当栈中没有应用上下文时同时也会创建应用上下文 with app.test_request_context(): # 在线程中创建请求上下文,当栈中没有应用上下文时同时也会创建应用上下文
# 将主线程请求上下文栈中的request和session放入子线程的请求上下文栈顶 # 将主线程请求上下文栈中的request和session放入子线程的请求上下文栈顶
_request_ctx_stack.top.request = request _request_ctx_stack.top.request = request
session['dispatcher_trigger_type'] = '手动' # 调度触发类型 session['dispatcher_trigger_type'] = DISPATCHER_TRIGGER_TYPE.BY_HAND # 调度触发类型
_request_ctx_stack.top.session = session _request_ctx_stack.top.session = session
ProjectDispatcher(project_id=project_id).run() ProjectDispatcher(project_id=project_id).run()
except Exception as e: except Exception as e:
@ -1009,7 +1010,7 @@ def apscheduler_async_project_run(project_id, app, request, session):
with app.test_request_context(): # 在线程中创建请求上下文,当栈中没有应用上下文时同时也会创建应用上下文 with app.test_request_context(): # 在线程中创建请求上下文,当栈中没有应用上下文时同时也会创建应用上下文
# 将主线程请求上下文栈中的request和session放入子线程的请求上下文栈顶 # 将主线程请求上下文栈中的request和session放入子线程的请求上下文栈顶
_request_ctx_stack.top.request = request _request_ctx_stack.top.request = request
session['dispatcher_trigger_type'] = '定时任务' # 调度触发类型 session['dispatcher_trigger_type'] = DISPATCHER_TRIGGER_TYPE.BY_SCHEDULE # 调度触发类型
_request_ctx_stack.top.session = session _request_ctx_stack.top.session = session
ProjectDispatcher(project_id=project_id).run() ProjectDispatcher(project_id=project_id).run()
except Exception as e: except Exception as e: