diff --git a/hb_server/apps/hrm/views.py b/hb_server/apps/hrm/views.py index e41ee48..dafe59c 100644 --- a/hb_server/apps/hrm/views.py +++ b/hb_server/apps/hrm/views.py @@ -1,3 +1,4 @@ +from functools import update_wrapper from django.shortcuts import render from django.utils import timezone from rest_framework.response import Response @@ -5,6 +6,7 @@ from rest_framework.viewsets import ModelViewSet, GenericViewSet from rest_framework.mixins import UpdateModelMixin, RetrieveModelMixin, CreateModelMixin, ListModelMixin from apps.hrm.filters import ClockRecordFilterSet from apps.hrm.services import HRMService +from apps.hrm.tasks import update_all_user_facedata_cache from apps.system.mixins import CreateUpdateModelAMixin, OptimizationMixin from apps.hrm.models import ClockRecord, Employee from apps.hrm.serializers import ClockRecordListSerializer, EmployeeSerializer, FaceClockCreateSerializer, FaceLoginSerializer @@ -42,6 +44,8 @@ class EmployeeViewSet(CreateUpdateModelAMixin, OptimizationMixin, UpdateModelMix f_l = HRMService.get_facedata_from_img(photo) if f_l: serializer.save(update_by=request.user, face_data = f_l) + # 更新人脸缓存 + update_all_user_facedata_cache.delay() return Response() return Response('头像识别失败', status=status.HTTP_400_BAD_REQUEST) serializer.save(update_by=request.user)