fix: print leader with --color flag

phette23 · Jan 29, 2025 · 64dd893 · 64dd893
1 parent a1080a5
commit 64dd893
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/marcgrep/color.py b/marcgrep/color.py
@@ -77,6 +77,12 @@ def color_field(field: Field, invert: bool) -> None:
 
 
 def color_record(record: Record, invert: bool) -> None:
+    # create COLORS global if we have not already
+    if not globals().get("COLOR"):
+        _setup_colors(invert)
+
+    cprint("LDR", COLORS["TAG"], end=" ")
+    cprint(record.leader, COLORS["DATA"])
     for field in record:
         color_field(field, invert)
     click.echo()
diff --git a/test/test_cli.py b/test/test_cli.py
@@ -180,3 +180,9 @@ def test_glob(self):
         )
         assert result.exit_code == 0
         assert len(result.output.splitlines()) == 1
+
+    def test_color_leader(self):
+        runner = CliRunner()
+        result = runner.invoke(main, ["--color", ONE_RECORD])
+        assert result.exit_code == 0
+        assert result.output.splitlines()[0] == "LDR 03292namaa2200793uu 4500"