R360-ModelEvaluator/scripts/organize_project.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
项目结构优化脚本
用于重新组织项目文件结构，创建更清晰的目录布局
"""

import os
import shutil
from pathlib import Path


def create_project_structure():
    """创建标准的项目目录结构"""
    base_dir = Path.cwd()
    
    # 定义目录结构
    directories = [
        'src',                    # 源代码目录
        'src/core',              # 核心模块
        'src/utils',             # 工具模块
        'src/config',            # 配置模块
        'tests',                 # 测试目录
        'docs',                  # 文档目录
        'examples',              # 示例代码
        'scripts',               # 脚本目录
        'data',                  # 数据目录
        'data/models',           # 模型文件
        'data/samples',          # 样本数据
        'output',                # 输出目录
        'logs',                  # 日志目录
    ]
    
    # 创建目录
    for directory in directories:
        dir_path = base_dir / directory
        dir_path.mkdir(parents=True, exist_ok=True)
        print(f"✓ 创建目录: {directory}")
    
    return base_dir, directories


def move_files_to_structure(base_dir):
    """将现有文件移动到新的目录结构中"""
    
    # 文件移动映射
    file_moves = {
        # 核心模块
        'detector.py': 'src/core/',
        'image_processor.py': 'src/core/',
        'model_inference.py': 'src/core/',
        'post_processor.py': 'src/core/',
        'file_manager.py': 'src/utils/',
        'report_generator.py': 'src/utils/',
        
        # 配置模块
        'config.py': 'src/config/',
        
        # 主入口
        'main.py': 'src/',
        
        # 测试文件
        'test_refactored.py': 'tests/',
        
        # 脚本
        'organize_project.py': 'scripts/',
        
        # 模型文件
        'model/': 'data/models/',
    }
    
    # 执行文件移动
    for src_path, dst_dir in file_moves.items():
        src_full = base_dir / src_path
        dst_full = base_dir / dst_dir
        
        if src_full.exists():
            if src_full.is_dir():
                # 移动目录
                dst_path = dst_full / src_full.name
                if dst_path.exists():
                    shutil.rmtree(dst_path)
                shutil.move(str(src_full), str(dst_full))
                print(f"✓ 移动目录: {src_path} -> {dst_dir}")
            else:
                # 移动文件
                dst_path = dst_full / src_full.name
                if dst_path.exists():
                    dst_path.unlink()
                shutil.move(str(src_full), str(dst_full))
                print(f"✓ 移动文件: {src_path} -> {dst_dir}")
        else:
            print(f"⚠ 文件不存在: {src_path}")


def create_init_files(base_dir):
    """创建__init__.py文件"""
    init_files = [
        'src/__init__.py',
        'src/core/__init__.py',
        'src/utils/__init__.py',
        'src/config/__init__.py',
        'tests/__init__.py',
    ]
    
    for init_file in init_files:
        init_path = base_dir / init_file
        if not init_path.exists():
            init_path.write_text('# -*- coding: utf-8 -*-\n')
            print(f"✓ 创建: {init_file}")


def update_imports_in_files(base_dir):
    """更新文件中的导入路径"""
    # 需要更新导入的文件
    files_to_update = [
        'src/main.py',
        'src/core/detector.py',
        'tests/test_refactored.py',
    ]
    
    # 导入路径映射
    import_mapping = {
        'from config import': 'from src.config.config import',
        'from detector import': 'from src.core.detector import',
        'from image_processor import': 'from src.core.image_processor import',
        'from model_inference import': 'from src.core.model_inference import',
        'from post_processor import': 'from src.core.post_processor import',
        'from file_manager import': 'from src.utils.file_manager import',
        'from report_generator import': 'from src.utils.report_generator import',
    }
    
    for file_path in files_to_update:
        full_path = base_dir / file_path
        if full_path.exists():
            try:
                content = full_path.read_text(encoding='utf-8')
                
                # 更新导入语句
                for old_import, new_import in import_mapping.items():
                    content = content.replace(old_import, new_import)
                
                full_path.write_text(content, encoding='utf-8')
                print(f"✓ 更新导入: {file_path}")
            except Exception as e:
                print(f"⚠ 更新失败 {file_path}: {e}")


def create_setup_files(base_dir):
    """创建项目配置文件"""
    
    # 创建setup.py
    setup_content = '''#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from setuptools import setup, find_packages

with open("README.md", "r", encoding="utf-8") as fh:
    long_description = fh.read()

with open("requirements.txt", "r", encoding="utf-8") as fh:
    requirements = [line.strip() for line in fh if line.strip() and not line.startswith("#")]

setup(
    name="uav-detector",
    version="2.0.0",
    author="UAV Detection Team",
    description="基于ONNX的无人机检测系统",
    long_description=long_description,
    long_description_content_type="text/markdown",
    packages=find_packages(),
    classifiers=[
        "Development Status :: 4 - Beta",
        "Intended Audience :: Developers",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
    ],
    python_requires=">=3.8",
    install_requires=requirements,
    entry_points={
        "console_scripts": [
            "uav-detect=src.main:main",
        ],
    },
)
'''
    
    setup_path = base_dir / 'setup.py'
    setup_path.write_text(setup_content, encoding='utf-8')
    print("✓ 创建: setup.py")
    
    # 创建.gitignore
    gitignore_content = '''# Python
__pycache__/
*.py[cod]
*$py.class
*.so
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
*.egg-info/
.installed.cfg
*.egg
PIPFILE.lock

# Virtual Environment
.venv/
venv/
ENV/
env/

# IDE
.vscode/
.idea/
*.swp
*.swo
*~

# OS
.DS_Store
.DS_Store?
._*
.Spotlight-V100
.Trashes
ehthumbs.db
Thumbs.db

# Project specific
output/
logs/
*.log
data/samples/*
!data/samples/.gitkeep
'''
    
    gitignore_path = base_dir / '.gitignore'
    if not gitignore_path.exists():
        gitignore_path.write_text(gitignore_content, encoding='utf-8')
        print("✓ 创建: .gitignore")


def create_placeholder_files(base_dir):
    """创建占位文件"""
    placeholders = [
        ('data/samples/.gitkeep', ''),
        ('logs/.gitkeep', ''),
        ('output/.gitkeep', ''),
        ('docs/API.md', '# API 文档\n\n待完善...\n'),
        ('examples/basic_usage.py', '#!/usr/bin/env python3\n# -*- coding: utf-8 -*-\n\n"""基本使用示例"""\n\n# 待完善...\n'),
    ]
    
    for file_path, content in placeholders:
        full_path = base_dir / file_path
        if not full_path.exists():
            full_path.write_text(content, encoding='utf-8')
            print(f"✓ 创建占位文件: {file_path}")


def main():
    """主函数"""
    print("🚀 开始项目结构重构...")
    print("=" * 50)
    
    try:
        # 1. 创建目录结构
        print("\n📁 创建目录结构...")
        base_dir, directories = create_project_structure()
        
        # 2. 移动文件
        print("\n📦 移动文件到新结构...")
        move_files_to_structure(base_dir)
        
        # 3. 创建__init__.py文件
        print("\n📄 创建__init__.py文件...")
        create_init_files(base_dir)
        
        # 4. 更新导入路径
        print("\n🔄 更新导入路径...")
        update_imports_in_files(base_dir)
        
        # 5. 创建配置文件
        print("\n⚙️ 创建项目配置文件...")
        create_setup_files(base_dir)
        
        # 6. 创建占位文件
        print("\n📝 创建占位文件...")
        create_placeholder_files(base_dir)
        
        print("\n" + "=" * 50)
        print("🎉 项目结构重构完成！")
        print("\n新的项目结构:")
        print("├── src/                 # 源代码")
        print("│   ├── core/            # 核心模块")
        print("│   ├── utils/           # 工具模块")
        print("│   ├── config/          # 配置模块")
        print("│   └── main.py          # 主入口")
        print("├── tests/               # 测试代码")
        print("├── docs/                # 文档")
        print("├── examples/            # 示例代码")
        print("├── scripts/             # 脚本")
        print("├── data/                # 数据文件")
        print("│   ├── models/          # 模型文件")
        print("│   └── samples/         # 样本数据")
        print("├── output/              # 输出目录")
        print("├── logs/                # 日志目录")
        print("├── setup.py             # 安装配置")
        print("├── requirements.txt     # 依赖列表")
        print("└── README.md            # 项目说明")
        
    except Exception as e:
        print(f"❌ 重构过程中出现错误: {e}")
        return False
    
    return True


if __name__ == '__main__':
    success = main()
    if success:
        print("\n✅ 可以使用以下命令测试新结构:")
        print("   python -m src.main --help")
        print("   python -m pytest tests/")
    else:
        print("\n❌ 重构失败，请检查错误信息")