mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-25 14:50:24 +00:00
43b5e97b95
* refactor(drf_renderer): 添加 ExcelRenderer 支持导出excel文件格式; 优化CSVRenderer, 抽象 BaseRenderer * perf(renderer): 支持导出资源详情 * refactor(drf_parser): 添加 ExcelParser 支持导入excel文件格式; 优化CSVParser, 抽象 BaseParser * refactor(drf_parser): 添加 ExcelParser 支持导入excel文件格式; 优化CSVParser, 抽象 BaseParser 2 * perf(renderer): 捕获renderer处理异常 * perf: 添加excel依赖包 * perf(drf): 优化导入导出错误日志 * perf: 添加依赖包 pyexcel-io==0.6.4 * perf: 添加依赖包pyexcel-xlsx==0.6.0 * feat: 修改drf/renderer&parser变量命名 * feat: 修改drf/renderer的bug * feat: 修改drf/renderer&parser变量命名 Co-authored-by: Bai <bugatti_it@163.com>
31 lines
645 B
Python
31 lines
645 B
Python
# ~*~ coding: utf-8 ~*~
|
|
#
|
|
|
|
import codecs
|
|
import unicodecsv
|
|
from six import BytesIO
|
|
|
|
from .base import BaseFileRenderer
|
|
|
|
|
|
class CSVFileRenderer(BaseFileRenderer):
|
|
media_type = 'text/csv'
|
|
format = 'csv'
|
|
|
|
writer = None
|
|
buffer = None
|
|
|
|
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):
|
|
self.writer.writerow(row)
|
|
|
|
def get_rendered_value(self):
|
|
value = self.buffer.getvalue()
|
|
return value
|