# ~*~ 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
