From d7772701ae20bed4e6ab631768b010c9dcfb8884 Mon Sep 17 00:00:00 2001 From: Lan Date: Sat, 5 Jul 2025 00:02:04 +0800 Subject: [PATCH] feat:code style --- apps/base/utils.py | 6 +----- apps/base/views.py | 8 +------- 2 files changed, 2 insertions(+), 12 deletions(-) diff --git a/apps/base/utils.py b/apps/base/utils.py index 961c399..3f9bdbe 100644 --- a/apps/base/utils.py +++ b/apps/base/utils.py @@ -22,10 +22,8 @@ async def get_file_path_name(file: UploadFile) -> Tuple[str, str, str, str, str] filename = await sanitize_filename(unquote(file.filename)) # 使用 UUID 作为子目录名 base_path = f"share/data/{today.strftime('%Y/%m/%d')}/{file_uuid}" - # 如果设置了存储路径，将其添加到基础路径中 path = f"{storage_path}/{base_path}" if storage_path else base_path - prefix, suffix = os.path.splitext(filename) # 保持原始文件名 save_path = f"{path}/{filename}" @@ -43,9 +41,7 @@ async def get_chunk_file_path_name(file_name: str, upload_id: str) -> Tuple[str, return path, suffix, prefix, file_name, save_path -async def get_expire_info( - expire_value: int, expire_style: str -) -> Tuple[Optional[datetime.datetime], int, int, str]: +async def get_expire_info(expire_value: int, expire_style: str) -> Tuple[Optional[datetime.datetime], int, int, str]: """获取过期信息""" expired_count, used_count = -1, 0 now = datetime.datetime.now() diff --git a/apps/base/views.py b/apps/base/views.py index 2744c70..6bdbd35 100644 --- a/apps/base/views.py +++ b/apps/base/views.py @@ -64,18 +64,12 @@ async def share_file( ip: str = Depends(ip_limit["upload"]), ): await validate_file_size(file, settings.uploadSize) - if expire_style not in settings.expireStyle: raise HTTPException(status_code=400, detail="过期时间类型错误") - - expired_at, expired_count, used_count, code = await get_expire_info( - expire_value, expire_style - ) + expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style) path, suffix, prefix, uuid_file_name, save_path = await get_file_path_name(file) - file_storage: FileStorageInterface = storages[settings.file_storage]() await file_storage.save_file(file, save_path) - await create_file_code( code=code, prefix=prefix,