实现文件上传功能

main.py

@@ -1,9 +1,8 @@
 import os
-import io
 import uvicorn
 
 from pathlib                    import Path
-from fastapi                    import FastAPI, Header
+from fastapi                    import FastAPI, UploadFile, File, Header
 from fastapi.responses          import StreamingResponse
 from fastapi.middleware.cors    import CORSMiddleware
 from fastapi.staticfiles        import StaticFiles
@@ -11,20 +10,28 @@ from config                     import ServerConfig
 
 
 app = FastAPI()
-app.mount('/home', StaticFiles(directory="static", html=True), name="static")
+app.mount('/home', StaticFiles(directory='static', html=True), name='static')
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=['*'],
     allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"]
+    allow_methods=['*'],
+    allow_headers=['*'],
 )
 
 
+@app.post('/upload')
+async def upload_file(file: UploadFile = File(...)):
+    video_path = os.path.join(ServerConfig.video_upload_path, file.filename + '.mp4')
+    with open(video_path, 'wb') as video:
+        video.write(await file.read())
+    return {'status': 200, 'id': os.path.splitext(file.filename)[0]}
+
+
 @app.get("/fetch/{id}")
 async def fetch_file(id: str, range: str = Header(None)) -> StreamingResponse:
-    video_path: Path = Path(os.path.join(ServerConfig.video_play_path, id + '.mp4'))
-    video_size: int  = video_path.stat().st_size
+    video_path = Path(os.path.join(ServerConfig.video_play_path, id + '.mp4'))
+    video_size = video_path.stat().st_size
     start      = 0
     end        = video_size - 1
     chunk_size = min(video_size, ServerConfig.video_chunk_size)
@@ -36,7 +43,7 @@ async def fetch_file(id: str, range: str = Header(None)) -> StreamingResponse:
         start, end  = range.replace('bytes=', '').split('-')
         start       = int(start)
         end         = int(end) if end else video_size - 1
-        chunk_size: int  = min(end - start + 1, ServerConfig.video_chunk_size)
+        chunk_size  = min(end - start + 1, ServerConfig.video_chunk_size)
         headers['Content-Range']       = f'bytes {start}-{end}/{video_size}'
         headers['Accept-Ranges']       = 'bytes'
         headers['Content-Disposition'] = 'inline'

requirements.txt

@@ -6,3 +6,4 @@ opencv-python==4.9.0.80
 opencv-contrib-python==4.9.0.80
 uvicorn==0.29.0
 fastapi==0.110.2
+python-multipart==0.0.9

static/js/index.js

@@ -26,7 +26,6 @@ function onSelectVideoButtonClicked() {
 function onUploadVideoChanged(files) {
     for (var index = 0; index < files.length; index++) {
         var file = files[index];
-        console.log(index);
         if (file.type != 'video/mp4') {
             errorModal('错误', `文件${file.name}的格式错误`);
             return;
@@ -70,6 +69,7 @@ function onFileUploadFinished(index, file, data) {
     $(`#stat-label-${index}`).html('正在提取');
     $(`#stat-prog-${index}`).css({'width': '0%'});
     $(`#stat-prog-${index}`).html('0%');
+    
 }
 
 /**
@@ -114,7 +114,11 @@ function uploadFile(index, file) {
         processData: false,                         // 不处理发送的数据
         contentType: false,                         // 不设置内容类型
         success: function(data) {
+            if (data.status != 200) {
+                //TODO: 添加错误处理
+            } else {
                 onFileUploadFinished(index, file, data);
+            }
         },
         error: function(xhr, status, error) {
             if (xhr.status == 0) {
@@ -187,7 +191,3 @@ $(document).ready(function() {
         onUploadVideoChanged(files);
     });
 });
-
-$(document).ready(function() {
-    playFile(0, null, '123');
-});