实现文件上传功能

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,35 +10,43 @@ 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
-    start            = 0
-    end              = video_size - 1
-    chunk_size       = min(video_size, ServerConfig.video_chunk_size)
-    headers          = { 
-                         'Content-Type': 'video/mp4', 
-                         'Content-Disposition': f'attachment; filename="{id}.mp4"',
-                       }
+    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)
+    headers    = { 
+                    'Content-Type': 'video/mp4', 
+                    'Content-Disposition': f'attachment; filename="{id}.mp4"',
+                 }
     if range:
-        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)
-        headers['Content-Range']        = f'bytes {start}-{end}/{video_size}'
-        headers['Accept-Ranges']        = 'bytes'
-        headers['Content-Disposition']  = 'inline'
+        start, end  = range.replace('bytes=', '').split('-')
+        start       = int(start)
+        end         = int(end) if end else video_size - 1
+        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'
     def file_reader():
         with open(video_path, 'rb') as video:
             video.seek(start)

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) {
-            onFileUploadFinished(index, file, 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');
-});