From b35d708ea1e56a8d79b2f4e5651c57ebc99e7f8d Mon Sep 17 00:00:00 2001
From: "Chris St. Pierre" <chris.a.st.pierre@gmail.com>
Date: Wed, 11 Jul 2012 10:48:44 -0400
Subject: added feature to allow clients to declare their version to server

---
 src/lib/Bcfg2/version.py | 115 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 115 insertions(+)
 create mode 100644 src/lib/Bcfg2/version.py

(limited to 'src/lib/Bcfg2/version.py')

diff --git a/src/lib/Bcfg2/version.py b/src/lib/Bcfg2/version.py
new file mode 100644
index 000000000..ac10dac94
--- /dev/null
+++ b/src/lib/Bcfg2/version.py
@@ -0,0 +1,115 @@
+import re
+
+__version__ = "1.3.0"
+
+class Bcfg2VersionInfo(tuple):
+    v_re = re.compile(r'(\d+)(\w+)(\d+)')
+
+    def __new__(cls, vstr):
+        (major, minor, rest) = vstr.split(".")
+        match = cls.v_re.match(rest)
+        if match:
+            micro, releaselevel, serial = match.groups()
+        else:
+            micro = rest
+            releaselevel = 'final'
+            serial = 0
+        return tuple.__new__(cls, [int(major), int(minor), int(micro),
+                                   releaselevel, int(serial)])
+
+    def __init__(self, vstr):
+        tuple.__init__(self)
+        self.major, self.minor, self.micro, self.releaselevel, self.serial = \
+            tuple(self)
+    
+    def __repr__(self):
+        return "(major=%s, minor=%s, micro=%s, releaselevel=%s, serial=%s)" % \
+            tuple(self)
+
+    def _release_cmp(self, r1, r2):
+        if r1 == r2:
+            return 0
+        elif r1 == "final":
+            return -1
+        elif r2 == "final":
+            return 1
+        elif r1 == "rc":
+            return -1
+        elif r2 == "rc":
+            return 1
+            # should never get to anything past this point
+        elif r1 == "pre":
+            return -1
+        elif r2 == "pre":
+            return 1
+        else:
+            # wtf?
+            return 0
+
+    def __gt__(self, version):
+        if version is None:
+            # older bcfg2 clients didn't report their version, so we
+            # handle this case specially and assume that any reported
+            # version is newer than any indeterminate version
+            return True
+        try:
+            for i in range(3):
+                if self[i] > version[i]:
+                    return True
+                elif self[i] < version[i]:
+                    return False
+            rel = self._release_cmp(self[3], version[3])
+            if rel < 0:
+                return True
+            elif rel > 0:
+                return False
+            if self[4] > version[4]:
+                return True
+            else:
+                return False
+        except TypeError:
+            return self > Bcfg2VersionInfo(version)
+
+    def __lt__(self, version):
+        if version is None:
+            # older bcfg2 clients didn't report their version, so we
+            # handle this case specially and assume that any reported
+            # version is newer than any indeterminate version
+            return False
+        try:
+            for i in range(3):
+                if self[i] < version[i]:
+                    return True
+                elif self[i] > version[i]:
+                    return False
+            rel = self._release_cmp(self[3], version[3])
+            if rel > 0:
+                return True
+            elif rel < 0:
+                return False
+            if self[4] < version[4]:
+                return True
+            else:
+                return False
+        except TypeError:
+            return self < Bcfg2VersionInfo(version)
+
+    def __eq__(self, version):
+        if version is None:
+            # older bcfg2 clients didn't report their version, so we
+            # handle this case specially and assume that any reported
+            # version is newer than any indeterminate version
+            return False
+        try:
+            rv = True
+            for i in range(len(self)):
+                rv &= self[i] == version[i]
+            return rv
+        except TypeError:
+            return self == Bcfg2VersionInfo(version)
+
+    def __ge__(self, version):
+        return not self < version
+
+    def __le__(self, version):
+        return not self > version
-- 
cgit v1.2.3-1-g7c22