| author | Tero Marttila <terom@paivola.fi> |
| Wed, 10 Oct 2012 22:45:50 +0300 | |
| changeset 3 | 5990b188c54b |
| parent 1 | 731d2df704f0 |
| child 4 | b09436772d46 |
| permissions | -rw-r--r-- |
| 0 | 1 |
from pvl.verkko import db, web |
2 |
||
3 |
from pvl.html import tags as html |
|
4 |
||
5 |
import socket # dns |
|
6 |
||
7 |
import logging; log = logging.getLogger('pvl.verkko.hosts')
|
|
8 |
||
|
1
731d2df704f0
fixup index + non-chunked response (?) + hosts + evil hardcoded db url
Tero Marttila <terom@paivola.fi>
parents:
0
diff
changeset
|
9 |
# XXX: this should actually be DHCPHost |
| 0 | 10 |
class Host (object) : |
11 |
DATE_FMT = '%Y%m%d' |
|
12 |
||
13 |
def __init__ (self, ip, mac, name=None) : |
|
14 |
self.ip = ip |
|
15 |
self.mac = mac |
|
16 |
self.name = name |
|
17 |
||
18 |
def render_mac (self) : |
|
19 |
if not self.mac : |
|
20 |
return None |
|
21 |
||
22 |
elif len(self.mac) > (6 * 2 + 5) : |
|
23 |
return u'???' |
|
24 |
||
25 |
else : |
|
26 |
return unicode(self.mac) |
|
27 |
||
28 |
def render_name (self) : |
|
29 |
if self.name : |
|
30 |
return self.name.decode('ascii', 'replace')
|
|
31 |
else : |
|
32 |
return None |
|
33 |
||
34 |
def when (self) : |
|
35 |
return '{frm} - {to}'.format(
|
|
36 |
frm = self.first_seen.strftime(self.DATE_FMT), |
|
37 |
to = self.last_seen.strftime(self.DATE_FMT), |
|
38 |
) |
|
39 |
||
40 |
def dns (self) : |
|
41 |
""" |
|
42 |
Reverse-DNS lookup. |
|
43 |
""" |
|
44 |
||
45 |
if not self.ip : |
|
46 |
return None |
|
47 |
||
48 |
sockaddrs = set(sockaddr for family, socktype, proto, canonname, sockaddr in socket.getaddrinfo(self.ip, 0, 0, 0, 0, socket.AI_NUMERICHOST)) |
|
49 |
||
50 |
for sockaddr in sockaddrs : |
|
51 |
try : |
|
52 |
host, port = socket.getnameinfo(sockaddr, socket.NI_NAMEREQD) |
|
53 |
except socket.gaierror : |
|
54 |
continue |
|
55 |
||
56 |
return host |
|
57 |
||
58 |
def __unicode__ (self) : |
|
59 |
return u"{host.ip} ({host.mac})".format(host=self)
|
|
60 |
||
61 |
db.mapper(Host, db.dhcp_hosts, properties=dict( |
|
62 |
id = db.dhcp_hosts.c.rowid, |
|
63 |
#_mac = db.dhcp_hosts.c.mac, |
|
64 |
#_name = db.dhcp_hosts.c.name, |
|
65 |
)) |
|
66 |
||
| 3 | 67 |
HOST_ATTRS = {
|
| 0 | 68 |
'id': Host.id, |
69 |
'ip': Host.ip, |
|
70 |
'mac': Host.mac, |
|
71 |
'name': Host.name, |
|
72 |
'seen': Host.last_seen, |
|
73 |
} |
|
| 3 | 74 |
HOST_SORT = Host.last_seen.desc() |
| 0 | 75 |
|
76 |
def render_hosts (hosts, title=None) : |
|
77 |
COLS = ( |
|
78 |
#title sort |
|
79 |
('#', None),
|
|
80 |
('IP', 'ip'),
|
|
81 |
('MAC', 'mac'),
|
|
82 |
('Hostname', 'name'),
|
|
83 |
('Seen', 'seen'),
|
|
84 |
) |
|
85 |
||
86 |
return html.table( |
|
87 |
html.caption(title) if title else None, |
|
88 |
html.thead( |
|
89 |
html.tr( |
|
90 |
html.th( |
|
91 |
html.a(href='?sort={name}'.format(name=sort))(title) if sort else (title)
|
|
92 |
) for title, sort in COLS |
|
93 |
) |
|
94 |
), |
|
95 |
html.tbody( |
|
96 |
html.tr(class_=('alternate' if i % 2 else None), id=host.id)(
|
|
97 |
html.td(class_='id')( |
|
98 |
html.a(href='/hosts/{host.id}'.format(host=host))(
|
|
99 |
'#' #host['rowid']) |
|
100 |
) |
|
101 |
), |
|
102 |
html.td(class_='ip')( |
|
103 |
html.a(href='/hosts/ip/{host.ip}'.format(host=host))(host.ip)
|
|
104 |
), |
|
105 |
html.td(class_='mac')( |
|
106 |
html.a(href='/hosts/mac/{host.mac}'.format(host=host))(
|
|
107 |
host.render_mac() |
|
108 |
) |
|
109 |
), |
|
110 |
html.td(host.render_name()), |
|
111 |
html.td(host.when()), |
|
112 |
) for i, host in enumerate(hosts) |
|
113 |
) |
|
114 |
) |
|
115 |
||
116 |
def render_host (host, hosts) : |
|
117 |
title = 'DHCP Host: {host}'.format(host=host)
|
|
118 |
||
119 |
attrs = ( |
|
120 |
('IP', host.ip),
|
|
121 |
('MAC', host.mac),
|
|
122 |
('Hostname', host.name),
|
|
123 |
('DNS', host.dns()),
|
|
124 |
) |
|
125 |
||
126 |
return ( |
|
127 |
html.h2('Host'),
|
|
128 |
html.dl( |
|
129 |
(html.dt(title), html.dd(value)) for title, value in attrs |
|
130 |
), |
|
131 |
||
132 |
html.h2('Related'),
|
|
133 |
render_hosts(hosts), |
|
134 |
||
|
1
731d2df704f0
fixup index + non-chunked response (?) + hosts + evil hardcoded db url
Tero Marttila <terom@paivola.fi>
parents:
0
diff
changeset
|
135 |
html.a(href='/hosts')(html('«'), 'Back'),
|
| 0 | 136 |
) |
137 |
||
| 3 | 138 |
class Handler (web.Handler) : |
139 |
TITLE = "DHCP Hosts" |
|
| 0 | 140 |
|
| 3 | 141 |
def index (self) : |
142 |
return render_hosts(self.hosts) |
|
| 0 | 143 |
|
| 3 | 144 |
def host (self, id) : |
145 |
host = self.hosts.get(id) |
|
| 0 | 146 |
|
| 3 | 147 |
if not host : |
148 |
raise web.NotFound("No such host: {id}".format(id=id))
|
|
149 |
||
150 |
hosts = self.hosts.filter((Host.ip == host.ip) | (Host.mac == host.mac)) |
|
151 |
||
152 |
# XXX |
|
153 |
#self.title = "DHCP Host: {host}".format(host=unicode(host))
|
|
154 |
||
155 |
return render_host(host, hosts) |
|
| 0 | 156 |
|
| 3 | 157 |
def list (self, attr, value) : |
| 0 | 158 |
# fake host |
159 |
host = { 'ip': None, 'mac': None, 'name': None }
|
|
| 3 | 160 |
|
161 |
if attr not in HOST_ATTRS : |
|
162 |
raise web.BadRequest("Invalid attribute: {attr}".format(attr=attr))
|
|
163 |
||
| 0 | 164 |
host[attr] = value |
165 |
||
166 |
host = Host(**host) |
|
167 |
||
168 |
# query |
|
| 3 | 169 |
attr = HOST_ATTRS[attr] |
| 0 | 170 |
log.debug("%s == %s", attr, value)
|
171 |
||
| 3 | 172 |
hosts = self.hosts.filter(attr == value) |
| 0 | 173 |
|
| 3 | 174 |
# XXX |
175 |
#self.title = "DHCP Hosts: {value}".format(value=value)
|
|
| 0 | 176 |
|
| 3 | 177 |
return render_host(host, hosts) |
178 |
||
179 |
def process (self) : |
|
180 |
hosts = self.db.query(Host) |
|
| 0 | 181 |
|
| 3 | 182 |
# sort ? |
183 |
sort = self.request.args.get('sort')
|
|
| 0 | 184 |
|
| 3 | 185 |
if sort : |
186 |
sort = HOST_ATTRS[sort] |
|
187 |
else : |
|
188 |
sort = HOST_SORT |
|
| 0 | 189 |
|
| 3 | 190 |
log.debug("sort: %s", sort)
|
191 |
||
192 |
hosts = hosts.order_by(sort) |
|
193 |
||
194 |
# store |
|
195 |
self.hosts = hosts |
|
196 |
||
197 |
def render (self) : |
|
198 |
# index |
|
199 |
if not self.path : |
|
200 |
return self.index() |
|
201 |
||
202 |
# id |
|
203 |
elif len(self.path) == 1 : |
|
204 |
try : |
|
205 |
id, = self.path |
|
206 |
id = int(id) |
|
207 |
except ValueError as ex : |
|
208 |
raise web.BadRequest("Invalid host ID: {id}: {ex}".format(id=id, ex=ex))
|
|
209 |
||
210 |
return self.host(id) |
|
211 |
||
212 |
# query |
|
213 |
elif len(self.path) == 2 : |
|
214 |
attr, value = self.path |
|
215 |
||
216 |
return self.list(attr, value) |
|
217 |
||
218 |
else : |
|
219 |
raise web.NotFound |
|
220 |