jeirmeister/repo-to-text
1
0
Fork 0
mirror of https://github.com/kirill-markin/repo-to-text.git synced 2025-12-06 03:22:23 -08:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
Kirill Markin 2024-06-08 11:21:11 +02:00
parent b3282bfc6e
commit efaf4654fa
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
repo_to_text/main.py
View file

@ -84,7 +84,11 @@ def save_repo_to_text(path='.') -> None:
output_file = 'repo_structure.txt' output_file = 'repo_structure.txt'
with open(output_file, 'w') as file: with open(output_file, 'w') as file:
file.write(tree_structure + '\n') project_name = os.path.basename(os.path.abspath(path))
file.write(f'Project: {project_name}\n\n')
file.write('Directory Structure:\n')
file.write('```\n.\n')
file.write(tree_structure + '\n' + '```\n')
logging.debug('Tree structure written to file') logging.debug('Tree structure written to file')
for root, _, files in os.walk(path): for root, _, files in os.walk(path):
@ -97,7 +101,7 @@ def save_repo_to_text(path='.') -> None:
relative_path = relative_path.replace('./', '', 1) relative_path = relative_path.replace('./', '', 1)
file.write(f'\n{relative_path}\n') file.write(f'\nContents of {relative_path}:\n')
file.write('```\n') file.write('```\n')
try: try:
with open(file_path, 'r', encoding='utf-8') as f: with open(file_path, 'r', encoding='utf-8') as f: