alihan/files-to-prompt
1
0
Fork 0
mirror of https://github.com/simonw/files-to-prompt.git synced 2025-10-23 00:02:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Warn and continue on binary files, closes #5

Browse Source
This commit is contained in:
Simon Willison
2024-04-07 23:03:40 -07:00
parent b123b19ae8
commit 84df8a64ae
2 changed files with 48 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
files_to_prompt/cli.py
View File

@@ -27,13 +27,17 @@ def process_path(
path, include_hidden, ignore_gitignore, gitignore_rules, ignore_patterns
):
if os.path.isfile(path):
with open(path, "r") as f:
file_contents = f.read()
click.echo(path)
click.echo("---")
click.echo(file_contents)
click.echo()
click.echo("---")
try:
with open(path, "r") as f:
file_contents = f.read()
click.echo(path)
click.echo("---")
click.echo(file_contents)
click.echo()
click.echo("---")
except UnicodeDecodeError:
warning_message = f"Warning: Skipping file {path} due to UnicodeDecodeError"
click.echo(click.style(warning_message, fg="red"), err=True)
elif os.path.isdir(path):
for root, dirs, files in os.walk(path):
if not include_hidden:
@@ -62,14 +66,20 @@ def process_path(
for file in files:
file_path = os.path.join(root, file)
with open(file_path, "r") as f:
file_contents = f.read()
try:
with open(file_path, "r") as f:
file_contents = f.read()
click.echo(file_path)
click.echo("---")
click.echo(file_contents)
click.echo()
click.echo("---")
click.echo(file_path)
click.echo("---")
click.echo(file_contents)
click.echo()
click.echo("---")
except UnicodeDecodeError:
warning_message = (
f"Warning: Skipping file {file_path} due to UnicodeDecodeError"
)
click.echo(click.style(warning_message, fg="red"), err=True)
@click.command()