Function upload_files

pstext/main.py:61–86

(background_tasks: BackgroundTasks, db: Session = Depends(get_db), file: UploadFile = File(...))

@application.post("/")
async def upload_files(background_tasks: BackgroundTasks, db: Session = Depends(get_db), file: UploadFile = File(...)):
    filename = file.filename
    title = filename.split('.')[0]
    ext = filename.split('.')[1]

    # check if uploading file existed already
    db_title = crud.get_audio_by_title(db, title=title)
    if db_title:
        raise HTTPException(status_code=400, detail="Audio already uploaded")    # 将传过来的文件保存

    # save file
    out_file_path = audio_raw_path / file.filename
    async with aiofiles.open(out_file_path, 'wb') as out_file:
        content = await file.read()  # async read
        size = "{}MB".format(round(len(content)/(1024*1024),3))
        await out_file.write(content)  # async write
    data = {"title": title,
            "size": size,
            "extension": ext
            }
    #create audio info
    result = crud.create_audio_info(db = db, data = data)
    # background tasks for ffmpeg file
    background_tasks.add_task(bg_task, out_file_path, title, db)

    return result


@application.get("/", response_model=schemas.ReadAudioData)