fix: use click.echo instead of print

phette23 · Nov 2, 2024 · a1080a5 · a1080a5
1 parent e721c8a
commit a1080a5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/marcgrep/cli.py b/marcgrep/cli.py
@@ -71,17 +71,17 @@ def main(
                                 if color:
                                     color_field(f, invert)
                                 else:
-                                    print(f)
+                                    click.echo(f)
                         else:
                             if color:
                                 color_record(record, invert)
                             else:
-                                print(record)
+                                click.echo(record)
                 if limit and counter >= limit:
                     break
 
         if count:
-            print(f"{file.name}: {matched_records}")
+            click.echo(f"{file.name}: {matched_records}")
 
     # non-zero exit if no records match
     return exit(0 if any_matches else 1)

diff --git a/marcgrep/color.py b/marcgrep/color.py
@@ -1,6 +1,7 @@
 from os import environ
 from typing import get_args
 
+import click
 from pymarc import Field, Record
 from termcolor import cprint
 from termcolor._types import Color
@@ -72,10 +73,10 @@ def color_field(field: Field, invert: bool) -> None:
         for subfield in field.subfields:
             cprint(f"{DELIMITER}{subfield.code}", COLORS["SUBFIELD"], end="")
             cprint(f"{subfield.value}", COLORS["DATA"], end="")
-    print()
+    click.echo()
 
 
 def color_record(record: Record, invert: bool) -> None:
     for field in record:
         color_field(field, invert)
-    print()
+    click.echo()