From 7cebb33898f81253f78ea237c3fb27bece5f776e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lu=C3=ADs=20Murta?= <luis@murta.dev>
Date: Fri, 23 Jul 2021 21:22:39 +0100
Subject: [PATCH] Adds '__version__' and completes --version option

---
 pfbudget/__init__.py |  4 ++++
 pfbudget/runnable.py | 10 +++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

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"