Handle running outside a venv for the version command

thebeanogamer · May 26, 2023 · 10dc2bb · 10dc2bb
1 parent dee63d5
commit 10dc2bb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/hstsparser.py b/hstsparser.py
@@ -110,7 +110,10 @@ def date_round(date: datetime.datetime) -> datetime.datetime:
 group.add_argument("--firefox", action="store_true", help="Process a Firefox database")
 group.add_argument("--chrome", action="store_true", help="Process a Chrome database")
 
-__version__ = importlib.metadata.version("hstsparser")
+try:
+    __version__ = importlib.metadata.version("hstsparser")
+except importlib.metadata.PackageNotFoundError:
+    __version__ = "0.0.1"
 
 parser.add_argument('--version', action='version', version=f'%(prog)s {__version__}')