perf:用户导入功能优化

This commit is contained in:
ktianc 2023-07-16 21:40:10 +08:00
parent 0d0b3a7308
commit 0807d9c02a
3 changed files with 10 additions and 5 deletions

View File

@ -77,6 +77,7 @@ const handleImport = async () => {
}
const downloadTemplate = async () => {
ElMessage.info('正在下载请稍等!')
const res = await getImportTemplateApi()
if (res) {
const a = document.createElement('a')
@ -90,6 +91,7 @@ const downloadTemplate = async () => {
}
const downloadErrorFile = async (row: Recordable) => {
ElMessage.info('正在下载请稍等!')
const a = document.createElement('a')
a.style.display = 'none'
a.href = row.error_url

View File

@ -22,7 +22,7 @@ class User(BaseModel):
nickname: Optional[str] = None
avatar: Optional[str] = None
is_active: Optional[bool] = True
is_staff: Optional[bool] = False
is_staff: Optional[bool] = True
gender: Optional[str] = "0"
is_wx_server_openid: Optional[bool] = False

View File

@ -57,15 +57,18 @@ class ImportManage:
if file.content_type not in cls.file_type:
raise CustomException(msg="文件类型必须为xlsx类型", code=status.HTTP_ERROR)
async def get_table_data(self) -> None:
async def get_table_data(self, header_row: int = 1, data_row: int = 2) -> None:
"""
获取表格数据与表头
:param header_row: 表头在第几行
:param data_row: 数据开始行
"""
self.__filename = await FileManage.save_tmp_file(self.file)
es = ExcelManage()
es.open_sheet(self.__filename, read_only=True)
self.__table_header = es.get_header(1, len(self.headers), asterisk=True)
self.__table_data = es.readlines(min_row=2, max_col=len(self.headers))
es.open_sheet(file=self.__filename, read_only=True)
self.__table_header = es.get_header(header_row, len(self.headers), asterisk=True)
self.__table_data = es.readlines(min_row=data_row, max_col=len(self.headers))
es.close()
def check_table_data(self) -> None: