diff options
Diffstat (limited to 'junos-bootstrap/httpd/server_http.py')
| -rw-r--r-- | junos-bootstrap/httpd/server_http.py | 110 |
1 files changed, 110 insertions, 0 deletions
diff --git a/junos-bootstrap/httpd/server_http.py b/junos-bootstrap/httpd/server_http.py new file mode 100644 index 0000000..20e5de5 --- /dev/null +++ b/junos-bootstrap/httpd/server_http.py @@ -0,0 +1,110 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- + +from http.server import BaseHTTPRequestHandler, HTTPServer +from string import Template +import time +import psycopg2 +import psycopg2.extras +import sys + +def main(): + # + # Settings + # + settings = dict( + db = dict( + user = 'bootstrap', + password = 'asdf', + dbname = 'bootstrap', + host = 'localhost' + ), + http = dict( + host = 'localhost', + port = 9000 + ) + ) + + # + # Connect to DB + # + try: + connect_params = ("dbname='%s' user='%s' host='%s' password='%s'" % (settings['db']['dbname'], settings['db']['user'], settings['db']['host'], settings['db']['password'])) + conn = psycopg2.connect(connect_params) + # cur = conn.cursor() + cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor) + cur.execute("""SELECT * from switches""") + rows = cur.fetchall() + print ("\nSwitches in DB during server_http.py startup:") + for row in rows: + print (" --> %s, connected to %s port %s" % (row['hostname'], row['distro_name'], row['distro_phy_port'])) + + except (psycopg2.DatabaseError, psycopg2.OperationalError) as e: + print ('Error: %s' % e) + sys.exit(1) + + except: + print(sys.exc_info()[0]) + sys.exit(1) + + def template_get(model): + return open(model + '.template').read() + + def template_parse(template_src, hostname): + cur.execute("SELECT * FROM switches WHERE hostname = '%s'" % hostname) + if(cur.rowcount == 1): + row = cur.fetchall()[0] + print(' --> DB response ok, populating template') + d={ + 'hostname': row['hostname'], + 'distro_name': row['distro_name'], + 'distro_phy_port': row['distro_phy_port'], + 'mgmt_addr': row['mgmt_addr'], + 'mgmt_cidr': row['mgmt_cidr'], + 'mgmt_gw': row['mgmt_gw'], + 'mgmt_vlan': row['mgmt_vlan'] + } + return Template(template_src).safe_substitute(d) + else: + print(' --> No hits in DB for hostname "%s", cannot continue' % hostname) + return False + + class httpd(BaseHTTPRequestHandler): + def do_GET(self): + print('[%s] Incoming request: source:%s path:%s ' % (time.asctime(), self.client_address[0], self.path)) + if '/tg15-edge/' in self.path: + hostname = self.path.split('/tg15-edge/')[1] + if len(hostname) > 0: + print(' --> hostname "%s" accepted, fetching info from DB' % hostname) + template_parsed = template_parse(template_get('ex2200'), hostname) + if template_parsed: + print(' --> sending response to client') + self.send_response(200) + self.send_header("Content-type", "text/plain") + self.end_headers() + self.wfile.write(bytes(template_parsed, "utf-8")) + print(' --> success - %s bytes sent to client' % len(template_parsed)) + else: + print(' --> error - template could not be populated') + else: + print(' --> rejected due to missing hostname') + else: + print(' --> rejected due to bad path') + # silence stderr from BaseHTTPRequestHandler + # source: http://stackoverflow.com/questions/3389305/how-to-silent-quiet-httpserver-and-basichttprequesthandlers-stderr-output + def log_message(self, format, *args): + return + + httpd_instance = HTTPServer((settings['http']['host'], settings['http']['port']), httpd) + print("\n[%s] Server Starts - %s:%s" % (time.asctime(), settings['http']['host'], settings['http']['port'])) + + try: + httpd_instance.serve_forever() + except KeyboardInterrupt: + pass + + httpd_instance.server_close() + print("\n\n[%s] HTTP Server stopped\n" % time.asctime()) + +if __name__ == "__main__": + main() |