浏览代码

no message

libushang 4 周之前
父节点
当前提交
3d5b678c6f
共有 1 个文件被更改,包括 42 次插入1 次删除
  1. 42 1
      routers/api/resourceProvison/MaterialReserveManagement/material.py

+ 42 - 1
routers/api/resourceProvison/MaterialReserveManagement/material.py

@@ -21,6 +21,10 @@ from utils.resource_provision_util import *
 from common.barcode import create_bar,create_qr
 import json
 import traceback
+import xlrd
+
+# 目录在文档上传接口写死
+UPLOAD_mergefile_PATH = '/data/upload/mergefile'
 
 router = APIRouter()
 
@@ -686,4 +690,41 @@ async def gettreeselect(request: Request,material_name: str = Query(None, descri
         "code": 200,
         "msg": "操作成功",
         "data": data
-    }
+    }
+
+# 导入
+@router.post('/import')
+async def import_doc(
+    request: Request, 
+    db: Session = Depends(get_db), 
+    body = Depends(remove_xss_json), 
+    user_id = Depends(valid_access_token)
+):
+    
+    try:
+        filename = body['filename']
+        if len(filename) == 0:
+            raise Exception()
+        
+        file = filename[0]
+        url = file['url']
+        file_path = f"{UPLOAD_mergefile_PATH}/uploads/{url}"
+        file_path = os.path.abspath(file_path)
+        print(file_path)
+        
+        book = xlrd.open_workbook(file_path)
+        sheet = book.sheet_by_index(0)
+
+        data = []
+
+        return {
+            'code': 200,
+            'msg': '导入成功'
+        }
+    except Exception:
+        traceback.print_exc()
+        return {
+            'code': 500,
+            'msg': '导入发生异常'
+        }
+