from rest_framework.exceptions import ParseError, APIException import logging import time from apps.hrm.models import ClockRecord, Employee from apps.third.errors import DH_REQUEST_ERROR from apps.third.models import TDevice from apps.third.tapis import dhapis from apps.third.dahua import dhClient from apps.utils.tools import rannum, ranstr from datetime import datetime, timedelta from django.utils import timezone from dateutil import tz myLogger = logging.getLogger('log') class HrmService: @classmethod def sync_dahua_employee(cls, ep: Employee, old_photo='', start_time=None, end_time=None): """同步大华信息(员工/卡片/门禁) Args: ep (Employee): 人员实例 old_photo (str, optional): 原照片地址. Defaults to ''. start_time (_type_, optional): 开人脸卡起始时间. Defaults to None. end_time (_type_, optional): 开人脸卡结束时间. Defaults to None. Returns: _type_: _description_ """ dh_id = ep.third_info.get('dh_id', None) dh_photo = ep.third_info.get('dh_photo', None) dh_face_card = ep.third_info.get('dh_face_card', None) departmentId = 1 if ep.belong_dept: try: departmentId = ep.belong_dept.third_info['dh_id'] except Exception: pass if ep.third_info.get('dh_id', None): # 如果有大华信息 dh_id = ep.third_info['dh_id'] dh_photo = ep.third_info['dh_photo'] json_data = { "service": "ehs", "id": dh_id, "name": ep.name, "code": ep.number if ep.number else ranstr(6), "paperType": 111, "paperNumber": ep.id_number, "paperAddress": "default", "departmentId": departmentId, "phone": ep.phone, "email": ep.email, "sex": 1 if ep.gender == '男' else 2, "biosignatureTypeList": [3], "personBiosignatures": [{ "type": 3, "index": 1, "path": dh_photo }] } if ep.photo != old_photo and ep.photo: _, res = dhClient.request(**dhapis['person_img_upload'], file_path_rela=ep.photo) dh_photo = res["fileUrl"] json_data.update( { "biosignatureTypeList": [3], "personBiosignatures": [{ "type": 3, "index": 1, "path": dh_photo }] } ) dhClient.request(**dhapis['person_update'], json=json_data) else: _, res = dhClient.request(**dhapis['person_gen_id']) dh_id = res['id'] json_data = { "service": "ehs", "id": dh_id, "name": ep.name, "code": ep.number if ep.number else ranstr(6), "paperType": 111, "paperNumber": ep.id_number, "paperAddress": "default", "departmentId": departmentId, "phone": ep.phone, "email": ep.email, "sex": 1 if ep.gender == '男' else 2 } _, res = dhClient.request(**dhapis['person_img_upload'], file_path_rela=ep.photo) dh_photo = res["fileUrl"] json_data.update( { "biosignatureTypeList": [3], "personBiosignatures": [{ "type": 3, "index": 1, "path": dh_photo }] } ) _, res = dhClient.request(**dhapis['person_add'], json=json_data) ep = cls.save(ep, data={'dh_id': dh_id, 'dh_photo': dh_photo}) # 开人脸卡 dh_face_card = cls.open_face_card( ep=ep, dh_id=dh_id, departmentId=departmentId, start_time=start_time, end_time=end_time) # 授予门禁权限 dh_dchannels = cls.door_auth(ep=ep) return {'dh_id': dh_id, 'dh_photo': dh_photo, 'dh_face_card': dh_face_card, 'dh_dchannels': dh_dchannels} @classmethod def open_face_card(cls, ep, dh_id, departmentId, start_time, end_time): """开人脸卡/有卡就更新卡时间 """ if ep.third_info.get('dh_face_card', None): cardNumber = ep.third_info.get('dh_face_card') # 如果有人脸卡就执行更新操作 if start_time is None: # 如果时间段未提供,跳过更新操作 pass else: startDate = start_time.strftime("%Y-%m-%d %H:%M:%S") endDate = end_time.strftime("%Y-%m-%d %H:%M:%S") json_data = { "cardNumber": cardNumber, "startDate": startDate, "endDate": endDate, "departmentId": departmentId, } _, res = dhClient.request(**dhapis['card_update'], json=json_data) return cardNumber else: _, res = dhClient.request(**dhapis['card_gen_id']) cardId = res['id'] cardNumber = str(ep.id)[3:8] + rannum(5) now = datetime.now() if start_time is None: # 如果未规定时间范围, 默认1小时 startDate = now.strftime("%Y-%m-%d %H:%M:%S") endDate = (now+timedelta(minutes=60)).strftime("%Y-%m-%d %H:%M:%S") else: startDate = start_time.strftime("%Y-%m-%d %H:%M:%S") endDate = end_time.strftime("%Y-%m-%d %H:%M:%S") json_data = { "id": cardId, "cardNumber": cardNumber, "category": 0, "cardType": 0, "personId": dh_id, "departmentId": departmentId, "startDate": startDate, "endDate": endDate } _, res = dhClient.request(**dhapis['card_add'], json=json_data) time.sleep(6) # 等待确保生成卡片 cls.save(ep, data={'dh_face_card': cardNumber}) return cardNumber @classmethod def door_auth(cls, ep: Employee, dh_dchannels: list = []): """授予门禁权限 """ from apps.third.models import TDevice dh_face_card = ep.third_info.get('dh_face_card', None) if dh_dchannels: pass else: # 查找可授予的门禁通道号 dh_dchannels = list(TDevice.objects.filter(type=TDevice.DEVICE_DCHANNEL, access_list__contains=ep.type).values_list('code', flat=True)) if dh_dchannels: details = [] for i in dh_dchannels: details.append({ "privilegeType": 1, "resouceCode": i }) json_data = { "cardNumbers": [dh_face_card], "timeQuantumId": 1, "cardPrivilegeDetails": details } is_ok, res = dhClient.request(**dhapis['card_door_authority'], json=json_data, raise_exception=False) if is_ok == 'fail' and '44999999' in res['code']: time.sleep(10) dhClient.request(**dhapis['card_door_authority'], json=json_data) elif is_ok == 'fail': raise ParseError(**res) elif is_ok == 'error': raise APIException(**DH_REQUEST_ERROR) cls.save(ep, data={'dh_dchannels': dh_dchannels}) return dh_dchannels @classmethod def save(cls, ep: Employee, data: dict): """更新third_info """ ti = ep.third_info ti.update(data) ep.third_info = ti ep.save() return ep @classmethod def swipe(cls, data: dict): """ 刷卡事件/用于记录考勤 """ from apps.vm.models import Visit from apps.rpm.models import Rpj nodeCode = data['info']['nodeCode'] id_number = data['info']['extend'].get('paperNumber', None) swip_time = data['info']['extend']['swingTime'] if id_number: device = TDevice.objects.filter(code=nodeCode).first() if device: tzinfo = tz.gettz('Asia/Shanghai') s_time_f = datetime.strptime(swip_time, "%Y-%m-%d %H:%M:%S").replace(tzinfo=tzinfo) first_time = datetime(year=s_time_f.year, month=s_time_f.month, day=s_time_f.day, hour=0, minute=0, second=0, tzinfo=tzinfo) end_time = datetime(year=s_time_f.year, month=s_time_f.month, day=s_time_f.day, hour=23, minute=59, second=59, tzinfo=tzinfo) if device.is_clock: # 如果设置为关联考勤 # myLogger.info(data['info']['extend']) if data['info']['extend']['enterOrExit'] == 1: # 如果是进门 ep = Employee.objects.filter(id_number=id_number, type__in=["employee", "remployee"]).first() # 如果是内部/相关方员工创建上班打卡记录(更新) if ep: cr_10 = ClockRecord.objects.filter( type=10, employee=ep, create_time__gte=first_time, create_time__lte=end_time).first() if cr_10: if s_time_f < cr_10.create_time: cr_10.create_time = s_time_f cr_10.trigger = 'door' cr_10.detail = data['info']['extend'] cr_10.save() else: cr_10 = ClockRecord() cr_10.type = 10 cr_10.employee = ep cr_10.trigger = 'door' cr_10.detail = data['info']['extend'] cr_10.create_time = s_time_f cr_10.save() ep.is_atwork = True ep.last_check_time = s_time_f ep.save() elif data['info']['extend']['enterOrExit'] == 2: # 如果是出门 id_number = data['info']['extend']['paperNumber'] if id_number: ep = Employee.objects.filter(id_number=id_number, type__in=["employee", "remployee"]).first() # 如果是内部/相关方员工创建下班打卡记录(更新) if ep: cr_20 = ClockRecord.objects.filter( type=20, employee=ep, create_time__gte=first_time, create_time__lte=end_time).first() if cr_20: if s_time_f > cr_20.create_time: cr_20.create_time = s_time_f cr_20.trigger = 'door' cr_20.detail = data['info']['extend'] cr_20.save() else: cr_20 = ClockRecord() cr_20.type = 20 cr_20.employee = ep cr_20.trigger = 'door' cr_20.detail = data['info']['extend'] cr_20.create_time = s_time_f cr_20.save() ep.is_atwork = False ep.last_check_time = s_time_f ep.save() # 进行相关方/访客项目更新 Visit.objects.filter(state=Visit.V_ENTER, visitors__employee__id_number=id_number).update( state=Visit.V_WORKING) Rpj.objects.filter(state=Rpj.RPJ_ENTER, remployees__employee__id_number=id_number).update( state=Rpj.RPJ_WORKING) # 此处可触发安全帽事件逻辑