feat: 证书修改背景

This commit is contained in:
caoqianming 2023-09-01 14:09:24 +08:00
parent 1378eb0de2
commit 66abf6b402
1 changed files with 15 additions and 16 deletions

View File

@ -35,10 +35,7 @@ class CertificateViewSet(CreateUpdateCustomMixin, ModelViewSet):
return [AllowAny()]
return super().get_permissions()
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:
def make_img_x(self, instance):
titles = []
if instance.是否内审员:
titles.append('内审员')
@ -50,6 +47,12 @@ class CertificateViewSet(CreateUpdateCustomMixin, ModelViewSet):
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:
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()