Переглянути джерело

Make "gptbot -v" actually output the correct version

Kumi 11 місяців тому
батько
коміт
2b7813f715
1 змінених файлів з 9 додано та 2 видалено
  1. 9 2
      src/gptbot/__main__.py

+ 9 - 2
src/gptbot/__main__.py

@@ -5,11 +5,18 @@ from configparser import ConfigParser
 
 import signal
 import asyncio
-
+import pkg_resources
 
 def sigterm_handler(_signo, _stack_frame):
     exit()
 
+def get_version():
+    try:
+        package_version = pkg_resources.get_distribution("matrix_gptbot").version
+    except pkg_resources.DistributionNotFound:
+        return None
+    return package_version
+
 
 if __name__ == "__main__":
     # Parse command line arguments
@@ -25,7 +32,7 @@ if __name__ == "__main__":
         "-v",
         help="Print version and exit",
         action="version",
-        version="GPTBot v0.1.1",
+        version=f"GPTBot {get_version() or '- version unknown'}",
     )
     args = parser.parse_args()