aiogram3-uow-taskiq-skeleton/collect_context.pyw at main · PyDevDeep/aiogram3-uow-taskiq-skeleton · GitHub

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
89
90
91
92
import os

OUTPUT_FILE = "project_context.txt"
# Видалили .ftl, щоб не плодити копії перекладів
INCLUDE_EXTENSIONS = (".py", ".yaml", ".ini")
INCLUDE_FILES = (
    "requirements.txt",
    "Dockerfile",
    "docker-compose.yml",
    "alembic.ini",
    ".gitignore",
    # Додали конкретний шлях як Single Source of Truth для правил
    os.path.join("app", "locales", "en", "parsing_rules.json"),
)
EXCLUDE_DIRS = {
    "venv",
    "tests",
    ".git",
    ".roo/__pycache__",
    ".pytest_cache",
    "env",
    "dist",
    "build",
    ".idea",
    ".vscode",
    "project_context.txt",
    "collect_context.py",
    "test_results.txt",
    "parser_test.txt",
    # Видаляємо інші мови з перебору, щоб не витрачати ресурси
    "de",
    "fr",
    "it",
    "pl",
    "uk",
}


def generate_tree(startpath):
    tree_str = "PROJECT STRUCTURE:\n"
    for root, dirs, files in os.walk(startpath):
        dirs[:] = [d for d in dirs if d not in EXCLUDE_DIRS]
        level = root.replace(startpath, "").count(os.sep)
        indent = " " * 4 * level
        tree_str += f"{indent}{os.path.basename(root)}/\n"
        subindent = " " * 4 * (level + 1)
        for f in files:
            full_path = os.path.relpath(os.path.join(root, f), startpath)
            if (
                f.endswith(INCLUDE_EXTENSIONS)
                or f in INCLUDE_FILES
                or full_path in INCLUDE_FILES
            ):
                tree_str += f"{subindent}{f}\n"
    return tree_str


def collect_project_data(root_path="."):
    with open(OUTPUT_FILE, "w", encoding="utf-8") as f_out:
        f_out.write(generate_tree(root_path))
        f_out.write("\n" + "=" * 50 + "\n")
        f_out.write("FILE CONTENTS BEGIN HERE (LOCALES IGNORED EXCEPT EN RULES)\n")
        f_out.write("=" * 50 + "\n")

        for root, dirs, files in os.walk(root_path):
            dirs[:] = [d for d in dirs if d not in EXCLUDE_DIRS]
            for file in files:
                full_path = os.path.join(root, file)
                relative_path = os.path.relpath(full_path, root_path)

                # Перевірка: розширення АБО пряме включення імені АБО повний відносний шлях
                if (
                    file.endswith(INCLUDE_EXTENSIONS)
                    or file in INCLUDE_FILES
                    or relative_path in INCLUDE_FILES
                ):
                    if file == os.path.basename(__file__) or file == OUTPUT_FILE:
                        continue

                    f_out.write(f"\n--- FILE: {relative_path} ---\n")
                    try:
                        with open(full_path, "r", encoding="utf-8") as f_in:
                            f_out.write(f_in.read())
                    except Exception as e:
                        f_out.write(f"[Error reading file: {e}]\n")
                    f_out.write("\n")

    print(f"Готово! Контекст оптимізовано: {os.path.abspath(OUTPUT_FILE)}")


if __name__ == "__main__":
    collect_project_data()