# -*- coding: utf-8 -*-
|
#
|
# @author:
|
# @date:
|
# @version:
|
# @description:
|
|
from fastapi import FastAPI, UploadFile
|
import asyncio
|
import os
|
from knowledgebase import utils
|
from knowledgebase.log import Log
|
|
UPLOAD_FILE_PATH = "upload_files"
|
if not utils.file_exists(UPLOAD_FILE_PATH):
|
os.mkdir(UPLOAD_FILE_PATH)
|
|
app = FastAPI(swagger_ui_parameters={"syntaxHighlight": False})
|
|
|
@app.get("/")
|
async def read_root():
|
return {"Hello": "World"}
|
|
|
# 上传文件，保存文件到本地，返回md5
|
@app.post("/UploadFile/")
|
async def upload_file(file: UploadFile):
|
data = await file.read()
|
md5 = await utils.get_md5_async(data)
|
ext = os.path.splitext(file.filename)[1]
|
save_path = f'./upload_files/{md5}{ext}'
|
Log.info(f'上传文件：{file.filename}，md5: {md5}')
|
|
await utils.save_to_file_async(data, save_path)
|
return {"md5": md5}
|
