From c3f85740d0302619d571c3067f57443d8849e265 Mon Sep 17 00:00:00 2001
From: Sven Klemm <sven@timescale.com>
Date: Sat, 25 Feb 2023 09:34:12 +0100
Subject: [PATCH] Add --version flag

---
 src/pgspot/cli.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/pgspot/cli.py b/src/pgspot/cli.py
index 1c61f1d..61daf6d 100755
--- a/src/pgspot/cli.py
+++ b/src/pgspot/cli.py
@@ -5,11 +5,18 @@
 from .visitors import visit_sql
 import sys
 
+from pgspot import __version__
+
 
 def run():
     parser = ArgumentParser(
         description="Spot vulnerabilities in PostgreSQL SQL scripts"
     )
+    parser.add_argument(
+        "--version",
+        action="version",
+        version="%(prog)s {version}".format(version=__version__),
+    )
     parser.add_argument(
         "-a",
         "--append",