106 lines
3.4 KiB
Python
Executable File
106 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
from pathlib import Path
|
|
import pyperclip
|
|
import questionary
|
|
|
|
# List of directories to exclude
|
|
EXCLUDED_DIRS = {'node_modules', '.next', '.venv', '.git', '__pycache__', '.idea', '.vscode', 'ui'}
|
|
|
|
def collect_files(project_path):
|
|
"""
|
|
Collects files from the project directory, excluding specified directories and filtering by extensions.
|
|
Returns a list of file paths relative to the project directory.
|
|
"""
|
|
collected_files = []
|
|
|
|
for root, dirs, files in os.walk(project_path):
|
|
# Exclude specified directories
|
|
dirs[:] = [d for d in dirs if d not in EXCLUDED_DIRS]
|
|
|
|
for file in files:
|
|
file_path = Path(root) / file
|
|
relative_path = file_path.relative_to(project_path)
|
|
collected_files.append(relative_path)
|
|
|
|
return collected_files
|
|
|
|
def main():
|
|
# Parse command-line arguments
|
|
parser = argparse.ArgumentParser(description='Generate Markdown from selected files.')
|
|
parser.add_argument('path', nargs='?', default='.', help='Path to the project directory')
|
|
args = parser.parse_args()
|
|
|
|
project_path = Path(args.path).resolve()
|
|
if not project_path.is_dir():
|
|
print(f"Error: '{project_path}' is not a directory.")
|
|
sys.exit(1)
|
|
|
|
# Collect files from the project directory
|
|
file_list = collect_files(project_path)
|
|
|
|
if not file_list:
|
|
print("No files found in the project directory with the specified extensions.")
|
|
sys.exit(1)
|
|
|
|
# Sort file_list for better organization
|
|
file_list.sort()
|
|
|
|
# Interactive file selection using questionary
|
|
print("\nSelect the files you want to include:")
|
|
selected_files = questionary.checkbox(
|
|
"Press space to select files, and Enter when you're done:",
|
|
choices=[str(f) for f in file_list]
|
|
).ask()
|
|
|
|
if not selected_files:
|
|
print("No files selected.")
|
|
sys.exit(1)
|
|
|
|
# Generate markdown
|
|
markdown_lines = []
|
|
markdown_lines.append('')
|
|
|
|
for selected_file in selected_files:
|
|
file_path = project_path / selected_file
|
|
try:
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
# Determine the language for code block from file extension
|
|
language = file_path.suffix.lstrip('.')
|
|
markdown_lines.append(f'{selected_file}')
|
|
markdown_lines.append(f'```{language}')
|
|
markdown_lines.append(content)
|
|
markdown_lines.append('```')
|
|
markdown_lines.append('')
|
|
except Exception as e:
|
|
print(f"Error reading file {selected_file}: {e}")
|
|
|
|
markdown_text = '\n'.join(markdown_lines)
|
|
|
|
# Write markdown to file
|
|
output_file = 'output.md'
|
|
with open(output_file, 'w', encoding='utf-8') as f:
|
|
f.write(markdown_text)
|
|
print(f"\nMarkdown file '{output_file}' has been generated.")
|
|
|
|
# Copy markdown content to clipboard
|
|
pyperclip.copy(markdown_text)
|
|
print("Markdown content has been copied to the clipboard.")
|
|
|
|
if __name__ == "__main__":
|
|
# Check if required libraries are installed
|
|
try:
|
|
import questionary
|
|
import pyperclip
|
|
except ImportError as e:
|
|
missing_module = e.name
|
|
print(f"Error: Missing required module '{missing_module}'.")
|
|
print(f"Please install it by running: pip install {missing_module}")
|
|
sys.exit(1)
|
|
|
|
main()
|