mirror of https://github.com/jumpserver/jumpserver
29 lines
684 B
Python
29 lines
684 B
Python
# ~*~ coding: utf-8 ~*~
|
|
#
|
|
|
|
import chardet
|
|
import unicodecsv
|
|
|
|
from .base import BaseFileParser
|
|
|
|
|
|
class CSVFileParser(BaseFileParser):
|
|
|
|
media_type = 'text/csv'
|
|
|
|
@staticmethod
|
|
def _universal_newlines(stream):
|
|
"""
|
|
保证在`通用换行模式`下打开文件
|
|
"""
|
|
for line in stream.splitlines():
|
|
yield line
|
|
|
|
def generate_rows(self, stream_data):
|
|
detect_result = chardet.detect(stream_data)
|
|
encoding = detect_result.get("encoding", "utf-8")
|
|
lines = self._universal_newlines(stream_data)
|
|
csv_reader = unicodecsv.reader(lines, encoding=encoding)
|
|
for row in csv_reader:
|
|
yield row
|