HTTPServer.py

import folium
import sqlite3
from tornado.web import (Application,
                         RequestHandler)
from tornado.httpserver import HTTPServer
from tornado.ioloop import IOLoop
from random import randint

LPORT = 8081
db = sqlite3.connect('scan.db', timeout=5)
cursor = db.cursor()

def get_status(code):
    code = int(code)

    if (code == -1):
        return "None"

    elif (code == 0):
        return "Authenticate"

    elif (code == 1):
        return "Associated"

    elif (code == 2):
        return "IBSS JOINED"

    else:
        return "Error!"

def get_location(location_string, signal):
    (latitude, longitude) = location_string.strip().split(',')
    latitude = float(latitude)
    longitude = float(longitude)

    return(latitude, longitude)

def get_vendor(mac):
    mac = mac.replace(':', '').upper()
    db = sqlite3.connect('mac-vendor.db')
    cursor = db.cursor()

    cursor.execute('SELECT vendor FROM vendors WHERE mac = {} LIMIT 1;'.format(repr(mac)))

    result = cursor.fetchone()

    db.close()

    if (result == None):
        return "Unknown"

    return result[0]

class Handler(RequestHandler):
    def get(self):
        self.render('index.html')

class THandler(RequestHandler):
    def get(self):
        bssid2loc = self.get_argument('target', None)
        cursor.execute(''.join([
                "CREATE TABLE IF NOT EXISTS router(",
                "id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,",
                "bssid TEXT NOT NULL,",
                "ssid TEXT NOT NULL,",
                "frequency INTEGER NOT NULL,",
                "signal INTEGER NOT NULL,",
                "seen_ms_ago INTEGER NOT NULL,",
                "status INTEGER NOT NULL,",
                "registred TEXT NOT NULL,",
                "location TEXT);"
            
            ]))
        cursor.execute('SELECT * FROM router;')
        result = cursor.fetchall()
        marks = {}
        Map = None

        for _ in result:
            bssid = _[1]
            vendor = get_vendor(bssid[:8])
            location = get_location(_[8], _[4])
            color_for_mark = 'red'

            if (bssid2loc == bssid):
                Map = folium.Map(location=location)
                color_for_mark = 'blue'

            html = ['<h4>Information of {} ({})</h4>'.format(bssid, vendor),
                    '<strong>ID:</strong> {}'.format(_[0]),
                    '<strong>BSSID:</strong> {}'.format(bssid),
                    '<strong>SSID:</strong> {}'.format(_[2]),
                    '<strong>Frequency:</strong> {}'.format(_[3]),
                    '<strong>Signal:</strong> {}'.format(_[4]),
                    '<strong>Seen ms ago:</strong> {}'.format(_[5]),
                    '<strong>Status:</strong> {}'.format(get_status(_[6])),
                    '<strong>Registred:</strong> {}'.format(_[7]),
                    '<strong>Location:</strong> {},{}'.format(*location)
            ]

            iframe = folium.IFrame('<br>'.join(html), width=300, height=200)
            mark = folium.Marker(location=location,
                                 popup=folium.Popup(iframe, max_width=300, sticky=True),
                                 icon=folium.Icon(color=color_for_mark, icon_color='white', icon='map-marker', prefix='fa'),
                                 draggable=True)
            if (marks.get(vendor) == None):
                marks[vendor] = []
            
            marks[vendor].append(mark)

        if not (Map):
            Map = folium.Map()

        for i in marks:
            group_name = folium.FeatureGroup(name=i)

            for j in marks[i]:
                j.add_to(group_name)

            group_name.add_to(Map)

        folium.LayerControl().add_to(Map)

        self.write(Map._repr_html_())

app = Application([(r'/', Handler), 
                   (r'/tracked', THandler)])

try:
    httpd = HTTPServer(app, ssl_options={
                                'keyfile'   : 'keyfile',
                                'certfile'  : 'certfile'

                            })

except Exception as Except:
    print('Certificate or password does not exist, you aren\'t using it!. Exception:', Except)
    httpd = HTTPServer(app)

httpd.listen(LPORT)

print("Listening on", LPORT, '...')

try:
    IOLoop().current().start()

except KeyboardInterrupt:
    db.close()