From 94f606a9b4c9b2922c867cf53955c4f87254ed96 Mon Sep 17 00:00:00 2001
From: Narayan Desai <desai@mcs.anl.gov>
Date: Wed, 15 Jul 2009 22:03:34 +0000
Subject: GroupPatterns: implement NodeRange (node[[001-096,099]]) support

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@5330 ce84e21b-d406-0410-9b95-82705330c041
---
 schemas/grouppatterns.xsd               |  2 ++
 src/lib/Server/Plugins/GroupPatterns.py | 51 +++++++++++++++++++++++++++++----
 2 files changed, 48 insertions(+), 5 deletions(-)

diff --git a/schemas/grouppatterns.xsd b/schemas/grouppatterns.xsd
index 9d11c9c13..5e2abed3f 100644
--- a/schemas/grouppatterns.xsd
+++ b/schemas/grouppatterns.xsd
@@ -11,6 +11,8 @@
     <xsd:sequence>
       <xsd:element name="NamePattern" type="xsd:string" minOccurs='1' 
                    maxOccurs='1'/>
+      <xsd:element name="NameRange" type="xsd:string" minOccurs='1'
+                   maxOccurs='1'/>
       <xsd:element name="Group" type="xsd:string" minOccurs='1' 
                    maxOccurs='1'/>
     </xsd:sequence>
diff --git a/src/lib/Server/Plugins/GroupPatterns.py b/src/lib/Server/Plugins/GroupPatterns.py
index 1a5d590c1..56195602b 100644
--- a/src/lib/Server/Plugins/GroupPatterns.py
+++ b/src/lib/Server/Plugins/GroupPatterns.py
@@ -2,13 +2,49 @@
 import re, lxml.etree
 import Bcfg2.Server.Plugin
 
+class PackedDigitRange(object):
+    def __init__(self, digit_range):
+        self.sparse = list()
+        self.ranges = list()
+        for item in digit_range.split(','):
+            if '-' in item:
+                self.ranges.append(tuple([int(x) for x in item.split('-')]))
+            else:
+                self.sparse.append(int(item))
+
+    def includes(self, other):
+        iother = int(other)
+        return iother in self.sparse or True in \
+               [iother in range(rng[0], rng[1]+1) for rng in self.ranges]
+
 class PatternMap(object):
-    def __init__(self, pattern, groupname):
+    range_finder = '\\[\\[[\d\-,]+\\]\\]'
+    def __init__(self, pattern, rangestr, groupname):
         self.pattern = pattern
-        self.re = re.compile(pattern)
+        self.rangestr = rangestr
         self.groupname = groupname
+        if pattern != None:
+            self.re = re.compile(pattern)
+            self.process = self.process_re
+        elif rangestr != None:
+            self.process = self.process_range
+            self.re = re.compile('^' + re.subn(self.range_finder, '(\d+)', rangestr)[0])
+            dmatcher = re.compile(re.subn(self.range_finder, '\\[\\[([\d\-,]+)\\]\\]', rangestr)[0])
+            self.dranges = [PackedDigitRange(x) for x in dmatcher.match(rangestr).groups()]
+        else:
+            raise Exception
+
+    def process_range(self, name):
+        match = self.re.match(name)
+        if not match:
+            return None
+        digits = match.groups()
+        for i in range(len(digits)):
+            if not self.dranges[i].includes(digits[i]):
+                return None
+        return self.groupname
 
-    def process(self, name):
+    def process_re(self, name):
         match = self.re.match(name)
         if not match:
             return None
@@ -32,9 +68,14 @@ class PatternFile(Bcfg2.Server.Plugin.SingleXMLFileBacked):
             return
         for entry in parsed.findall('GroupPattern'):
             try:
-                pat = entry.find('NamePattern').text
+                pat = None
+                rng = None
+                if entry.find('NamePattern') is not None:
+                    pat = entry.find('NamePattern').text
+                if entry.find('NameRange') is not None:
+                    rng = entry.find('NameRange').text
                 grp = entry.find('Group').text
-                self.patterns.append(PatternMap(pat, grp))
+                self.patterns.append(PatternMap(pat, rng, grp))
             except:
                 Bcfg2.Server.Plugin.logger.error(\
                     "GroupPatterns: Failed to initialize pattern %s" % \
-- 
cgit v1.2.3-1-g7c22