4 from monitor.wrapper import plc
5 from monitor.database.info.model import *
8 def dsites_from_lsites(l_sites):
12 if not site['login_base'] in d_sites:
13 d_sites[site['login_base']] = site
14 id2lb[site['site_id']] = site['login_base']
16 #print "Two sites have the same login_base value %s!" % site['login_base']
19 return (d_sites, id2lb)
21 def dsn_from_dsln(d_sites, id2lb, l_nodes):
26 if id2lb[id] not in lb2hn:
30 # this won't reach sites without nodes, which I guess isn't a problem.
31 if node['site_id'] in id2lb.keys():
32 login_base = id2lb[node['site_id']]
34 print "%s has a foreign site_id %s" % (node['hostname'],
38 print i, " ", id2lb[i]
39 raise Exception, "Node has missing site id!! %s %d" %(node['hostname'], node['site_id'])
40 if not login_base in dsn:
41 lb2hn[login_base] = []
43 dsn[login_base]['plc'] = d_sites[login_base]
44 dsn[login_base]['monitor'] = {} # event log, or something
46 hostname = node['hostname']
47 lb2hn[login_base].append(node)
48 dsn[login_base][hostname] = {}
49 dsn[login_base][hostname]['plc'] = node
50 dsn[login_base][hostname]['comon'] = {}
51 dsn[login_base][hostname]['monitor'] = {}
53 hn2lb[hostname] = login_base
54 return (dsn, hn2lb, lb2hn)
71 print "initing plccache"
73 dbsites = PlcSite.query.all()
74 l_sites = [ s.plc_site_stats for s in dbsites ]
77 dbnodes = PlcNode.query.all()
78 l_nodes = [ s.plc_node_stats for s in dbnodes ]
81 dbpcus = PlcPCU.query.all()
82 l_pcus = [ s.plc_pcu_stats for s in dbpcus ]
84 print "dsites_from_lsites"
85 (d_sites,id2lb) = dsites_from_lsites(l_sites)
87 (plcdb, hn2lb, lb2hn) = dsn_from_dsln(d_sites, id2lb, l_nodes)
95 def GetNodesByIds(ids):
98 node = PlcNode.get_by(node_id=node_id)
99 ret.append(node.plc_node_stats)
102 def GetNodesBySite(loginbase):
103 site = PlcSite.get_by(loginbase=loginbase)
104 return GetNodesByIds(site.plc_site_stats['node_ids'])
106 def GetNodeByName(hostname):
107 node = PlcNode.get_by(hostname=hostname)
108 return node.plc_node_stats
110 def GetSitesByName(sitelist):
112 for site in sitelist:
113 site = PlcSite.get_by(loginbase=site)
114 ret.append(site.plc_site_stats)
117 def GetSitesById(idlist):
119 for site_id in idlist:
120 site = PlcSite.get_by(site_id=site_id)
121 ret.append(site.plc_site_stats)
124 def deleteExtra(l_plc, objectClass=PlcSite, dbKey='loginbase', plcKey='login_base'):
125 dbobjs = objectClass.query.all()
126 dbobj_key = [ getattr(s, dbKey) for s in dbobjs ]
127 plcobj_key = [ s[plcKey] for s in l_plc ]
128 extra_key = set(dbobj_key) - set(plcobj_key)
129 for obj in extra_key:
130 print "deleting %s" % obj
131 dbobj = objectClass.get_by(**{dbKey : obj})
135 l_sites = plc.api.GetSites({'peer_id':None},
136 ['login_base', 'site_id', 'abbreviated_name', 'latitude',
137 'longitude', 'max_slices', 'slice_ids', 'node_ids',
138 'enabled', 'date_created' ])
139 l_nodes = plc.api.GetNodes({'peer_id':None},
140 ['hostname', 'node_id', 'ports', 'site_id', 'boot_state',
141 'version', 'last_updated', 'date_created', 'key',
142 'last_contact', 'pcu_ids', 'interface_ids'])
143 l_pcus = plc.api.GetPCUs()
147 dbsite = PlcSite.findby_or_create(site_id=site['site_id'])
148 dbsite.loginbase = site['login_base']
149 dbsite.date_checked = datetime.now()
150 dbsite.plc_site_stats = site
151 deleteExtra(l_sites, PlcSite, 'loginbase', 'login_base')
152 deleteExtra(l_sites, HistorySiteRecord, 'loginbase', 'login_base')
157 dbpcu = PlcPCU.findby_or_create(pcu_id=pcu['pcu_id'])
158 dbpcu.date_checked = datetime.now()
159 dbpcu.plc_pcu_stats = pcu
160 deleteExtra(l_pcus, PlcPCU, 'pcu_id', 'pcu_id')
161 deleteExtra(l_pcus, HistoryPCURecord, 'plc_pcuid', 'pcu_id')
162 deleteExtra(l_pcus, FindbadPCURecord, 'plc_pcuid', 'pcu_id')
167 dbnode = PlcNode.findby_or_create(node_id=node['node_id'])
168 dbnode.hostname = node['hostname']
169 dbnode.date_checked = datetime.now()
170 dbnode.plc_node_stats = node
171 deleteExtra(l_nodes, PlcNode, 'hostname', 'hostname')
172 deleteExtra(l_nodes, HistoryNodeRecord, 'hostname', 'hostname')
173 deleteExtra(l_nodes, FindbadNodeRecord, 'hostname', 'hostname')
180 if __name__ == '__main__':