summaryrefslogtreecommitdiffstats
path: root/src/lib/Server/Core.py
blob: b481033c44d1a128d241cc2c744b8329b8d29508 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/bin/python
# $Id$

from os import stat
from stat import ST_MODE, S_ISDIR
from syslog import syslog, LOG_ERR

from Error import PublishError
from Types import Clause

import _fam

class fam(object):
    '''The fam object contains alteration monitors'''
    
    def __init__(self):
        self.fm = _fam.open()
        self.users = {}
        self.handles = {}

    def fileno(self):
        return self.fm.fileno()

    def AddMonitor(self, path, obj=None):
        m = stat(path)[ST_MODE]
        if S_ISDIR(m):
            h = self.fm.monitorDirectory(path, None)
            self.handles[h.requestID()] = h
        else:
            h = self.fm.monitorFile(path, None)
            self.handles[h.requestID()] = h
        if obj != None:
            self.users[h.requestID()] = obj
        return h.requestID()

    def HandleEvent(self):
        event = self.fm.nextEvent()
        id = event.requestID
        if self.users.has_key(id):
            self.users[id].HandleEvent(event)

class PublishedValue(object):
    def __init__(self,owner,key,value):
        self.owner=owner
        self.key=key
        self.value=value

    def Update(self,owner,value):
        if owner != self.owner:
            raise PublishError, (self.key,owner)
        self.value=value

class Core(object):
    def __init__(self, repository, structures, generators):
        self.datastore = repository
        self.fam = fam()
        self.pubspace = {}
        self.structures = []
        for structure in structures:
            m = getattr(__import__("Bcfg2.Server.Structures.%s"%(structure)).Server.Structures, structure)
            s = getattr(m, structure)
            self.structures.append(s(self, self.datastore))
        self.generators = []
        for generator in generators:
            m=getattr(__import__("Bcfg2.Server.Generators.%s"%(generator)).Server.Generators,generator)
            g = getattr(m, generator)
            self.generators.append(g(self, self.datastore))
        # we need to inventory and setup generators
        # Process generator requirements
        for g in self.generators:
            for prq in g.__requires__:
                if not self.pubspace.has_key(prq):
                    raise GeneratorError, (g.name, prq)
            g.CompleteSetup()

    def PublishValue(self,owner,key,value):
        if not self.pubspace.has_key(key):
            # This is a new entry
            self.pubspace[key]=PublishedValue(owner,key,value)
        else:
            # This is an old entry. Update can fai
            try:
                self.pubspace[key].Update(owner,value)
            except PublishError,e:
                syslog(LOG_ERR, "Publish conflict for %s. Owner %s, Modifier %s"%(key,self.pubspace[key].owner,owner))

    def ReadValue(self,key):
        if self.pubspace.has_key(key):
            return self.pubspace[key].value
        raise KeyError,key

    def GetStructures(self, metadata):
        return reduce(lambda x,y:x+y, map(lambda x:x.Construct(metadata), self.structures))

    def BindStructure(self, structure, metadata):
        for entry in structure.getchildren():
            self.Bind(entry, metadata)

    def Bind(self, entry, metadata):
        g = [x for x in self.generators if x.__provides__.get(entry.tag, {}).has_key(entry.attrib['name'])]
        if len(g) == 1:
            return g[0].__provides__[entry.tag][entry.attrib['name']](entry, metadata)
        elif len(g) > 1:
            print "Data Integrity error for %s %s"%(entry.tag, entry.attrib['name'])
        else:
            for g in self.generators:
                if hasattr(g, "FindHandler"):
                    try:
                        return g.FindHandler(entry)(entry, metadata)
                    except:
                        print g, "failed"
            raise KeyError, (entry.tag,entry.attrib['name'])