mirror of https://github.com/jumpserver/jumpserver
19 lines
525 B
Python
19 lines
525 B
Python
import pyexcel
|
|
from django.utils.translation import gettext as _
|
|
|
|
from .base import BaseFileParser
|
|
|
|
|
|
class ExcelFileParser(BaseFileParser):
|
|
media_type = 'text/xlsx'
|
|
|
|
def generate_rows(self, stream_data):
|
|
try:
|
|
workbook = pyexcel.get_book(file_type='xlsx', file_content=stream_data)
|
|
except Exception:
|
|
raise Exception(_('Invalid excel file'))
|
|
# 默认获取第一个工作表sheet
|
|
sheet = workbook.sheet_by_index(0)
|
|
rows = sheet.rows()
|
|
return rows
|