update to new module path

(Logical change 1.43)


git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@257 ce84e21b-d406-0410-9b95-82705330c041

1 files changed, 79 insertions, 2 deletions

diff --git a/src/lib/Server/Generator.py b/src/lib/Server/Generator.py
index eb66404a2..1ea602947 100644
--- a/src/lib/Server/Generator.py
+++ b/src/lib/Server/Generator.py
@@ -1,8 +1,7 @@
 #!/usr/bin/env python
 # $Id$
 
-from Error import GeneratorError
-
+from elementtree.ElementTree import XML
 from syslog import syslog, LOG_ERR
 
 class Generator(object):
@@ -58,3 +57,81 @@ class Generator(object):
 
     def AcceptProbeData(self, client, probedata):
         return
+
+class FileBacked(object):
+    '''This object caches file data in memory.
+    HandleEvent is called whenever fam registers an event.
+    Index can parse the data into member data as required.
+    This object is meant to be used as a part of DirectoryBacked.'''
+    
+    def __init__(self, name):
+        self.name = name
+        self.HandleEvent()
+
+    def HandleEvent(self, event=None):
+        self.data = file(self.name).read()
+        self.Index()
+
+    def Index(self):
+        pass
+
+class DirectoryBacked(object):
+    '''This object is a coherent cache for a filesystem hierarchy of files.'''
+    __child__ = FileBacked
+
+    def __init__(self, name, fam):
+        self.name = name
+        self.fam = fam
+        self.entries = {}
+        self.inventory = False
+        fam.AddMonitor(name, self)
+
+    def __getitem__(self, key):
+        return self.entries[key]
+
+    def __iter__(self):
+        return self.entries.iteritems()
+
+    def AddEntry(self, name):
+        if self.entries.has_key(name):
+            print "got multiple adds"
+        else:
+            self.entries[name] = self.__child__('%s/%s'%(self.name, name))
+            self.entries[name].HandleEvent()
+
+    def HandleEvent(self, event):
+        action = event.code2str()
+        if action == 'exists':
+            if event.filename != self.name:
+                self.AddEntry(event.filename)
+        elif action == 'created':
+            self.AddEntry(event.filename)
+        elif action == 'changed':
+            self.entries[event.filename].HandleEvent(event)
+        elif action == 'deleted':
+            if self.entries.has_key(event.filename):
+                del self.entries[event.filename]
+        elif action in ['endExist']:
+            pass
+        else:
+            print "Got unknown event %s %s %s"%(event.requestID, event.code2str(), event.filename)
+
+class XMLFileBacked(FileBacked):
+    '''This object is a coherent cache for an XML file to be used as a part of DirectoryBacked.'''
+    __identifier__ = 'name'
+
+    def Index(self):
+        a = XML(self.data)
+        self.label = a.attrib[self.__identifier__]
+        self.entries = a.getchildren()
+
+    def __iter__(self):
+        return iter(self.entries)
+
+class SingleXMLFileBacked(XMLFileBacked):
+    '''This object is a coherent cache for an independent XML File.'''
+    def __init__(self,filename,fam):
+        XMLFileBacked.__init__(self, filename)
+        fam.AddMonitor(filename, self)
+
+