64b8aac86f
Co-authored-by: hajdul88 <52442977+hajdul88@users.noreply.github.com> Co-authored-by: hande-k <handekafkas7@gmail.com> Co-authored-by: Igor Ilic <igorilic03@gmail.com> Co-authored-by: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Co-authored-by: Igor Ilic <30923996+dexters1@users.noreply.github.com>
29 lines
1,013 B
Python
29 lines
1,013 B
Python
import os
|
|
import asyncio
|
|
import argparse
|
|
from cognee.tasks.repo_processor.get_repo_file_dependencies import get_repo_file_dependencies
|
|
from cognee.tasks.repo_processor.enrich_dependency_graph import enrich_dependency_graph
|
|
from cognee.tasks.repo_processor.expand_dependency_graph import expand_dependency_graph
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("repo_path", help="Path to the repository")
|
|
args = parser.parse_args()
|
|
|
|
repo_path = args.repo_path
|
|
if not os.path.exists(repo_path):
|
|
print(f"Error: The provided repository path does not exist: {repo_path}")
|
|
return
|
|
|
|
graph = asyncio.run(get_repo_file_dependencies(repo_path))
|
|
graph = asyncio.run(enrich_dependency_graph(graph))
|
|
graph = expand_dependency_graph(graph)
|
|
for node in graph.nodes:
|
|
print(f"Node: {node}")
|
|
for _, target, data in graph.out_edges(node, data=True):
|
|
print(f" Edge to {target}, data: {data}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|