Merge branch 'develop' of https://e.coding.net/ctcdevteam/hberp/hberp into develop

hb_server/apps/hrm/serializers.py

@@ -20,14 +20,15 @@ class EmployeeNotWorkRemarkSerializer(ModelSerializer):
         fields = ['not_work_remark']
 class FaceLoginSerializer(serializers.Serializer):
     base64 = serializers.CharField()
+    tolerance = serializers.FloatField(required=False, default=0.42)
 
 class FaceLoginPathSerializer(serializers.Serializer):
     path = serializers.CharField()
-    tolerance = serializers.FloatField(required=False, default=0.45)
+    tolerance = serializers.FloatField(required=False, default=0.42)
 
 class FaceClockCreateSerializer(serializers.Serializer):
     base64 = serializers.CharField()
-    tolerance = serializers.FloatField(required=False, default=0.45)
+    tolerance = serializers.FloatField(required=False, default=0.42)
 
 class ClockRecordListSerializer(serializers.ModelSerializer):
     create_by_ = UserSimpleSerializer(source='create_by', read_only=True)

hb_server/apps/hrm/services.py

@@ -10,7 +10,7 @@ from django.core.cache import cache
 class HRMService:
 
     @classmethod
-    def face_compare_from_path(cls, path, tolerance=0.45):
+    def face_compare_from_path(cls, path, tolerance=0.42):
         filepath = settings.BASE_DIR +path
         try:
             unknown_picture = face_recognition.load_image_file(filepath)
@@ -43,7 +43,7 @@ class HRMService:
             return None, '匹配多张人脸:' + user_name_str
 
     @classmethod
-    def face_compare_from_base64(cls, base64_data, tolerance=0.45):
+    def face_compare_from_base64(cls, base64_data, tolerance=0.42):
         filename = str(uuid.uuid4())
         filepath = settings.BASE_DIR +'/temp/' + filename +'.png'
         with open(filepath, 'wb') as f:

hb_server/apps/hrm/views.py

@@ -109,7 +109,7 @@ class ClockRecordViewSet(CreateModelMixin, ListModelMixin, GenericViewSet):
         if 8<=now_local.hour<=17:
             base64_data = base64.urlsafe_b64decode(tran64(
                 request.data.get('base64').replace(' ', '+')))
-            user, msg = HRMService.face_compare_from_base64(base64_data)
+            user, msg = HRMService.face_compare_from_base64(base64_data, request.data.get('tolerance', 0.42))
             if user:
                 ins, created = ClockRecord.objects.get_or_create(
                     create_by = user, create_time__hour__range = [8,18], 
@@ -159,8 +159,8 @@ class FaceLogin(CreateAPIView):
         """
         人脸识别登录
         """
-        base64_data = base64.urlsafe_b64decode(tran64(request.data.get('base64').replace(' ', '+')), request.data.get('tolerance', 0.45))
-        user, msg = HRMService.face_compare_from_base64(base64_data)
+        base64_data = base64.urlsafe_b64decode(tran64(request.data.get('base64').replace(' ', '+')))
+        user, msg = HRMService.face_compare_from_base64(base64_data, request.data.get('tolerance', 0.42))
         if user:
             refresh = RefreshToken.for_user(user)
             # 可设为在岗
@@ -198,7 +198,7 @@ class FacePathLogin(CreateAPIView):
         """
         人脸识别登录-文件地址
         """
-        user, msg = HRMService.face_compare_from_path(request.data.get('path'), request.data.get('tolerance', 0.45))
+        user, msg = HRMService.face_compare_from_path(request.data.get('path'), request.data.get('tolerance', 0.42))
         if user:
             refresh = RefreshToken.for_user(user)
             # 可设为在岗

hb_server/apps/pm/serializers.py

@@ -43,12 +43,19 @@ class SubProductionPlanListSerializer(DynamicFieldsSerializerMixin, serializers.
     leader_1_ = UserSimpleSerializer(source='leader_1', read_only=True)
     leader_2_ = UserSimpleSerializer(source='leader_2', read_only=True)
     leader_3_ = UserSimpleSerializer(source='leader_3', read_only=True)
+    first_test_ = serializers.SerializerMethodField()
     class Meta:
         model=SubProductionPlan
         fields = '__all__'
     
     def get_plan_product_(self, obj):
         return MaterialSimpleSerializer(instance=obj.production_plan.product).data
+    
+    def get_first_test_(self, obj):
+        tr = obj.first_test
+        if obj.first_test:
+            return {'id':tr.id, 'is_submited':tr.is_submited}
+        return None