4 from monitor.wrapper import plc
5 from monitor.database.info.model import *
7 def dsites_from_lsites(l_sites):
11 if not site['login_base'] in d_sites:
12 d_sites[site['login_base']] = site
13 id2lb[site['site_id']] = site['login_base']
15 #print "Two sites have the same login_base value %s!" % site['login_base']
18 return (d_sites, id2lb)
20 def dsn_from_dsln(d_sites, id2lb, l_nodes):
25 if id2lb[id] not in lb2hn:
29 # this won't reach sites without nodes, which I guess isn't a problem.
30 if node['site_id'] in id2lb.keys():
31 login_base = id2lb[node['site_id']]
33 print "%s has a foreign site_id %s" % (node['hostname'],
37 print i, " ", id2lb[i]
38 raise Exception, "Node has missing site id!! %s %d" %(node['hostname'], node['site_id'])
39 if not login_base in dsn:
40 lb2hn[login_base] = []
42 dsn[login_base]['plc'] = d_sites[login_base]
43 dsn[login_base]['monitor'] = {} # event log, or something
45 hostname = node['hostname']
46 lb2hn[login_base].append(node)
47 dsn[login_base][hostname] = {}
48 dsn[login_base][hostname]['plc'] = node
49 dsn[login_base][hostname]['comon'] = {}
50 dsn[login_base][hostname]['monitor'] = {}
52 hn2lb[hostname] = login_base
53 return (dsn, hn2lb, lb2hn)
71 dbsites = PlcSite.query.all()
72 l_sites = [ s.plc_site_stats for s in dbsites ]
74 dbnodes = PlcNode.query.all()
75 l_nodes = [ s.plc_node_stats for s in dbnodes ]
77 dbpcus = PlcPCU.query.all()
78 l_pcus = [ s.plc_pcu_stats for s in dbpcus ]
80 (d_sites,id2lb) = dsites_from_lsites(l_sites)
81 (plcdb, hn2lb, lb2hn) = dsn_from_dsln(d_sites, id2lb, l_nodes)
89 def GetNodesByIds(ids):
92 node = PlcNode.get_by(node_id=node_id)
93 ret.append(node.plc_node_stats)
96 def GetNodesBySite(loginbase):
97 site = PlcSite.get_by(loginbase=loginbase)
98 return GetNodesByIds(site.plc_site_stats['node_ids'])
100 def GetNodeByName(hostname):
101 node = PlcNode.get_by(hostname=hostname)
102 return node.plc_node_stats
104 def GetSitesByName(sitelist):
106 for site in sitelist:
107 site = PlcSite.get_by(loginbase=site)
108 ret.append(site.plc_site_stats)
112 l_sites = plc.api.GetSites({'peer_id':None},
113 ['login_base', 'site_id', 'abbreviated_name', 'latitude',
114 'longitude', 'max_slices', 'slice_ids', 'node_ids',
115 'enabled', 'date_created' ])
116 l_nodes = plc.api.GetNodes({'peer_id':None},
117 ['hostname', 'node_id', 'ports', 'site_id',
118 'version', 'last_updated', 'date_created',
119 'last_contact', 'pcu_ids', 'nodenetwork_ids'])
120 l_pcus = plc.api.GetPCUs()
124 dbsite = PlcSite.findby_or_create(site_id=site['site_id'])
125 dbsite.loginbase = site['login_base']
126 dbsite.date_checked = datetime.now()
127 dbsite.plc_site_stats = site
129 # TODO: delete old records.
134 dbnode = PlcNode.findby_or_create(node_id=node['node_id'])
135 dbnode.hostname = node['hostname']
136 dbnode.date_checked = datetime.now()
137 dbnode.plc_node_stats = node
139 # TODO: delete old records.
144 dbpcu = PlcPCU.findby_or_create(pcu_id=pcu['pcu_id'])
145 dbpcu.date_checked = datetime.now()
146 dbpcu.plc_pcu_stats = pcu
148 # TODO: delete old records.
155 if __name__ == '__main__':