Added a version command

seahyc · Oct 10, 2024 · 52a37b3 · 52a37b3
1 parent cbbc121
commit 52a37b3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/ocbc_dbs_statement_parser/__pycache__/__init__.cpython-38.pyc b/src/ocbc_dbs_statement_parser/__pycache__/__init__.cpython-38.pyc
diff --git a/src/ocbc_dbs_statement_parser/cli.py b/src/ocbc_dbs_statement_parser/cli.py
@@ -2,6 +2,7 @@
 import json
 from decimal import Decimal
 from .main import parse_bank_statement
+from . import __version__  # Import the version from your package
 
 def decimal_default(obj):
     if isinstance(obj, Decimal):
@@ -13,6 +14,7 @@ def cli():
     parser.add_argument("pdf_path", help="Path to the PDF file")
     parser.add_argument("--debug", action="store_true", help="Enable debug output")
     parser.add_argument("--verify", action="store_true", help="Verify transaction totals")
+    parser.add_argument("--version", action="version", version=f"%(prog)s {__version__}")
     args = parser.parse_args()
 
     result = parse_bank_statement(args.pdf_path, args.debug, args.verify)