Add --version flag (#11)

inverse · Aug 22, 2022 · 917a15f · 917a15f
1 parent 36a8a6a
commit 917a15f
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+# 0.4.0
+
+- Add `--version` flag
+
 # 0.3.1
 
 - Validate `--status-code` is an integer

diff --git a/cert_host_scraper/cli.py b/cert_host_scraper/cli.py
@@ -1,11 +1,15 @@
 import logging
 import sys
+from pathlib import Path
 
 import click
 from requests import RequestException
 from rich.console import Console
 from rich.progress import track
 from rich.table import Table
+from single_source import get_version
+
+__version__ = get_version(__name__, Path(__file__).parent.parent)
 
 from cert_host_scraper import Options, Result, fetch_urls, validate_url
 
@@ -21,6 +25,7 @@ def validate_status_code(
 
 @click.group()
 @click.option("--debug", is_flag=True, help="Whether to enable debug level output")
+@click.version_option(__version__, message="%(prog)s: %(version)s")
 def cli(debug: bool):
     log_level = logging.DEBUG if debug else logging.INFO
     logging.basicConfig(level=log_level)

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,6 +20,7 @@ beautifulsoup4 = "^4.10.0"
 click = "^8.0.3"
 aiohttp = {extras = ["speedups"], version = "^3.8.1"}
 rich = "^11.0.0"
+single-source = "^0.3.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1.2"