From 9192d9ffdaf82941bcde7264f225ccd1d5e8c7f3 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Wed, 19 Apr 2023 19:20:52 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BA=BA=E5=91=98=E5=AF=BC=E5=87=BAexce?=
 =?UTF-8?q?l=E4=BF=A1=E6=81=AFbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/hrm/views.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/apps/hrm/views.py b/apps/hrm/views.py
index 728640fb..f42dbbc8 100755
--- a/apps/hrm/views.py
+++ b/apps/hrm/views.py
@@ -223,12 +223,14 @@ class EmployeeViewSet(CustomModelViewSet):
         """
         field_data = ['人员类型', '人员', '手机号', '身份证号', '所属部门', '在职状态', '定位卡号']
         queryset = self.filter_queryset(self.get_queryset())
+        if queryset.count() > 1000:
+            raise ParseError('数据量超过1000,请筛选后导出')
         odata = EmployeeSerializer(queryset, many=True).data
         # 处理数据
         data = []
         for i in odata:
             data.append(
-                [epTypeOptions['type'],
+                [epTypeOptions[i['type']],
                  i['name'],
                  i['phone'],
                  i['id_number'],
@@ -236,7 +238,7 @@ class EmployeeViewSet(CustomModelViewSet):
                  epStateOptions[i['job_state']],
                  i['blt_'].get('code', '') if 'blt_' in i else '']
             )
-        return Response({'path': export_excel(field_data, data, '打卡记录')})
+        return Response({'path': export_excel(field_data, data, '人员信息')})
 
 
 class ClockRecordViewSet(ListModelMixin, CustomGenericViewSet):