Mercurial > code > home > repos > homeauto
annotate service/tomatoWifi/wifi.py @ 51:d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
Ignore-this: dc012ad1d539d3a1c37c779e63c8cf4b
| author | drewp@bigasterisk.com |
|---|---|
| date | Mon, 31 Dec 2012 15:01:04 -0800 |
| parents | 0ab069867c64 |
| children | 875a37be1228 |
| rev | line source |
|---|---|
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
1 import re, ast, logging, socket |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
2 from twisted.internet.defer import inlineCallbacks, returnValue |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
3 from cyclone.httpclient import fetch |
| 36 | 4 from rdflib import Literal, Graph |
| 0 | 5 |
| 6 log = logging.getLogger() | |
| 7 | |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
8 class Router(object): |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
9 def __repr__(self): |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
10 return repr(self.__dict__) |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
11 |
| 0 | 12 class Wifi(object): |
| 13 """ | |
| 14 gather the users of wifi from the tomato routers | |
| 15 """ | |
| 16 def __init__(self, tomatoConfig="/my/site/magma/tomato_config.js", | |
| 17 accessN3="/my/proj/openid_proxy/access.n3"): | |
| 18 | |
| 19 # ideally this would all be in the same rdf store, with int and | |
| 20 # ext versions of urls | |
| 21 | |
| 22 txt = open(tomatoConfig).read().replace('\n', '') | |
| 23 self.knownMacAddr = jsValue(txt, 'knownMacAddr') | |
| 24 tomatoUrl = jsValue(txt, 'tomatoUrl') | |
| 25 | |
| 26 g = Graph() | |
| 27 g.parse(accessN3, format="n3") | |
| 28 repl = {'/tomato1/' : None, '/tomato2/' : None} | |
| 29 for k in repl: | |
| 30 rows = list(g.query(''' | |
| 31 PREFIX p: <http://bigasterisk.com/openid_proxy#> | |
| 32 SELECT ?prefix WHERE { | |
| 33 [ | |
| 34 p:requestPrefix ?public; | |
| 35 p:proxyUrlPrefix ?prefix | |
| 36 ] | |
| 37 }''', initBindings={"public" : Literal(k)})) | |
| 38 repl[k] = str(rows[0][0]) | |
| 39 | |
| 40 self.routers = [] | |
| 41 for url in tomatoUrl: | |
| 42 name = url | |
| 43 for k, v in repl.items(): | |
| 44 url = url.replace(k, v) | |
| 45 | |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
46 r = Router() |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
47 r.url = url.replace('root:admin@', '') |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
48 r.headers = {'Authorization': ['Basic cm9vdDphZG1pbg==']} |
| 0 | 49 r.name = {'tomato1' : 'bigasterisk3', |
| 50 'tomato2' : 'bigasterisk4'}[name.split('/')[1]] | |
| 51 self.routers.append(r) | |
| 52 | |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
53 @inlineCallbacks |
| 0 | 54 def getPresentMacAddrs(self): |
| 55 aboutIp = {} | |
| 56 byMac = {} # mac : [ip] | |
| 57 | |
| 58 for router in self.routers: | |
| 59 log.debug("GET %s", router) | |
| 60 try: | |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
61 data = yield fetch(router.url, headers=router.headers, |
|
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
62 timeout=2) |
| 0 | 63 except socket.error: |
| 64 log.warn("get on %s failed" % router) | |
| 65 continue | |
| 66 | |
| 67 for (ip, mac, iface) in jsValue(data, 'arplist'): | |
| 68 aboutIp.setdefault(ip, {}).update(dict( | |
| 69 ip=ip, | |
| 70 router=router.name, | |
| 71 mac=mac, | |
| 72 iface=iface, | |
| 73 )) | |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
74 |
| 0 | 75 byMac.setdefault(mac, set()).add(ip) |
| 76 | |
| 77 for (name, ip, mac, lease) in jsValue(data, 'dhcpd_lease'): | |
| 78 if lease.startswith('0 days, '): | |
| 79 lease = lease[len('0 days, '):] | |
| 80 aboutIp.setdefault(ip, {}).update(dict( | |
| 81 router=router.name, | |
| 82 rawName=name, | |
| 83 mac=mac, | |
| 84 lease=lease | |
| 85 )) | |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
86 |
| 0 | 87 byMac.setdefault(mac, set()).add(ip) |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
88 |
| 0 | 89 for iface, mac, signal in jsValue(data, 'wldev'): |
| 90 matched = False | |
| 91 for addr in aboutIp.values(): | |
| 92 if (addr['router'], addr['mac']) == (router.name, mac): | |
| 93 addr.update(dict(signal=signal, iface=iface)) | |
| 94 matched = True | |
| 95 if not matched: | |
| 96 aboutIp["mac-%s-%s" % (router, mac)] = dict( | |
| 97 router=router.name, | |
| 98 mac=mac, | |
| 99 signal=signal, | |
| 100 ) | |
| 101 | |
| 102 ret = [] | |
| 103 for addr in aboutIp.values(): | |
| 104 if addr.get('ip') in ['192.168.1.1', '192.168.1.2', '192.168.0.2']: | |
| 105 continue | |
| 106 try: | |
| 107 addr['name'] = self.knownMacAddr[addr['mac']] | |
| 108 except KeyError: | |
| 109 addr['name'] = addr.get('rawName') | |
| 110 if addr['name'] in [None, '*']: | |
| 111 addr['name'] = 'unknown' | |
| 112 ret.append(addr) | |
| 113 | |
|
51
d2842eedd56d
rewrite tomatowifi from restkit to cyclone httpclient
drewp@bigasterisk.com
parents:
36
diff
changeset
|
114 returnValue(ret) |
| 0 | 115 |
| 116 | |
| 117 def jsValue(js, variableName): | |
| 118 # using literal_eval instead of json parser to handle the trailing commas | |
| 119 val = re.search(variableName + r'\s*=\s*(.*?);', js, re.DOTALL).group(1) | |
| 120 return ast.literal_eval(val) |