diff --git a/main.py b/main.py index 2125aee..d4bfa66 100644 --- a/main.py +++ b/main.py @@ -61,39 +61,6 @@ async def admin(): ) -@app.get(f'/{settings.ADMIN_ADDRESS}/files', dependencies=[Depends(admin_required)]) -async def get_files(): - files = [] - for i, j, k in os.walk(f'{LOCAL_ROOT}'): - files.extend([{'name': _.split('/')[-1], 'path': i} for _ in k]) - return {'data': files} - - -@app.post(f'/{settings.ADMIN_ADDRESS}/files', dependencies=[Depends(admin_required)]) -async def share_file(name=Form(...), path=File(...), s: AsyncSession = Depends(get_session)): - file_path = path + f'/{name}' - key = uuid.uuid4().hex - code = await get_code(s) - now = datetime.datetime.now() - path = f"{settings.LOCAL_ROOT}/upload/{now.year}/{now.month}/{now.day}/" - if not os.path.exists(path): - os.makedirs(path) - text = f"{path}/{f'{key}{name}'}" - exp_time = datetime.datetime.now() + datetime.timedelta(days=1) - s.add(Codes(code=code, text=text.replace(f'{settings.LOCAL_ROOT}/', ''), type='file', name=name, count=-1, - exp_time=exp_time, key=key)) - await s.commit() - shutil.move(file_path, text) - return { - 'detail': '分享成功!请在24小时内使用分享使用!', - 'data': { - 'code': code, - 'text': text, - 'name': name, - } - } - - @app.post(f'/{settings.ADMIN_ADDRESS}', dependencies=[Depends(admin_required)], description='查询数据库列表') async def admin_post(page: int = Form(default=1), size: int = Form(default=10), s: AsyncSession = Depends(get_session)): infos = (await s.execute(select(Codes).offset((page - 1) * size).limit(size))).scalars().all() diff --git a/settings.py b/settings.py index 4baec6b..8be69a6 100644 --- a/settings.py +++ b/settings.py @@ -19,8 +19,6 @@ class Settings: DATABASE_URL = config('DATABASE_URL', cast=str, default=f"sqlite+aiosqlite:///{DATABASE_FILE}") # 数据存储文件夹,文件就不暴露在静态资源里面了 DATA_ROOT = './data/' + config('DATA_ROOT', cast=str, default=f"static") - # 本地文件夹 - LOCAL_ROOT = './data/' + config('LOCAL_ROOT', cast=str, default=f"local") # 静态文件夹URL STATIC_URL = config('STATIC_URL', cast=str, default="/static") # 开启上传 diff --git a/templates/admin.html b/templates/admin.html index 65ca06c..2211409 100644 --- a/templates/admin.html +++ b/templates/admin.html @@ -73,25 +73,6 @@ > - - - - - - -
-
文件名:${ file.name }
-
路径:${ file.path }
-
-
- - 分享 - -
-
-
-