mirror of https://github.com/jumpserver/jumpserver
18 lines
472 B
Python
18 lines
472 B
Python
import pyexcel
|
|
|
|
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
|