adulau/pdns-qof-server
1
0
Fork 0
mirror of https://github.com/adulau/pdns-qof-server.git synced 2024-11-22 10:07:10 +00:00
Code Issues Projects Releases Packages Wiki Activity

Basic REST API to pdns-qos-server using Python Tornado

Browse source
This commit is contained in:
Alexandre Dulaunoy 2013-12-24 14:14:01 +01:00
parent 6a0b4bf5e5
commit 2de1a7a364
1 changed files with 29 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
bin/qof-server.py
View file

@ -14,6 +14,11 @@
# #
# Copyright (c) 2013 Alexandre Dulaunoy - a@foo.be # Copyright (c) 2013 Alexandre Dulaunoy - a@foo.be
from datetime import date
import tornado.escape
import tornado.ioloop
import tornado.web
import iptools import iptools
import redis import redis
import json import json
@ -56,7 +61,7 @@ def getRecord(t = None):
if r.smembers(rec): if r.smembers(rec):
rrval = {} rrval = {}
for v in r.smembers(rec): for v in r.smembers(rec):
rdata = v.decode(encoding='UTF-8') rdata = v.decode(encoding='UTF-8').strip()
rrval['time_first'] = getFirstSeen(t1=t, t2=rdata) rrval['time_first'] = getFirstSeen(t1=t, t2=rdata)
rrval['time_last'] = getLastSeen(t1=t, t2=rdata) rrval['time_last'] = getLastSeen(t1=t, t2=rdata)
if rrval['time_first'] is None: if rrval['time_first'] is None:
@ -86,8 +91,27 @@ def JsonQOF(rrfound = None):
rrqof = rrqof + json.dumps(rr) + "\n" rrqof = rrqof + json.dumps(rr) + "\n"
return rrqof return rrqof
class InfoHandler(tornado.web.RequestHandler):
def get(self):
response = { 'version': 'git',
'software': 'pdns-qof-server' }
self.write(response)
if __name__ == "__main__": class QueryHandler(tornado.web.RequestHandler):
def get(self, q):
print ("query: "+q)
if iptools.ipv4.validate_ip(q) or iptools.ipv6.validate_ip(q):
for x in getAssociatedRecords(q):
self.write(JsonQOF(getRecord(x)))
else:
self.write(JsonQOF(getRecord(t = q.strip())))
application = tornado.web.Application([
(r"/query/(.*)",QueryHandler),
(r"/info", InfoHandler)
])
if __name__ == "test":
qq = ["foo.be", "8.8.8.8"] qq = ["foo.be", "8.8.8.8"]
@ -97,4 +121,7 @@ if __name__ == "__main__":
print (JsonQOF(getRecord(x))) print (JsonQOF(getRecord(x)))
else: else:
print (JsonQOF(getRecord(t = q))) print (JsonQOF(getRecord(t = q)))
else:
application.listen(8888)
tornado.ioloop.IOLoop.instance().start()