# from __future__ import absolute_import, unicode_literals import contextlib import os import sys import traceback import logging from celery import Celery # set the default Django settings module for the 'celery' program. os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'settings.config') import django django.setup() app = Celery('loonflow') # Using a string here means the worker doesn't have to serialize # the configuration object to child processes. # - namespace='CELERY' means all celery-related configuration keys # should have a `CELERY_` prefix. app.config_from_object('django.conf:settings', namespace='CELERY') # Load task modules from all registered Django app configs. app.autodiscover_tasks() import json import requests from apps.ticket.models import TicketRecord from apps.workflow.models import Transition, State, WorkflowScript, Workflow, CustomNotice from service.account.account_base_service import account_base_service_ins from service.common.constant_service import constant_service_ins from service.ticket.ticket_base_service import TicketBaseService, ticket_base_service_ins from service.common.common_service import CommonService, common_service_ins from service.workflow.workflow_transition_service import WorkflowTransitionService, workflow_transition_service_ins from django.conf import settings try: from StringIO import StringIO except ImportError: from io import StringIO logger = logging.getLogger('django') @app.task(bind=True) def debug_task(self): print('Request: {0!r}'.format(self.request)) @app.task def test_task(a, b): print('a:', a) print('b:', b) print(a+b) @contextlib.contextmanager def stdoutIO(stdout=None): old = sys.stdout if stdout is None: try: # for python2 stdout = StringIO.StringIO() except Exception: stdout = StringIO() sys.stdout = stdout yield stdout sys.stdout = old @app.task def run_flow_task(ticket_id, script_id_str, state_id, action_from='loonrobot'): """ 执行工作流脚本 :param script_id_star:通过脚本id来执行, 保存的是字符串 :param ticket_id: :param state_id: :param action_from: :return: """ script_id = int(script_id_str) ticket_obj = TicketRecord.objects.filter(id=ticket_id, is_deleted=False).first() if ticket_obj.participant == script_id_str and ticket_obj.participant_type_id == constant_service_ins.PARTICIPANT_TYPE_ROBOT: ## 校验脚本是否合法 # 获取脚本名称 script_obj = WorkflowScript.objects.filter(id=script_id, is_deleted=False, is_active=True).first() if not script_obj: return False, '脚本未注册或非激活状态' script_file = os.path.join(settings.MEDIA_ROOT, script_obj.saved_name.name) globals = {'ticket_id': ticket_id, 'action_from': action_from} # 如果需要脚本执行完成后，工单不往下流转(也就脚本执行失败或调用其他接口失败的情况)，需要在脚本中抛出异常 try: with stdoutIO() as s: # execfile(script_file, globals) # for python 2 exec(open(script_file, encoding='utf-8').read(), globals) script_result = True # script_result_msg = ''.join(s.buflist) script_result_msg = ''.join(s.getvalue()) except Exception as e: logger.error(traceback.format_exc()) script_result = False script_result_msg = e.__str__() logger.info('*' * 20 + '工作流脚本回调,ticket_id:[%s]' % ticket_id + '*' * 20) logger.info('*******工作流脚本回调，ticket_id:{}*****'.format(ticket_id)) # 因为上面的脚本执行时间可能会比较长，为了避免db session失效，重新获取ticket对象 ticket_obj = TicketRecord.objects.filter(id=ticket_id, is_deleted=False).first() # 新增处理记录,脚本后只允许只有一个后续直连状态 transition_obj = Transition.objects.filter(source_state_id=state_id, is_deleted=False).first() new_ticket_flow_dict = dict(ticket_id=ticket_id, transition_id=transition_obj.id, suggestion=script_result_msg, participant_type_id=constant_service_ins.PARTICIPANT_TYPE_ROBOT, participant='脚本:(id:{}, name:{})'.format(script_obj.id, script_obj.name), state_id=state_id, creator='loonrobot') ticket_base_service_ins.add_ticket_flow_log(new_ticket_flow_dict) if not script_result: # 脚本执行失败，状态不更新,标记任务执行结果 ticket_obj.script_run_last_result = False ticket_obj.save() return False, script_result_msg # 自动执行流转 flag, msg = ticket_base_service_ins.handle_ticket(ticket_id, dict(username='loonrobot', suggestion='脚本执行完成后自行流转', transition_id=transition_obj.id), False, True) if flag: logger.info('******脚本执行成功,工单基础信息更新完成, ticket_id:{}******'.format(ticket_id)) return flag, msg else: return False, '工单当前处理人为非脚本，不执行脚本' @app.task def timer_transition(ticket_id, state_id, date_time, transition_id): """ 定时器流转 :param ticket_id: :param state_id: :param date_time: :param transition_id: :return: """ # 需要满足工单此状态后续无其他操作才自动流转 # 查询该工单此状态所有操作 # flow_log_set, msg = TicketBaseService().get_ticket_flow_log(ticket_id, 'loonrobot', per_page=1000) flag, result = ticket_base_service_ins.get_ticket_flow_log(ticket_id, 'loonrobot', per_page=1000) if flag is False: return False, result flow_log_list = result.get('ticket_flow_log_restful_list') for flow_log in flow_log_list: if flow_log.get('state').get('state_id') == state_id and flow_log.get('gmt_created') > date_time: return True, '后续有操作，定时器失效' # 执行流转 handle_ticket_data = dict(transition_id=transition_id, username='loonrobot', suggestion='定时器流转') ticket_base_service_ins.handle_ticket(ticket_id, handle_ticket_data, True) @app.task def send_ticket_notice(ticket_id): """ 发送工单通知 :param ticket_id: :return: """ # 获取工单信息 # 获取工作流信息，获取工作流的通知信息 # 获取通知信息的标题和内容模板 # 将通知内容，通知标题，通知人，作为hook的请求参数 ticket_obj = TicketRecord.objects.filter(id=ticket_id, is_deleted=0).first() if not ticket_obj: return False, 'ticket is not exist or has been deleted' workflow_id = ticket_obj.workflow_id workflow_obj = Workflow.objects.filter(id=workflow_id, is_deleted=0).first() notices = workflow_obj.notices if not notices: return True, 'no notice defined' notice_str_list = notices.split(',') notice_id_list = [int(notice_str) for notice_str in notice_str_list] send_notice_result_list = [] title_template = workflow_obj.title_template content_template = workflow_obj.content_template # 获取工单所有字段的变量 flag, ticket_value_info = ticket_base_service_ins.get_ticket_all_field_value(ticket_id) if flag is False: return False, ticket_value_info title_result = title_template.format(**ticket_value_info) content_result = content_template.format(**ticket_value_info) # 获取工单最后一条操作记录 flag, result = ticket_base_service_ins.get_ticket_flow_log(ticket_id, 'loonrobot') flow_log_list = result.get('ticket_flow_log_restful_list') last_flow_log = flow_log_list[0] participant_info_list = [] participant_username_list = [] from apps.account.models import LoonUser if ticket_obj.participant_type_id == constant_service_ins.PARTICIPANT_TYPE_PERSONAL: participant_username_list = [ticket_obj.participant] elif ticket_obj.participant_type_id == constant_service_ins.PARTICIPANT_TYPE_MULTI: participant_username_list = ticket_obj.participant.split(',') elif ticket_obj.participant_type_id == constant_service_ins.PARTICIPANT_TYPE_ROLE: flag, participant_username_list = account_base_service_ins.get_role_username_list(ticket_obj.participant) if flag is False: return False, participant_username_list elif ticket_obj.participant_type_id == constant_service_ins.PARTICIPANT_TYPE_DEPT: flag, participant_username_list = account_base_service_ins.get_dept_username_list(ticket_obj.participant) if not flag: return flag, participant_username_list if participant_username_list: participant_queryset = LoonUser.objects.filter(username__in=participant_username_list, is_deleted=0) for participant_0 in participant_queryset: participant_info_list.append(dict(username=participant_0.username, alias=participant_0.alias, phone=participant_0.phone, email=participant_0.email)) params = {'title_result': title_result, 'content_result': content_result, 'participant': ticket_obj.participant, 'participant_type_id': ticket_obj.participant_type_id, 'multi_all_person': ticket_obj.multi_all_person, 'ticket_value_info': ticket_value_info, 'last_flow_log': last_flow_log, 'participant_info_list': participant_info_list} for notice_id in notice_id_list: notice_obj = CustomNotice.objects.filter(id=notice_id, is_deleted=0).first() if not notice_obj: continue hook_url = notice_obj.hook_url from service.workflow.workflow_base_service import workflow_base_service_ins flag, msg = workflow_base_service_ins.hook_host_valid_check(hook_url) if not flag: return False, msg hook_token = notice_obj.hook_token # gen signature flag, headers = common_service_ins.gen_signature_by_token(hook_token) try: r = requests.post(hook_url, headers=headers, json=params) result = r.json() if result.get('code') == 0: send_notice_result_list.append(dict(notice_id=notice_id, result='success', msg=result.get('msg', ''))) else: send_notice_result_list.append(dict(notice_id=notice_id, result='fail', msg=result.get('msg', ''))) except Exception as e: send_notice_result_list.append(dict(notice_id=notice_id, result='fail', msg=e.__str__())) return True, dict(send_notice_result_list=send_notice_result_list) @app.task def flow_hook_task(ticket_id): """ hook 任务 :param ticket_id: :return: """ # 查询工单状态 ticket_obj = TicketRecord.objects.filter(id=ticket_id, is_deleted=0).first() state_id = ticket_obj.state_id state_obj = State.objects.filter(id=state_id, is_deleted=0).first() participant_type_id = state_obj.participant_type_id if participant_type_id != constant_service_ins.PARTICIPANT_TYPE_HOOK: return False, '' hook_config = state_obj.participant hook_config_dict = json.loads(hook_config) hook_url = hook_config_dict.get('hook_url') hook_token = hook_config_dict.get('hook_token') wait = hook_config_dict.get('wait') extra_info = hook_config_dict.get('extra_info') from service.workflow.workflow_base_service import workflow_base_service_ins hook_check_flag, hook_result_msg = workflow_base_service_ins.hook_host_valid_check(hook_url) flag, msg = common_service_ins.gen_hook_signature(hook_token) if not flag: hook_check_flag, hook_result_msg = flag, msg flag, all_ticket_data = ticket_base_service_ins.get_ticket_all_field_value(ticket_id) flag, all_field_value_result = ticket_base_service_ins.get_ticket_all_field_value_json(ticket_id) all_ticket_data_json = all_field_value_result.get('all_field_value_json') if hook_check_flag: if extra_info is not None: all_ticket_data.update(dict(extra_info=extra_info)) try: r = requests.post(hook_url, headers=msg, json=all_ticket_data, timeout=10) result = r.json() except Exception as e: result = dict(code=-1, msg=e.__str__()) if result.get('code') == 0: # 调用成功 TicketBaseService().add_ticket_flow_log(dict(ticket_id=ticket_id, transition_id=0, suggestion=result.get('msg'), participant_type_id=constant_service_ins.PARTICIPANT_TYPE_HOOK, participant='hook', state_id=state_id, intervene_type_id=constant_service_ins.TRANSITION_INTERVENE_TYPE_HOOK, ticket_data=all_ticket_data_json, creator='loonrobot' )) if not wait: # 不等待hook目标回调，直接流转 flag, transition_queryset = workflow_transition_service_ins.get_state_transition_queryset(state_id) transition_id = transition_queryset[0].id # hook状态只支持一个流转 new_request_dict = {} new_request_dict.update({'transition_id': transition_id, 'suggestion': msg, 'username': 'loonrobot'}) # 执行流转 flag, msg = ticket_base_service_ins.handle_ticket(ticket_id, new_request_dict, by_hook=True) if not flag: return False, msg return True, '' hook_result_msg = result.get('msg') ticket_base_service_ins.update_ticket_field_value(ticket_id, {'script_run_last_result': False}) ticket_base_service_ins.add_ticket_flow_log( dict(ticket_id=ticket_id, transition_id=0, suggestion=hook_result_msg, participant_type_id=constant_service_ins.PARTICIPANT_TYPE_HOOK, participant='hook', state_id=state_id, ticket_data=all_ticket_data_json, creator='loonrobot'))