From 66abf6b402197f85d150d0d41593967b902c5e6b Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 1 Sep 2023 14:09:24 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=AF=81=E4=B9=A6=E4=BF=AE=E6=94=B9?=
 =?UTF-8?q?=E8=83=8C=E6=99=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/apps/edu/views.py | 31 +++++++++++++++----------------
 1 file changed, 15 insertions(+), 16 deletions(-)

diff --git a/server/apps/edu/views.py b/server/apps/edu/views.py
index 9c53baa..e62d5b7 100644
--- a/server/apps/edu/views.py
+++ b/server/apps/edu/views.py
@@ -35,21 +35,24 @@ class CertificateViewSet(CreateUpdateCustomMixin, ModelViewSet):
             return [AllowAny()]
         return super().get_permissions()
 
+    def make_img_x(self, instance):
+        titles = []
+        if instance.是否内审员:
+            titles.append('内审员')
+        if instance.是否授权签字人:
+            titles.append('授权签字人')
+        if instance.是否质量负责人:
+            titles.append('质量负责人')
+        if instance.是否最高管理者:
+            titles.append('最高管理者')
+        instance.证书地址 = make_img(instance.证书编号, instance.姓名, '、'.join(titles))
+        instance.save()
+
     def retrieve(self, request, *args, **kwargs):
         regen_img = request.query_params.get('regen_img', False)
         instance = self.get_object()
         if not instance.证书地址 or regen_img:
-            titles = []
-            if instance.是否内审员:
-                titles.append('内审员')
-            if instance.是否授权签字人:
-                titles.append('授权签字人')
-            if instance.是否质量负责人:
-                titles.append('质量负责人')
-            if instance.是否最高管理者:
-                titles.append('最高管理者')
-            instance.证书地址 = make_img(instance.证书编号, instance.姓名, '、'.join(titles))
-            instance.save()
+            self.make_img_x(instance)
         serializer = self.get_serializer(instance)
         return Response(serializer.data)
     
@@ -86,9 +89,5 @@ class CertificateViewSet(CreateUpdateCustomMixin, ModelViewSet):
             data = {}
             data = self.make_data(data, sheet, i)
             obj, created = Certificate.objects.update_or_create(defaults=data, 证书编号=data['证书编号'])
-            # if created:
-            #     print(f'已创建证书-{data["姓名"]}-{data["证书编号"]}')
-            # else:
-            #     print(f'已更新证书-{data["姓名"]}-{data["证书编号"]}')
             i = i + 1
-        return Response()
\ No newline at end of file
+        return Response()