From: Jason Ish <ish@unx.ca>
Date: Thu, 30 Nov 2017 21:57:22 +0000 (-0600)
Subject: include git revision in version output
X-Git-Tag: 1.0.0a1~28
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=64bc308d4e050edf6acfefa182704075f5d4be44;p=thirdparty%2Fsuricata-update.git

include git revision in version output
---

diff --git a/.gitignore b/.gitignore
index 6cde500..17b6cf9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,3 +7,6 @@ _work
 
 # Development update.yaml
 /update.yaml
+
+# The file containing the git revision.
+/suricata/update/revision.py*
diff --git a/setup.py b/setup.py
index e51d8ff..351f2bb 100644
--- a/setup.py
+++ b/setup.py
@@ -1,10 +1,22 @@
+import subprocess
 from setuptools import setup
 
-import suricata.update
+from suricata.update.version import version
+
+def write_revision():
+    try:
+        revision = subprocess.check_output(
+            ["git", "rev-parse", "--short", "HEAD"])
+        with open("./suricata/update/revision.py", "w") as fileobj:
+            fileobj.write("revision = '%s'" % (revision.strip()))
+    except Exception as err:
+        print("Failed to get current git revision: %s" % (err))
+
+write_revision()
 
 setup(
     name="suricata-update",
-    version=suricata.update.version,
+    version=version,
     description="Suricata Update Tool",
     author="Jason Ish",
     author_email="ish@unx.ca",
diff --git a/suricata/update/__init__.py b/suricata/update/__init__.py
index 9f7743d..e69de29 100644
--- a/suricata/update/__init__.py
+++ b/suricata/update/__init__.py
@@ -1 +0,0 @@
-version = "1.0.0.dev"
diff --git a/suricata/update/main.py b/suricata/update/main.py
index 9df9757..2918aba 100644
--- a/suricata/update/main.py
+++ b/suricata/update/main.py
@@ -52,6 +52,12 @@ from suricata.update import util
 from suricata.update import sources
 from suricata.update import commands
 
+from suricata.update.version import version
+try:
+    from suricata.update.revision import revision
+except:
+    revision = "unknown"
+
 # Initialize logging, use colour if on a tty.
 if len(logging.root.handlers) == 0 and os.isatty(sys.stderr.fileno()):
     logger = logging.getLogger()
@@ -1159,9 +1165,8 @@ def _main():
     if args.quiet:
         logger.setLevel(logging.WARNING)
 
-    logger.debug("This is suricata-update version %s; Python: %s" % (
-        suricata.update.version,
-        sys.version.replace("\n", "- ")))
+    logger.debug("This is suricata-update version %s (rev: %s); Python: %s" % (
+        version, revision, sys.version.replace("\n", "- ")))
 
     config = Config(args)
     try:
@@ -1189,7 +1194,7 @@ def _main():
         return dump_sample_configs()
 
     if args.version:
-        print("suricata-update version %s" % suricata.update.version)
+        print("suricata-update version %s (rev: %s)" % (version, revision))
         return 0
 
     if args.help:
diff --git a/suricata/update/version.py b/suricata/update/version.py
new file mode 100644
index 0000000..fd3222d
--- /dev/null
+++ b/suricata/update/version.py
@@ -0,0 +1,6 @@
+# Version format:
+# Release: 1.0.0
+# Beta: 1.0.0b1
+# Alpha: 1.0.0a1
+# Development: 1.0.0.dev0
+version = "1.0.0.dev0"