Explorar o código

修改Excel脚本启动请求可能会出错的问题

zhangxiao hai 1 ano
pai
achega
a1cda8f0a4
Modificáronse 1 ficheiros con 6 adicións e 0 borrados
  1. 6 0
      app/api/excel.py

+ 6 - 0
app/api/excel.py

@@ -70,6 +70,12 @@ async def upload_file(files: list[UploadFile] = File(...)):
 @router.websocket("/ws/excel")
 async def ws_excel(websocket: WebSocket):
     await websocket.accept()
+
+    create_dir_if_not_exists(SOURCE_FILES_PATH)
+    create_dir_if_not_exists(EXCEL_FILES_PATH)
+    clear_directory(SOURCE_FILES_PATH)
+    clear_directory(EXCEL_FILES_PATH)
+
     while True:
         data = await websocket.receive_text()
         try: