90 lines
3.5 KiB
Python
90 lines
3.5 KiB
Python
|
from datetime import datetime
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import xlrd
|
||
|
|
from app.utils.LanguageUtils import LANG_VIEWS_USE_LANG_T
|
||
|
|
|
||
|
|
|
||
|
|
class UploadUtils():
|
||
|
|
"""文件上传工具类"""
|
||
|
|
|
||
|
|
# 上传摄像头Excel文件
|
||
|
|
def upload_camera_xlsx(self, file, upload_dir, lang=None):
|
||
|
|
__ret = False
|
||
|
|
__msg = LANG_VIEWS_USE_LANG_T(lang, "msg_unknown_error")
|
||
|
|
__data = []
|
||
|
|
|
||
|
|
file_name = file.name
|
||
|
|
file_content_type = file.content_type
|
||
|
|
if 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet' == file_content_type and file_name.endswith(".xlsx"):
|
||
|
|
filename_dir = datetime.now().strftime("%Y%m%d%H%M%S") + "_" + file_name
|
||
|
|
abs_filedir = os.path.join(upload_dir, filename_dir)
|
||
|
|
if not os.path.exists(abs_filedir):
|
||
|
|
os.makedirs(abs_filedir)
|
||
|
|
|
||
|
|
abs_filepath = os.path.join(abs_filedir, file_name)
|
||
|
|
f = open(abs_filepath, 'wb')
|
||
|
|
f.write(file.read())
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
# 读取excel
|
||
|
|
wb = xlrd.open_workbook(abs_filepath)
|
||
|
|
sheet = wb.sheet_by_index(0)
|
||
|
|
|
||
|
|
if sheet.ncols in (8, 9, 10): # 兼容8/9/10列
|
||
|
|
for row in range(sheet.nrows):
|
||
|
|
if row > 0:
|
||
|
|
try:
|
||
|
|
row_cols = sheet.row_values(row)
|
||
|
|
|
||
|
|
code = str(row_cols[0]).strip()
|
||
|
|
nickname = str(row_cols[1]).strip()
|
||
|
|
pull_stream_url = str(row_cols[2]).strip()
|
||
|
|
pull_stream_ip = str(row_cols[3]).strip()
|
||
|
|
|
||
|
|
try:
|
||
|
|
pull_stream_port = int(row_cols[4])
|
||
|
|
except:
|
||
|
|
pull_stream_port = 554
|
||
|
|
|
||
|
|
username = str(row_cols[5]).strip()
|
||
|
|
password = str(row_cols[6]).strip()
|
||
|
|
remark = str(row_cols[7]).strip()
|
||
|
|
|
||
|
|
try:
|
||
|
|
is_audio = int(row_cols[8])
|
||
|
|
except:
|
||
|
|
is_audio = 0
|
||
|
|
try:
|
||
|
|
camera_device_id = str(row_cols[9]).strip()
|
||
|
|
except:
|
||
|
|
camera_device_id = code
|
||
|
|
|
||
|
|
d = {
|
||
|
|
'code': code,
|
||
|
|
'nickname': nickname,
|
||
|
|
'pull_stream_url': pull_stream_url,
|
||
|
|
'pull_stream_ip': pull_stream_ip,
|
||
|
|
'pull_stream_port': pull_stream_port,
|
||
|
|
'username': username,
|
||
|
|
'password': password,
|
||
|
|
'remark': remark,
|
||
|
|
'is_audio': is_audio,
|
||
|
|
'camera_device_id': camera_device_id,
|
||
|
|
}
|
||
|
|
__data.append(d)
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
__ret = True
|
||
|
|
__msg = LANG_VIEWS_USE_LANG_T(lang, "msg_success")
|
||
|
|
else:
|
||
|
|
__msg = LANG_VIEWS_USE_LANG_T(lang, "upload_xlsx_cols_incorrect")
|
||
|
|
|
||
|
|
if os.path.exists(abs_filedir):
|
||
|
|
shutil.rmtree(abs_filedir)
|
||
|
|
else:
|
||
|
|
__msg = LANG_VIEWS_USE_LANG_T(lang, "upload_xlsx_format_incorrect")
|
||
|
|
|
||
|
|
return __ret, __msg, __data
|