# -*- coding: utf-8 -*- import json from django.utils.translation import gettext_lazy as _ from mysite.ladon.ladonizer import ladonize from mysite.mobile.utils import SUCCESS_CODE, MESSAGE_CODE, interface_response, request_valid, online_employee_new, \ stamp2datetime, datetime2stamp, SYSTEM_EXCEPTION, DATA_EXCEPTION, user_photo, paging class BioTimeAppOverTime(object): """ 【Overtime】 """ @request_valid @ladonize(int, str, str, str, rtype=str) def category(self, source, device_token, language, token): """ get overtime category @param source: data source(1: IOS， 2：Android) @param device_token: Token for push message @param language: @param token: @rtype: {"code": 1, "error": "", "describe": "", "message": "", "data":[{"code": , "name": ""}} """ choices = [ {'code': 1, 'name': u'{0}'.format(_('overtime_option_normalOvertime'))}, {'code': 2, 'name': u'{0}'.format(_('overtime_option_weekendOvertime'))}, {'code': 3, 'name': u'{0}'.format(_('overtime_option_holidayOvertime'))}, ] return interface_response(SUCCESS_CODE, json.dumps(choices), '', 'successful') @request_valid @ladonize(int, int, int, str, int, str, str, str, rtype=str) def apply(self, overtime_type, start, end, remark, source, device_token, language, token): """ apply overtime @param overtime_type: ·（required）（data source BioTimeAppOverTime --> category） @param start: ·（required）end time @param end: ·（required) start time @param remark: ·（required）apply reason @param source: data source(1: IOS， 2：Android) @param device_token: Token for push message @param language: @param token: @rtype: """ from mysite.att.models.model_overtime import Overtime from mysite.att import models_choices from django.db.models import Q from mysite.personnel.models import Employee start = stamp2datetime(start) end = stamp2datetime(end) if start >= end: describe = u'{0}'.format(_('end time can not less then start time')) message = u'{0}'.format(_('end time can not less then start time')) return interface_response(MESSAGE_CODE, '', '', describe, message) applier = online_employee_new(device_token) emp_code = str(applier).split()[0] company_id = applier.company_id employee_id = Employee.objects.filter(emp_code=emp_code, company_id=company_id)[0].id select_status = [models_choices.CANCEL_AUDIT_SUCCESS, models_choices.REFUSE] obj_tmp = Overtime.objects.filter(employee_id=employee_id, start_time__lte=end, end_time__gte=start).exclude( audit_status__in=select_status) if obj_tmp: describe = u'{0}'.format(_('overtime_time_overlap') % emp_code) message = u'{0}'.format(_('overtime_time_overlap') % emp_code) return interface_response(MESSAGE_CODE, '', '', describe, message) try: obj = Overtime() obj.employee_id = employee_id obj.start_time = start obj.end_time = end obj.overtime_type = overtime_type obj.apply_reason = remark obj.save() data = { 'message': u'{0}'.format(_(u'Request already processing')) } return interface_response(SUCCESS_CODE, json.dumps(data), '', 'successful') except Exception as e: import traceback traceback.print_exc() return interface_response(MESSAGE_CODE, '', '', e, SYSTEM_EXCEPTION) @request_valid @ladonize(int, int, int, str, str, str, rtype=str) def my_application(self, approve_status, page_num, source, device_token, language, token): """ get own apply record(pending, approve, reject) @param approve_status: 0:approved&rejected, 1:pending, 2:approved，3：rejected @param page_num: page number(１５items/page) @param source: data source(1: IOS， 2：Android) @param device_token: Token for push message @param language: @param token: @rtype: success {"code":1,"error":"","describe":"","message":"","data":{"category":1," items":{"code": ID,"pin":"emp_code","name":"first_name","photo":"photo address","start": "start time(stamp)","end":"end time(stamp)","remark":"apply reason","category":"","apply_time":"", "approve_status":"int value","approve_describe":"status describe","approved_remark":"approve reason", "approved_time":"approve time"},]}} fail {"code":-10001,"error":"","describe":"","message":"Pop-up message","data":""} """ from mysite.sql_utils import get_sql, p_query from mysite.att import models_choices from mysite.mobile.choices import CATEGORY_OVERTIME if approve_status in (0, 1, 2, 3): emp = online_employee_new(device_token) if not approve_status: _approve_status = ' audit_status in (%s, %s, %s) ' % ( models_choices.AUDIT_SUCCESS, models_choices.REFUSE, models_choices.CANCEL_AUDIT_SUCCESS) elif approve_status in (models_choices.APPLICATION,): _approve_status = ' audit_status in (%s, %s) ' % (models_choices.APPLICATION, models_choices.AUDITING) elif approve_status in (models_choices.REFUSE,): _approve_status = ' audit_status in (%s, %s) ' % ( models_choices.REFUSE, models_choices.CANCEL_AUDIT_SUCCESS) else: _approve_status = ' audit_status in (%s) ' % approve_status where = ' u.emp_code = %(applier)s and %(audit_status)s ' % ( {'applier': emp.pk, 'audit_status': _approve_status}) sort_name = 'apply_time' if approve_status in (1,): # Apply sort_name = 'apply_time' page_num = page_num or 1 try: sql = get_sql('sql', sqlid='overtime_application', app="mobile", params={'where': where}) sql = paging(sql, page_num, sort_name) rows = p_query(sql) data = { 'category': CATEGORY_OVERTIME, 'items': [] } if rows: status = dict(models_choices.ALL_AUDIT_STATUS) types = dict(models_choices.OVERTIME_TYPE) items = [{'code': r[0], 'pin': r[1], 'name': r[2], 'photo': user_photo(r[1]), 'start': datetime2stamp(r[3]), 'end': datetime2stamp(r[4]), 'remark': u'{0}'.format(r[5]), 'category': u'{0}'.format(types.get(r[6], r[6])), 'apply_time': datetime2stamp(r[7]), 'approve_status': r[8], 'approve_describe': u'{0}'.format(status.get(r[8], r[8])), 'approved_remark': u'{0}'.format(r[10]), 'approved_time': datetime2stamp(r[7])} for r in rows] data['items'] = items return interface_response(SUCCESS_CODE, json.dumps(data), '', 'successful') except Exception as e: import traceback traceback.print_exc() return interface_response(MESSAGE_CODE, '', '', e, SYSTEM_EXCEPTION) else: describe = 'parameter approve_status={0} error'.format(approve_status) return interface_response(MESSAGE_CODE, '', '', describe, DATA_EXCEPTION) @request_valid @ladonize(int, int, int, int, str, str, str, rtype=str) def approval_list(self, approve_status, page_num, order_by, source, device_token, language, token): """ get approve list @param approve_status: ·（required）0:approved&rejected, 1:pending, 2:approved，3：rejected @param page_num: page number(１５items/page) @param order_by: sort(1: apply time，２：approve time), application page default１，approve page default２ @param source: data source(1: IOS， 2：Android) @param device_token: Token for push message @param language: @param token: @rtype: {"code": 1, "error": "", "describe": "", "message": "", "data":[{"id": ID, "pin": "emp_code", "name":"first_name", "photo":"photo address", "start": "", "end": "", "remark":"apply reason", "approve_status": ""},]} """ from mysite.att import models_choices from mysite.workflow.models import NodeInstance from django.contrib.contenttypes.models import ContentType from mysite.att.models import Overtime from django.db.models import Q if approve_status in (0, 1, 2, 3, 4, 5, 6): status = dict(models_choices.ALL_AUDIT_STATUS) types = dict(models_choices.OVERTIME_TYPE) if not approve_status: _approve_status = [2, 3] elif approve_status in (models_choices.APPLICATION,): _approve_status = [models_choices.APPLICATION, models_choices.AUDITING] else: _approve_status = [approve_status] emp = online_employee_new(device_token) emp_roles = emp.flow_role.all() data = [] try: if emp_roles: ct_ot = ContentType.objects.get_by_natural_key('att', 'overtime') if _approve_status == [2]: NodeInstance_obj = NodeInstance.objects.filter( workflow_instance__workflow_engine__content_type=ct_ot.id, node_engine__approver__in=emp_roles, state__in=_approve_status, approver_admin_id=None, approver_employee_id=emp.id ) else: NodeInstance_obj = NodeInstance.objects.filter( workflow_instance__workflow_engine__content_type=ct_ot.id, node_engine__approver__in=emp_roles, state__in=_approve_status, approver_admin_id=None, is_next_node=True, # approver_employee_id__isnull=False, ) NodeInstance_obj_without_depart = NodeInstance.objects.filter( workflow_instance__workflow_engine__content_type=ct_ot.id, node_engine__approver_by_overall=True, departments=None, node_engine__approver__in=emp_roles, state__in=_approve_status, is_next_node=True, approver_admin_id=None) NodeInstance_obj_with_depart = NodeInstance.objects.filter( workflow_instance__workflow_engine__content_type=ct_ot.id, node_engine__approver_by_overall=False, departments=emp.department, node_engine__approver__in=emp_roles, state__in=_approve_status, is_next_node=True, approver_admin_id=None) approve_nodes = NodeInstance.objects.filter( Q(id__in=NodeInstance_obj.values_list('id', flat=True)) | Q(id__in=NodeInstance_obj_with_depart.values_list('id', flat=True)) | Q(id__in=NodeInstance_obj_without_depart.values_list('id', flat=True))).values_list( 'workflow_instance__exception_id', 'workflow_instance__employee', 'state', 'remark', 'apply_time', 'is_last_node' ).distinct().exclude(workflow_instance__employee_id=emp.id) if approve_nodes: prv_exception = [] for r in approve_nodes: exception_id = r[0] last_node = r[-1] if not last_node: if _approve_status in ([2, 3], [2]): continue if exception_id in prv_exception: # continue apply_obj = Overtime.objects.filter(id=exception_id).values_list('overtime_type', 'start_time', 'end_time', 'apply_reason', 'apply_time', 'employee__emp_code', 'employee__first_name', 'audit_time') if apply_obj: apply_obj = apply_obj[0] res_data = { 'code': exception_id, 'pin': apply_obj[5], 'name': apply_obj[6], 'photo': user_photo(apply_obj[5]), 'start': datetime2stamp(apply_obj[1]), 'end': datetime2stamp(apply_obj[2]), 'remark': u'{0}'.format(apply_obj[3] and apply_obj[3] or ''), 'category': u'{0}'.format(types.get(apply_obj[0], apply_obj[0])), 'apply_time': datetime2stamp(apply_obj[4]), 'approve_status': r[2], 'approve_describe': u'{0}'.format(status.get(r[2], r[2])), 'approved_remark': u'{0}'.format(r[3]), 'approved_time': datetime2stamp(apply_obj[7]) } prv_exception.append(exception_id) data.append(res_data) data_filtered_list = [] if order_by == 1: data_filtered_list = sorted(data, key=lambda k: k['approved_time'], reverse=True) elif order_by == 2: data_filtered_list = sorted(data, key=lambda k: k['apply_time'], reverse=True) return interface_response(SUCCESS_CODE, json.dumps(data_filtered_list), '', 'successful') except Exception as e: import traceback traceback.print_exc() return interface_response(MESSAGE_CODE, '', '', SYSTEM_EXCEPTION, e) else: describe = _('approve_status_not_in (0,1,2,3)') return interface_response(MESSAGE_CODE, '', '', describe, DATA_EXCEPTION) @request_valid @ladonize(int, int, str, int, str, str, str, rtype=str) def approve(self, code, approve_status, remark, source, device_token, language, token): """ overtime approve @param code: ID @param approve_status: ·(required）audit status(approved/rejected) @param source: data source(1: IOS， 2：Android) @param device_token: push message Token @param language: @param token: @rtype: """ from mysite.att.models.model_overtime import Overtime from mysite.att import models_choices from mysite.workflow.models import NodeInstance from mysite.workflow.models.workflow_instance import WorkflowInstance import datetime if code: if approve_status not in (models_choices.AUDIT_SUCCESS, models_choices.REFUSE): describe = _('param_approve_status_out_of_range') return interface_response(MESSAGE_CODE, '', '', describe, DATA_EXCEPTION) approver = online_employee_new(device_token) objs = Overtime.objects.filter(abstractexception_ptr_id=code) if objs: try: obj = objs[0] audit_time = datetime.datetime.now() if approve_status == models_choices.AUDIT_SUCCESS: obj.audit_reason = remark obj.approver = str(approver).split()[1] obj.audit_time = audit_time obj.save() obj.workflowinstance.approve_current_node_by(approver, remark) nodes = NodeInstance.objects.filter(workflow_instance__exception=code).all().order_by('order') if nodes: for i, node in enumerate(nodes): index = i set_next_node = False current_node = node.is_next_node is_last_node = node.is_last_node if is_last_node: break if current_node: node.is_next_node = False node.save() set_next_node = True break if set_next_node: next_node = nodes[index + 1] next_node.is_next_node = True next_node.save() elif approve_status == models_choices.REFUSE: obj.audit_reason = remark obj.approver = str(approver).split()[1] obj.save() obj.workflowinstance.reject_current_node_by(approver, obj.audit_reason) state_ = 0 workflow_instance = WorkflowInstance.objects.filter( exception=obj.abstractexception_ptr_id).first() node_set = workflow_instance.nodeinstance_set.all().order_by('order') for i in node_set: if i.state == 3: state_ = i.state if state_ == 3: NodeInstance.objects.filter(id=i.id).update(state=state_) data = { 'message': u'{0}'.format(_(u'OK')) } return interface_response(SUCCESS_CODE, json.dumps(data), '', 'successful') except Exception as e: import traceback traceback.print_exc() return interface_response(MESSAGE_CODE, '', '', SYSTEM_EXCEPTION, e) else: describe = _('object_not_found') return interface_response(MESSAGE_CODE, '', '', describe, DATA_EXCEPTION) else: describe = _('object_id_not_found') return interface_response(MESSAGE_CODE, '', '', describe, DATA_EXCEPTION) @request_valid @ladonize(int, str, int, str, str, str, rtype=str) def revoke(self, code, remark, source, device_token, language, token): """ revoke approve @param code: Obj ID @param remark: audit_reason @param source: data source (1: IOS， 2：Android) @param device_token: Push message Token @param language: @param token: @rtype: """ import datetime from mysite.att.models import Overtime from mysite.att.models_choices import AUDIT_SUCCESS, CANCEL_AUDIT_SUCCESS obj = Overtime.objects.filter(id=code).first() if obj: if obj.audit_status == AUDIT_SUCCESS: user = online_employee_new(device_token) obj.audit_reason = 'Revoke by {0} and Remark is {1}'.format(user.name, remark) obj.audit_status = CANCEL_AUDIT_SUCCESS obj.approver = user.name obj.audit_time = datetime.datetime.now() obj._approve_user = user obj.save() data = { 'message': u'{0}'.format(_(u'OK')) } message = _('revoked_successful') return interface_response(SUCCESS_CODE, json.dumps(data), '', '', message) else: describe = _('only_approved_records_can_be_revoked') else: describe = _('workflow_instance_does_not_exists') if describe: return interface_response(MESSAGE_CODE, '', '', DATA_EXCEPTION, describe) return interface_response(MESSAGE_CODE, '', '', DATA_EXCEPTION, describe)