github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-19 18:34:28 +00:00
Code Issues Projects Releases Wiki Activity

fix: 给 view 增加 export 属性 (#10495)

Browse Source 
Co-authored-by: feng <1304903146@qq.com>
This commit is contained in:
fit2bot
2023-05-18 18:33:22 +08:00
committed by GitHub
parent 0596b74fa1
commit 2981bfffb1
2 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
apps/common/drf/renders/base.py
View File

@@ -184,19 +184,16 @@ class BaseFileRenderer(BaseRenderer):
self.write_column_titles(column_titles)
self.write_rows(rows)
self.after_render()
value = self.compress_into_zip_file(view, request, response)
value = self.get_rendered_value()
if getattr(view, 'export_as_zip', False) and self.template == 'export':
value = self.compress_into_zip_file(value, request, response)
except Exception as e:
logger.debug(e, exc_info=True)
value = 'Render error! ({})'.format(self.media_type).encode('utf-8')
return value
return value
def compress_into_zip_file(self, view, request, response):
value = self.get_rendered_value()
from accounts.models import Account
if str(view.model) not in (str(Account),) or self.template != 'export':
return value
def compress_into_zip_file(self, value, request, response):
filename_pattern = re.compile(r'filename="([^"]+)"')
content_disposition = response['Content-Disposition']
match = filename_pattern.search(content_disposition)