From deedd49dc5458ddf70e6e4538b405bae2cdf158c Mon Sep 17 00:00:00 2001 From: Bai Date: Tue, 15 Dec 2020 16:33:51 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=91=BD=E4=BB=A4?= =?UTF-8?q?=E8=AE=B0=E5=BD=95=E5=AF=BC=E5=87=BAexcel=E6=96=87=E4=BB=B6?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E6=9C=AA=E5=AE=9A=E4=B9=89=E7=9A=84=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/common/drf/renders/base.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/apps/common/drf/renders/base.py b/apps/common/drf/renders/base.py index deac735cc..57f035f26 100644 --- a/apps/common/drf/renders/base.py +++ b/apps/common/drf/renders/base.py @@ -27,11 +27,13 @@ class BaseFileRenderer(BaseRenderer): def set_response_disposition(self, response): serializer = self.serializer if response and hasattr(serializer, 'Meta') and hasattr(serializer.Meta, "model"): - model_name = serializer.Meta.model.__name__.lower() - now = datetime.now().strftime("%Y-%m-%d_%H-%M-%S") - filename = "{}_{}.{}".format(model_name, now, self.format) - disposition = 'attachment; filename="{}"'.format(filename) - response['Content-Disposition'] = disposition + filename_prefix = serializer.Meta.model.__name__.lower() + else: + filename_prefix = 'download' + now = datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + filename = "{}_{}.{}".format(filename_prefix, now, self.format) + disposition = 'attachment; filename="{}"'.format(filename) + response['Content-Disposition'] = disposition def get_rendered_fields(self): fields = self.serializer.fields