Skip to content

Commit 5561592

Browse files
committed
Add signatures to CLI
1 parent cb936bd commit 5561592

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

textractor/cli/cli.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -368,6 +368,8 @@ def textractor_cli():
368368
print(out.queries.pretty_print())
369369
if "ALL" in args.print or "EXPENSES" in args.print:
370370
print(out.expense_documents.pretty_print())
371+
if "ALL" in args.print or "SIGNATURES" in args.print:
372+
print(out.signatures.pretty_print())
371373
if "ALL" in args.print or "IDS" in args.print:
372374
print(out.identity_documents.pretty_print())
373375

@@ -383,6 +385,8 @@ def textractor_cli():
383385
entity_list += out.key_values
384386
if "ALL" in args.overlay or "QUERIES" in args.overlay:
385387
entity_list += out.queries
388+
if "ALL" in args.overlay or "SIGNATURES" in args.overlay:
389+
entity_list += out.signatures
386390
image = entity_list.visualize(
387391
with_text=True,
388392
font_size_ratio=args.font_size_ratio,

textractor/data/constants.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,8 @@ class CLIPrint(Enum):
219219
FORMS = 3
220220
QUERIES = 4
221221
EXPENSES = 5
222-
IDS = 6
222+
SIGNATURES = 6
223+
IDS = 7
223224

224225

225226
class CLIOverlay(Enum):
@@ -229,3 +230,4 @@ class CLIOverlay(Enum):
229230
TABLES = 3
230231
FORMS = 4
231232
QUERIES = 5
233+
SIGNATURE = 6

0 commit comments

Comments
 (0)