1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
#!/usr/bin/env python3
import requests,traceback
from jinja2 import Template,Environment,FileSystemLoader,TemplateNotFound
import json
import http.server
from enum import Enum
endpoints = "read/networks read/oplog read/snmp read/switches-management public/distro-tree public/config public/dhcp public/dhcp-summary public/ping public/switches public/switch-state".split()
objects = dict()
def getEndpoint(endpoint):
r = requests.get("http://localhost:8080/api/%s" % endpoint)
if (r.status_code != 200):
raise Exception("Bad status code for endpoint %s: %s" % (endpoint, r.status_code))
return r.json()
def updateData():
for a in endpoints:
objects[a] = getEndpoint(a)
env = Environment(loader=FileSystemLoader(['templates/','/opt/gondul/data/templates', '/opt/gondul/web/templates']),lstrip_blocks=True, trim_blocks=True)
class Mode(Enum):
Get = 1
Post = 2
class MyHandler(http.server.BaseHTTPRequestHandler):
options = dict()
def parse_options(self):
self.url = self.path[1:]
self.options = dict()
if self.url.find("?") != -1:
(self.url, tmpoptions) = self.url.split("?")
tmptuples = tmpoptions.split("&")
for a in tmptuples:
(x,y) = a.split("=")
self.options[x] = y
def generic(self, mode):
updateData()
self.parse_options()
body = ""
try:
if mode == Mode.Get:
template = env.get_template(self.url)
elif mode == Mode.Post:
length = self.headers.get('content-length')
if not length:
length = 0
content = self.rfile.read(int(length)).decode('UTF-8')
template = env.from_string(content)
else:
raise Exception("Invalid mode")
body = template.render(objects=objects, options=self.options)
self.send_response(200)
except TemplateNotFound as err:
body = "Template \"%s\" not found\n" % self.url
self.send_response(404)
except Exception as err:
body = ("Templating of \"%s\" failed to render. Most likely due to an error in the template. Error transcript:\n\n%s\n----\n\n%s\n" % (self.url, err, traceback.format_exc()))
if mode == Mode.Get:
self.send_response(400)
else:
self.send_response(500)
finally:
self.send_header('Cache-Control','max-age=5, s-maxage=1')
body = body.encode('UTF-8')
self.send_header('Content-Length', int(len(body)))
self.end_headers()
self.wfile.write(body)
self.wfile.flush()
def do_GET(self):
self.generic(Mode.Get)
def do_POST(self):
self.generic(Mode.Post)
def run(server_class=http.server.HTTPServer, handler_class=http.server.BaseHTTPRequestHandler):
server_address = ('localhost', 8081)
httpd = server_class(server_address, handler_class)
httpd.serve_forever()
run(handler_class=MyHandler)
|
