forked from Mirrors/Dispatcharr
77 lines
3.4 KiB
Python
Executable file
77 lines
3.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import os
|
|
|
|
# Specify the names of the script file and output file to exclude them
|
|
SCRIPT_NAME = "FileTree.py"
|
|
OUTPUT_FILE = "FileTree.txt"
|
|
EXCLUDED_FILES = {SCRIPT_NAME, OUTPUT_FILE, ".DS_Store", "__init__.py", "FileTree.old.txt"}
|
|
EXCLUDED_DIRS = {"__pycache__", "migrations", "static", "staticfiles", "media", ".venv", ".idea"} # Exclude directories like __pycache__
|
|
|
|
def generate_file_tree(output_file):
|
|
"""Generate a pretty file tree of the current directory and subdirectories."""
|
|
with open(output_file, 'w') as f:
|
|
for root, dirs, files in os.walk('.'): # Walk through the directory tree
|
|
# Remove excluded directories from the traversal
|
|
dirs[:] = [d for d in dirs if d not in EXCLUDED_DIRS]
|
|
level = root.count(os.sep)
|
|
indent = '│ ' * level
|
|
f.write(f"{indent}├── {os.path.basename(root)}/\n")
|
|
sub_indent = '│ ' * (level + 1)
|
|
for i, file in enumerate(files):
|
|
if file not in EXCLUDED_FILES:
|
|
connector = '└── ' if i == len(files) - 1 else '├── '
|
|
f.write(f"{sub_indent}{connector}{file}\n")
|
|
|
|
def append_file_contents(output_file):
|
|
"""Append contents of each file in the current directory and subdirectories to the output file, excluding specified files."""
|
|
# Determine the maximum width for the boxes
|
|
max_width = 20 # Default minimum width
|
|
file_paths = []
|
|
for root, dirs, files in os.walk('.'): # Walk through the directory tree
|
|
# Remove excluded directories from the traversal
|
|
dirs[:] = [d for d in dirs if d not in EXCLUDED_DIRS]
|
|
for file_name in files:
|
|
if file_name not in EXCLUDED_FILES:
|
|
file_path = os.path.join(root, file_name)
|
|
relative_path = os.path.relpath(file_path, start='.')
|
|
directory = os.path.dirname(relative_path)
|
|
base_name = os.path.basename(relative_path)
|
|
file_paths.append((directory, base_name))
|
|
max_width = max(max_width, len(directory) + 10, len(base_name) + 10)
|
|
|
|
max_width += 4 # Add padding for aesthetics
|
|
|
|
# Append file contents with uniform box size
|
|
with open(output_file, 'a') as f:
|
|
for directory, base_name in file_paths:
|
|
# Add the formatted header for the file
|
|
horizontal_line = f"┌{'─' * max_width}┐"
|
|
directory_line = f"│ Directory: {directory:<{max_width - 12}}│"
|
|
file_line = f"│ File: {base_name:<{max_width - 12}}│"
|
|
bottom_line = f"└{'─' * max_width}┘"
|
|
|
|
f.write(f"\n{horizontal_line}\n")
|
|
f.write(f"{directory_line}\n")
|
|
f.write(f"{file_line}\n")
|
|
f.write(f"{bottom_line}\n\n")
|
|
|
|
# Append the contents of the file
|
|
file_path = os.path.join(directory, base_name)
|
|
try:
|
|
with open(file_path, 'r', errors='ignore') as file:
|
|
f.write(file.read())
|
|
except Exception as e:
|
|
f.write(f"Error reading {file_path}: {e}\n")
|
|
|
|
# Add a visually distinct footer to signify the end of the file
|
|
f.write(f"\n========= END OF FILE =========\n")
|
|
f.write(f"File: {base_name}\n")
|
|
f.write(f"===============================\n\n")
|
|
|
|
def main():
|
|
generate_file_tree(OUTPUT_FILE)
|
|
append_file_contents(OUTPUT_FILE)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|