Spaces:
Running
Running
File size: 2,671 Bytes
7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 7dc78b3 0955e72 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 |
"""
A fork of github.com/donoceidon/repo2txt/blob/main/src/repo2txt/repo2txt.py
This version only includes the functionality to document the structure of a repository containing .md and .mdx files.
"""
import os
def should_ignore(item_path):
"""
Determine if a given item should be ignored.
Only includes .md and .mdx files, ignores hidden files and directories.
Args:
item_path (str): The path of the item (file or directory) to check.
Returns:
bool: True if the item should be ignored, False otherwise.
"""
item_name = os.path.basename(item_path)
# Ignore hidden files and directories
if item_name.startswith('.'):
return True
# If it's a file, only include .md and .mdx files
if os.path.isfile(item_path):
file_ext = os.path.splitext(item_name)[1].lower()
return file_ext not in ['.md', '.mdx']
# Include directories (they will be traversed)
return False
def make_tree(dir_path, prefix="", is_root=True):
"""
Recursively generate the directory tree as a string.
Args:
dir_path (str): The path of the directory to document.
prefix (str): Prefix string for line indentation and structure.
is_root (bool): Flag to indicate if the current directory is the root.
Returns:
str: The tree structure as a string.
"""
tree_string = ""
if is_root:
tree_string += "└── ./\n"
# Add the actual directory name as a child of ./
actual_dir_name = os.path.basename(dir_path)
if actual_dir_name:
tree_string += f" └── {actual_dir_name}\n"
prefix = " "
else:
prefix = " "
is_root = False
try:
items = os.listdir(dir_path)
except PermissionError:
return tree_string
items.sort()
# Filter out items that should be ignored
filtered_items = []
for item in items:
item_path = os.path.join(dir_path, item)
if not should_ignore(item_path):
filtered_items.append(item)
num_items = len(filtered_items)
for index, item in enumerate(filtered_items):
item_path = os.path.join(dir_path, item)
is_last_item = (index == num_items - 1)
new_prefix = "└── " if is_last_item else "├── "
child_prefix = " " if is_last_item else "│ "
tree_string += f"{prefix}{new_prefix}{item}\n"
if os.path.isdir(item_path):
next_prefix = prefix + child_prefix
tree_string += make_tree(item_path, next_prefix, is_root=False)
return tree_string |