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
|
import lxml.etree
import sqlalchemy
import sqlalchemy.orm
import Bcfg2.Server.Plugin
import Bcfg2.Server.Snapshots
from Bcfg2.Server.Snapshots.model import Snapshot
import time
class Snapshots(Bcfg2.Server.Plugin.Statistics,
Bcfg2.Server.Plugin.Plugin):
name = 'Snapshots'
experimental = True
def __init__(self, core, datastore):
Bcfg2.Server.Plugin.Plugin.__init__(self, core, datastore)
Bcfg2.Server.Plugin.Statistics.__init__(self)
self.session = Bcfg2.Server.Snapshots.setup_session()
def process_statistics(self, metadata, data):
return self.statistics_from_old_stats(metadata, data)
def statistics_from_old_stats(self, metadata, xdata):
# entries are name -> (modified, correct, start, desired, end)
# not sure we can get all of this from old format stats
t1 = time.time()
entries = dict([('Package', dict()),
('Service', dict()), ('Path', dict())])
extra = dict([('Package', dict()), ('Service', dict()),
('Path', dict())])
pdisp = {'Package': ['name', 'type', 'version'],
'Service': ['name', 'type', 'status']}
for entry in xdata.find('.//Bad'):
if entry.tag not in pdisp:
print "Not Found", entry.tag, entry.get('name')
continue
else:
edata = dict([(key, unicode(entry.get('current_%s' % key))) \
for key in pdisp[entry.tag]])
data = [False, False, edata, edata]
entries[entry.tag][entry.get('name')] = data
for entry in xdata.find('.//Modified'):
if entry.tag in pdisp:
if entry.get('name') in entries[entry.tag]:
entries[entry.tag][entry.get('name')][0] = True
else:
current = dict([(key, unicode(entry.get('current_%s' % key))) \
for key in pdisp[entry.tag]])
desired = dict([(key, unicode(entry.get(key))) \
for key in pdisp[entry.tag]])
data = [False, False, current, desired]
entries[entry.tag][entry.get('name')] = data
else:
print entry.tag, entry.get('name')
for entry in xdata.find('.//Extra'):
if entry.tag in pdisp:
current = dict([(key, unicode(entry.get(key))) for key in pdisp[entry.tag]])
extra[entry.tag][entry.get('name')] = current
else:
print "extra", entry.tag, entry.get('name')
t2 = time.time()
snap = Snapshot.from_data(self.session, metadata, entries, extra)
self.session.save(snap)
self.session.commit()
t3 = time.time()
return True
|