#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 项目代码导出工具 用于将整个电商项目的代码导出到文本文件中 """ import os import datetime from pathlib import Path class CodeExporter: def __init__(self, project_root=None): """ 初始化代码导出器 :param project_root: 项目根目录，默认为当前目录 """ self.project_root = Path(project_root) if project_root else Path('.') self.output_file = None # 需要导出的文件扩展名 self.include_extensions = { '.py', '.html', '.css', '.js', '.sql', '.txt', '.md', '.yml', '.yaml', '.json', '.xml', '.ini', '.cfg' } # 需要排除的目录 self.exclude_dirs = { 'venv', '.venv', 'env', '.env', '__pycache__', '.git', '.idea', '.vscode', 'node_modules', 'logs', 'temp', 'tmp', '.pytest_cache', '.coverage', 'htmlcov', 'dist', 'build' } # 需要排除的文件 self.exclude_files = { '.DS_Store', 'Thumbs.db', '.gitignore', '*.pyc', '*.pyo', '*.log', '*.tmp', '*.bak', '*.swp', '*.swo' } # 特殊处理的文件（即使没有扩展名也要包含） self.special_files = { 'Dockerfile', 'requirements.txt', 'README', 'LICENSE', 'Makefile', 'Procfile', '.dockerignore' } def should_include_file(self, file_path): """ 判断文件是否应该被包含在导出中 :param file_path: 文件路径 :return: bool """ file_name = file_path.name file_suffix = file_path.suffix.lower() # 检查特殊文件 if file_name in self.special_files: return True # 检查扩展名 if file_suffix in self.include_extensions: return True return False def should_exclude_dir(self, dir_path): """ 判断目录是否应该被排除 :param dir_path: 目录路径 :return: bool """ dir_name = dir_path.name return dir_name in self.exclude_dirs or dir_name.startswith('.') def get_file_info(self, file_path): """ 获取文件信息 :param file_path: 文件路径 :return: dict """ try: stat = file_path.stat() return { 'size': stat.st_size, 'modified': datetime.datetime.fromtimestamp(stat.st_mtime), 'relative_path': file_path.relative_to(self.project_root) } except Exception as e: return { 'size': 0, 'modified': datetime.datetime.now(), 'relative_path': file_path.relative_to(self.project_root), 'error': str(e) } def read_file_content(self, file_path): """ 读取文件内容 :param file_path: 文件路径 :return: str """ try: # 尝试用UTF-8编码读取 with open(file_path, 'r', encoding='utf-8') as f: return f.read() except UnicodeDecodeError: try: # 如果UTF-8失败，尝试GBK编码 with open(file_path, 'r', encoding='gbk') as f: return f.read() except UnicodeDecodeError: try: # 如果还是失败，尝试latin-1编码 with open(file_path, 'r', encoding='latin-1') as f: return f.read() except Exception as e: return f"[无法读取文件内容: {str(e)}]" except Exception as e: return f"[读取文件时发生错误: {str(e)}]" def scan_project(self): """ 扫描项目目录，获取所有需要导出的文件 :return: list """ files_to_export = [] for root, dirs, files in os.walk(self.project_root): root_path = Path(root) # 过滤掉需要排除的目录 dirs[:] = [d for d in dirs if not self.should_exclude_dir(root_path / d)] for file in files: file_path = root_path / file if self.should_include_file(file_path): file_info = self.get_file_info(file_path) files_to_export.append({ 'path': file_path, 'info': file_info }) # 按相对路径排序 files_to_export.sort(key=lambda x: str(x['info']['relative_path'])) return files_to_export def export_to_file(self, output_filename=None): """ 导出代码到文件 :param output_filename: 输出文件名 """ if not output_filename: timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") output_filename = f"project_code_export_{timestamp}.txt" self.output_file = output_filename files_to_export = self.scan_project() print(f"开始导出项目代码...") print(f"项目根目录: {self.project_root.absolute()}") print(f"找到 {len(files_to_export)} 个文件需要导出") print(f"输出文件: {output_filename}") with open(output_filename, 'w', encoding='utf-8') as output: # 写入文件头 self.write_header(output, files_to_export) # 写入每个文件的内容 for i, file_data in enumerate(files_to_export, 1): file_path = file_data['path'] file_info = file_data['info'] print(f"正在处理 ({i}/{len(files_to_export)}): {file_info['relative_path']}") self.write_file_section(output, file_path, file_info) # 写入文件尾 self.write_footer(output) print(f"\n✅ 导出完成！") print(f"输出文件: {output_filename}") print(f"文件大小: {os.path.getsize(output_filename) / 1024:.2f} KB") def write_header(self, output, files_to_export): """ 写入文件头部信息 """ output.write("=" * 80 + "\n") output.write("项目代码导出文件\n") output.write("=" * 80 + "\n") output.write(f"项目名称: 基于Python的线上电商系统\n") output.write(f"导出时间: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n") output.write(f"项目路径: {self.project_root.absolute()}\n") output.write(f"文件总数: {len(files_to_export)}\n") output.write("=" * 80 + "\n\n") # 写入文件目录 output.write("📁 文件目录:\n") output.write("-" * 50 + "\n") for file_data in files_to_export: file_info = file_data['info'] size_kb = file_info['size'] / 1024 if file_info['size'] > 0 else 0 output.write(f"{file_info['relative_path']} ({size_kb:.1f} KB)\n") output.write("\n" + "=" * 80 + "\n\n") def write_file_section(self, output, file_path, file_info): """ 写入单个文件的内容 """ relative_path = file_info['relative_path'] # 文件分隔符 output.write("🔸" + "=" * 78 + "\n") output.write(f"📄 文件: {relative_path}\n") output.write(f"📊 大小: {file_info['size']} bytes ({file_info['size'] / 1024:.2f} KB)\n") output.write(f"🕒 修改时间: {file_info['modified'].strftime('%Y-%m-%d %H:%M:%S')}\n") if 'error' in file_info: output.write(f"⚠️ 错误: {file_info['error']}\n") output.write("🔸" + "=" * 78 + "\n\n") # 文件内容 content = self.read_file_content(file_path) output.write(content) # 确保文件结尾有换行 if not content.endswith('\n'): output.write('\n') output.write("\n\n") def write_footer(self, output): """ 写入文件尾部信息 """ output.write("=" * 80 + "\n") output.write("导出完成\n") output.write(f"导出时间: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n") output.write("=" * 80 + "\n") def export_summary(self): """ 导出项目摘要信息 """ files_to_export = self.scan_project() # 按文件类型统计 type_stats = {} total_size = 0 for file_data in files_to_export: file_path = file_data['path'] file_info = file_data['info'] ext = file_path.suffix.lower() or '无扩展名' if ext not in type_stats: type_stats[ext] = {'count': 0, 'size': 0} type_stats[ext]['count'] += 1 type_stats[ext]['size'] += file_info['size'] total_size += file_info['size'] print("\n📊 项目统计信息:") print("-" * 50) print(f"总文件数: {len(files_to_export)}") print(f"总大小: {total_size / 1024:.2f} KB") print("\n📋 文件类型统计:") for ext, stats in sorted(type_stats.items(), key=lambda x: x[1]['count'], reverse=True): print(f"{ext:>10}: {stats['count']:>3} 个文件, {stats['size'] / 1024:>6.1f} KB") def main(): """ 主函数 """ print("🚀 项目代码导出工具") print("=" * 50) # 创建导出器 exporter = CodeExporter() # 显示项目摘要 exporter.export_summary() # 询问是否继续导出 print("\n" + "=" * 50) choice = input("是否继续导出完整代码到文件? (y/n): ").lower().strip() if choice in ['y', 'yes', '是']: # 询问输出文件名 output_name = input("请输入输出文件名 (直接回车使用默认名称): ").strip() if not output_name: output_name = None # 开始导出 exporter.export_to_file(output_name) else: print("取消导出。") if __name__ == "__main__": main()