于小丘
5835b4c626
Introduced utils/http/http_exceptions.py with common HTTP exception helpers for FastAPI. Updated main.py to use a global exception handler that logs and hides internal errors. Refined models/README.md to document new models and relationships, including PhysicalFile and UploadSession, and updated DTO and enum documentation. Simplified ThemeResponse in models/color.py. Improved models/download.py with type annotations, index changes, and import optimizations. Fixed a parameter type in clean.py. Co-Authored-By: 砂糖橘 <54745033+Foxerine@users.noreply.github.com>
239 lines
9.0 KiB
Python
239 lines
9.0 KiB
Python
import os
|
|
import shutil
|
|
from loguru import logger
|
|
import argparse
|
|
from typing import List, Tuple, Set
|
|
import time
|
|
|
|
Version = "2.1.0"
|
|
|
|
# 默认排除的目录
|
|
DEFAULT_EXCLUDE_DIRS = {"venv", "env", ".venv", ".env", "node_modules"}
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description='清理 Python 缓存文件')
|
|
parser.add_argument('-p', '--path', default=os.getcwd(),
|
|
help='要清理的目录路径，默认为当前工作目录')
|
|
parser.add_argument('-y', '--yes', action='store_true',
|
|
help='自动确认所有操作，不再询问')
|
|
parser.add_argument('--no-pycache', action='store_true',
|
|
help='不清理 __pycache__ 目录')
|
|
parser.add_argument('--no-nicegui', action='store_true',
|
|
help='不清理 .nicegui 目录')
|
|
parser.add_argument('--no-testdb', action='store_true',
|
|
help='不清理 test.db 文件')
|
|
parser.add_argument('--pyc', action='store_true',
|
|
help='清理 .pyc 文件')
|
|
parser.add_argument('--pytest-cache', action='store_true',
|
|
help='清理 .pytest_cache 目录')
|
|
parser.add_argument('--exclude', type=str, default="",
|
|
help='排除的目录，多个目录用逗号分隔')
|
|
parser.add_argument('--logger-file',
|
|
help='指定日志文件路径')
|
|
parser.add_argument('--dry-run', action='store_true',
|
|
help='仅列出将要删除的文件，不实际删除')
|
|
return parser.parse_args()
|
|
|
|
def confirm_action(message: str, auto_yes: bool = False) -> bool:
|
|
if auto_yes:
|
|
return True
|
|
return input(f"{message} (y/N): ").lower() == 'y'
|
|
|
|
def safe_remove(path: str, dry_run: bool = False) -> Tuple[bool, str]:
|
|
"""安全删除文件或目录"""
|
|
try:
|
|
if dry_run:
|
|
return True, f"DRY RUN: 将删除 {path}"
|
|
|
|
if os.path.isdir(path):
|
|
shutil.rmtree(path)
|
|
elif os.path.isfile(path):
|
|
os.remove(path)
|
|
return True, ""
|
|
except PermissionError as e:
|
|
return False, f"权限错误: {e}"
|
|
except OSError as e:
|
|
return False, f"系统错误: {e}"
|
|
except Exception as e:
|
|
return False, f"未知错误: {e}"
|
|
|
|
def get_excluded_dirs(exclude_arg: str) -> Set[str]:
|
|
"""获取要排除的目录列表"""
|
|
result = set(DEFAULT_EXCLUDE_DIRS)
|
|
if exclude_arg:
|
|
for item in exclude_arg.split(','):
|
|
item = item.strip()
|
|
if item:
|
|
result.add(item)
|
|
return result
|
|
|
|
def clean_pycache(root_dir: str, exclude_dirs: Set[str], dry_run: bool = False) -> List[str]:
|
|
"""清理 __pycache__ 目录"""
|
|
logger.info("开始清理 __pycache__ 目录pass")
|
|
cleaned_paths = []
|
|
|
|
for dirpath, dirnames, _ in os.walk(root_dir):
|
|
# 排除指定目录
|
|
for exclude in exclude_dirs:
|
|
if exclude in dirnames:
|
|
dirnames.remove(exclude)
|
|
|
|
if "__pycache__" in dirnames:
|
|
pycache_dir = os.path.join(dirpath, "__pycache__")
|
|
success, error = safe_remove(pycache_dir, dry_run)
|
|
if success:
|
|
cleaned_paths.append(pycache_dir)
|
|
else:
|
|
logger.error(f"无法清理 {pycache_dir}: {error}")
|
|
|
|
return cleaned_paths
|
|
|
|
def clean_pyc_files(root_dir: str, exclude_dirs: Set[str], dry_run: bool = False) -> List[str]:
|
|
"""清理 .pyc 文件"""
|
|
logger.info("开始清理 .pyc 文件pass")
|
|
cleaned_files = []
|
|
|
|
for dirpath, dirnames, filenames in os.walk(root_dir):
|
|
# 排除指定目录
|
|
for exclude in exclude_dirs:
|
|
if exclude in dirnames:
|
|
dirnames.remove(exclude)
|
|
|
|
for filename in filenames:
|
|
if filename.endswith('.pyc'):
|
|
file_path = os.path.join(dirpath, filename)
|
|
success, error = safe_remove(file_path, dry_run)
|
|
if success:
|
|
cleaned_files.append(file_path)
|
|
else:
|
|
logger.error(f"无法清理 {file_path}: {error}")
|
|
|
|
return cleaned_files
|
|
|
|
def clean_pytest_cache(root_dir: str, exclude_dirs: Set[str], dry_run: bool = False) -> List[str]:
|
|
"""清理 .pytest_cache 目录"""
|
|
logger.info("开始清理 .pytest_cache 目录pass")
|
|
cleaned_paths = []
|
|
|
|
for dirpath, dirnames, _ in os.walk(root_dir):
|
|
# 排除指定目录
|
|
for exclude in exclude_dirs:
|
|
if exclude in dirnames:
|
|
dirnames.remove(exclude)
|
|
|
|
if ".pytest_cache" in dirnames:
|
|
cache_dir = os.path.join(dirpath, ".pytest_cache")
|
|
success, error = safe_remove(cache_dir, dry_run)
|
|
if success:
|
|
cleaned_paths.append(cache_dir)
|
|
else:
|
|
logger.error(f"无法清理 {cache_dir}: {error}")
|
|
|
|
return cleaned_paths
|
|
|
|
def clean_nicegui(root_dir: str, dry_run: bool = False) -> Tuple[bool, str]:
|
|
"""清理 .nicegui 目录"""
|
|
logger.info("开始清理 .nicegui 目录pass")
|
|
nicegui_dir = os.path.join(root_dir, ".nicegui")
|
|
if os.path.exists(nicegui_dir) and os.path.isdir(nicegui_dir):
|
|
success, error = safe_remove(nicegui_dir, dry_run)
|
|
if success:
|
|
return True, nicegui_dir
|
|
else:
|
|
logger.error(f"无法清理 {nicegui_dir}: {error}")
|
|
return False, nicegui_dir
|
|
return False, nicegui_dir
|
|
|
|
def clean_testdb(root_dir: str, dry_run: bool = False) -> Tuple[bool, str, str]:
|
|
"""清理测试数据库文件"""
|
|
logger.info("开始清理 test.db 文件pass")
|
|
test_db = os.path.join(root_dir, "test.db")
|
|
if os.path.exists(test_db) and os.path.isfile(test_db):
|
|
success, error = safe_remove(test_db, dry_run)
|
|
if success:
|
|
return True, test_db, ""
|
|
else:
|
|
return False, test_db, error
|
|
return False, test_db, "文件不存在"
|
|
|
|
def main():
|
|
start_time = time.time()
|
|
args = parse_args()
|
|
root_dir = os.path.abspath(args.path)
|
|
exclude_dirs = get_excluded_dirs(args.exclude)
|
|
|
|
# 设置日志文件
|
|
if args.logger_file:
|
|
logger.set_logger_file(args.logger_file)
|
|
|
|
if not os.path.exists(root_dir):
|
|
logger.error(f"目录不存在 Directory not exists: {root_dir}")
|
|
return 1
|
|
|
|
logger.info(f"清理目录 Clean Directory: {root_dir}")
|
|
if args.dry_run:
|
|
logger.warning("模拟运行模式: 将只列出要删除的文件，不会实际删除")
|
|
|
|
if exclude_dirs:
|
|
logger.info(f"排除目录: {', '.join(exclude_dirs)}")
|
|
|
|
try:
|
|
total_cleaned = 0
|
|
|
|
# 清理 __pycache__
|
|
if not args.no_pycache and confirm_action("是否清理 __pycache__ 目录?", args.yes):
|
|
cleaned = clean_pycache(root_dir, exclude_dirs, args.dry_run)
|
|
for path in cleaned:
|
|
logger.info(f"已清理 Removed: {path}")
|
|
total_cleaned += len(cleaned)
|
|
|
|
# 清理 .pyc 文件
|
|
if args.pyc and confirm_action("是否清理 .pyc 文件?", args.yes):
|
|
cleaned = clean_pyc_files(root_dir, exclude_dirs, args.dry_run)
|
|
for path in cleaned:
|
|
logger.info(f"已清理 Removed: {path}")
|
|
total_cleaned += len(cleaned)
|
|
|
|
# 清理 .pytest_cache
|
|
if args.pytest_cache and confirm_action("是否清理 .pytest_cache 目录?", args.yes):
|
|
cleaned = clean_pytest_cache(root_dir, exclude_dirs, args.dry_run)
|
|
for path in cleaned:
|
|
logger.info(f"已清理 Removed: {path}")
|
|
total_cleaned += len(cleaned)
|
|
|
|
# 清理 .nicegui
|
|
if not args.no_nicegui and confirm_action("是否清理 .nicegui 目录?", args.yes):
|
|
cleaned, path = clean_nicegui(root_dir, args.dry_run)
|
|
if cleaned:
|
|
logger.info(f"已清理 Removed: {path}")
|
|
total_cleaned += 1
|
|
else:
|
|
logger.debug(f"未找到 Not found: {path}")
|
|
|
|
# 清理 test.db
|
|
if not args.no_testdb and confirm_action("是否清理 test.db 文件?", args.yes):
|
|
success, path, error = clean_testdb(root_dir, args.dry_run)
|
|
if success:
|
|
logger.info(f"已清理 Removed: {path}")
|
|
total_cleaned += 1
|
|
elif error == "文件不存在":
|
|
logger.debug(f"未找到 Not found: {path}")
|
|
else:
|
|
logger.error(f"清理失败 Failed: {error}")
|
|
|
|
except KeyboardInterrupt:
|
|
logger.warning("操作被用户中断")
|
|
return 1
|
|
except Exception as e:
|
|
logger.error(f"错误 Error: {e}")
|
|
return 1
|
|
else:
|
|
elapsed_time = time.time() - start_time
|
|
if args.dry_run:
|
|
logger.success(f"模拟清理结束，发现 {total_cleaned} 个可清理项目 (用时: {elapsed_time:.2f}秒)")
|
|
else:
|
|
logger.success(f"清理结束，共清理 {total_cleaned} 个项目 (用时: {elapsed_time:.2f}秒)")
|
|
return 0
|
|
|
|
if __name__ == "__main__":
|
|
exit(main()) |