4 from monitor.wrapper import plc
5 from monitor.database.info.model import *
11 if not str(obj[key]) in d:
12 d[str(obj[key])] = obj
14 print "Two objects have the same %s key %s!" % (key, obj[key])
18 def dpcus_from_lpcus(l_pcus):
19 d_pcus = d_from_l(l_pcus, 'pcu_id')
22 def dnodes_from_lnodes(l_nodes):
23 d_nodes = d_from_l(l_nodes, 'hostname')
26 def dsites_from_lsites(l_sites):
27 d_sites = d_from_l(l_sites, 'login_base')
30 def dsites_from_lsites_id(l_sites):
34 if not site['login_base'] in d_sites:
35 d_sites[site['login_base']] = site
36 id2lb[site['site_id']] = site['login_base']
38 #print "Two sites have the same login_base value %s!" % site['login_base']
41 return (d_sites, id2lb)
43 def dsn_from_dsln(d_sites, id2lb, l_nodes):
48 if id2lb[id] not in lb2hn:
52 # this won't reach sites without nodes, which I guess isn't a problem.
53 if node['site_id'] in id2lb.keys():
54 login_base = id2lb[node['site_id']]
56 print >>sys.stderr, "%s has a foreign site_id %s" % (node['hostname'],
60 print i, " ", id2lb[i]
61 raise Exception, "Node has missing site id!! %s %d" %(node['hostname'], node['site_id'])
62 if not login_base in dsn:
63 lb2hn[login_base] = []
65 dsn[login_base]['plc'] = d_sites[login_base]
66 dsn[login_base]['monitor'] = {} # event log, or something
68 hostname = node['hostname']
69 lb2hn[login_base].append(node)
70 dsn[login_base][hostname] = {}
71 dsn[login_base][hostname]['plc'] = node
72 dsn[login_base][hostname]['comon'] = {}
73 dsn[login_base][hostname]['monitor'] = {}
75 hn2lb[hostname] = login_base
76 return (dsn, hn2lb, lb2hn)
88 #print "IMPORTING PLCCACHE: ",
89 #traceback.print_stack()
96 print >>sys.stderr, "initing plccache"
98 print >>sys.stderr, "collecting plcsites"
99 dbsites = PlcSite.query.all()
100 l_sites = [ s.plc_site_stats for s in dbsites ]
102 print >>sys.stderr, "collecting plcnodes"
103 dbnodes = PlcNode.query.all()
104 l_nodes = [ s.plc_node_stats for s in dbnodes ]
106 print >>sys.stderr, "collecting plcpcus"
107 dbpcus = PlcPCU2.query.all()
111 for k in ['username', 'protocol', 'node_ids', 'ip',
112 'pcu_id', 'hostname', 'site_id', 'notes',
113 'model', 'password', 'ports']:
114 pcu[k] = getattr(s, k)
117 print >>sys.stderr, "building id2lb"
118 (d_sites,id2lb) = dsites_from_lsites_id(l_sites)
119 print >>sys.stderr, "building lb2hn"
120 (plcdb, hn2lb, lb2hn) = dsn_from_dsln(d_sites, id2lb, l_nodes)
128 def GetNodesByIds(ids):
131 node = PlcNode.get_by(node_id=node_id)
132 ret.append(node.plc_node_stats)
135 def GetNodesBySite(loginbase):
136 site = PlcSite.get_by(loginbase=loginbase)
137 return GetNodesByIds(site.plc_site_stats['node_ids'])
139 def GetNodeByName(hostname):
140 print "GetNodeByName %s" % hostname
141 node = PlcNode.get_by(hostname=hostname)
142 return node.plc_node_stats
144 def GetSitesByName(sitelist):
146 for site in sitelist:
147 site = PlcSite.get_by(loginbase=site)
148 ret.append(site.plc_site_stats)
151 def GetSitesById(idlist):
153 for site_id in idlist:
154 site = PlcSite.get_by(site_id=site_id)
155 ret.append(site.plc_site_stats)
158 def deleteExtra(l_plc, objectClass=PlcSite, dbKey='loginbase', plcKey='login_base'):
159 dbobjs = objectClass.query.all()
160 dbobj_key = [ getattr(s, dbKey) for s in dbobjs ]
161 plcobj_key = [ s[plcKey] for s in l_plc ]
162 extra_key = set(dbobj_key) - set(plcobj_key)
163 for obj in extra_key:
164 print >>sys.stderr, "deleting %s" % obj
165 dbobj = objectClass.get_by(**{dbKey : obj})
169 l_sites = plc.api.GetSites({'peer_id':None},
170 ['login_base', 'site_id', 'abbreviated_name', 'latitude',
171 'longitude', 'max_slices', 'slice_ids', 'node_ids',
172 'enabled', 'date_created' ])
173 l_nodes = plc.api.GetNodes({'peer_id':None},
174 ['hostname', 'node_id', 'ports', 'site_id', 'boot_state', 'run_level',
175 'version', 'last_updated', 'date_created', 'key',
176 'last_contact', 'pcu_ids', 'interface_ids'])
177 l_pcus = plc.api.GetPCUs()
179 print >>sys.stderr, "sync sites"
181 dbsite = PlcSite.findby_or_create(site_id=site['site_id'])
182 dbsite.loginbase = site['login_base']
183 dbsite.date_checked = datetime.now()
184 dbsite.plc_site_stats = site
185 deleteExtra(l_sites, PlcSite, 'loginbase', 'login_base')
186 deleteExtra(l_sites, HistorySiteRecord, 'loginbase', 'login_base')
189 print >>sys.stderr, "sync pcus"
191 dbpcu = PlcPCU2.findby_or_create(pcu_id=pcu['pcu_id'])
192 dbpcu.date_checked = datetime.now()
193 for key in pcu.keys():
194 print >>sys.stderr, "setting %s = %s" % (key, pcu[key])
195 setattr(dbpcu, key, pcu[key])
197 deleteExtra(l_pcus, PlcPCU2, 'pcu_id', 'pcu_id')
198 deleteExtra(l_pcus, HistoryPCURecord, 'plc_pcuid', 'pcu_id')
199 deleteExtra(l_pcus, FindbadPCURecord, 'plc_pcuid', 'pcu_id')
202 print >>sys.stderr, "sync nodes"
204 dbnode = PlcNode.findby_or_create(node_id=node['node_id'])
205 dbnode.hostname = node['hostname']
206 dbnode.date_checked = datetime.now()
207 dbnode.plc_node_stats = node
208 deleteExtra(l_nodes, PlcNode, 'node_id', 'node_id')
209 deleteExtra(l_nodes, HistoryNodeRecord, 'plc_nodeid', 'node_id')
210 deleteExtra(l_nodes, PlcNode, 'hostname', 'hostname')
211 deleteExtra(l_nodes, HistoryNodeRecord, 'hostname', 'hostname')
212 deleteExtra(l_nodes, FindbadNodeRecord, 'hostname', 'hostname')
219 if __name__ == '__main__':