diff --git a/pfbudget/__init__.py b/pfbudget/__init__.py index ae757c3..6a66eb8 100644 --- a/pfbudget/__init__.py +++ b/pfbudget/__init__.py @@ -1,3 +1,7 @@ +__all__ = ["run", "parse_data", "categorize_data"] +__author__ = "Luís Murta" +__version__ = "0.1" + from .categories import categorize_data from .parsers import parse_data from .runnable import run diff --git a/pfbudget/runnable.py b/pfbudget/runnable.py index 5e00372..027d4db 100644 --- a/pfbudget/runnable.py +++ b/pfbudget/runnable.py @@ -1,5 +1,6 @@ from pathlib import Path import argparse +import re from .categories import categorize_data from .database import DBManager @@ -41,7 +42,14 @@ def argparser() -> argparse.ArgumentParser: parser = argparse.ArgumentParser( description="does cool finance stuff", parents=[help] ) - parser.add_argument("--version") + parser.add_argument( + "--version", + action="version", + version=re.search( + r'__version__\s*=\s*[\'"]([^\'"]*)[\'"]', + open("pfbudget/__init__.py").read(), + ).group(1), + ) subparsers = parser.add_subparsers( dest="command", required=True, help="sub-command help"