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

fix: Fixed csv file export for 0 chars is not appear

Browse Source
This commit is contained in:
Bai
2024-04-28 17:51:00 +08:00
committed by 老广
parent 7408ed0f03
commit 38175d6b57
3 changed files with 36 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
apps/common/drf/renders/csv.py
View File

@@ -6,31 +6,36 @@ import unicodecsv
from six import BytesIO
from .base import BaseFileRenderer
from ..const import CSV_FILE_ESCAPE_CHARS
class CSVFileRenderer(BaseFileRenderer):
media_type = 'text/csv'
format = 'csv'
writer = None
buffer = None
escape_chars = tuple(CSV_FILE_ESCAPE_CHARS)
def initial_writer(self):
csv_buffer = BytesIO()
csv_buffer.write(codecs.BOM_UTF8)
csv_writer = unicodecsv.writer(csv_buffer, encoding='utf-8')
self.buffer = csv_buffer
self.writer = csv_writer
def write_row(self, row):
def __render_row(self, row):
row_escape = []
for d in row:
if isinstance(d, str) and d.strip().startswith(('=', '@')):
if isinstance(d, str) and d.strip().startswith(self.escape_chars):
d = "'{}".format(d)
row_escape.append(d)
else:
row_escape.append(d)
self.writer.writerow(row_escape)
row_escape.append(d)
return row_escape
def write_row(self, row):
row = self.__render_row(row)
self.writer.writerow(row)
def get_rendered_value(self):
value = self.buffer.getvalue()