53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
import threading
|
|
from wsgiref.simple_server import make_server, WSGIRequestHandler
|
|
from operator import itemgetter
|
|
from departures import Departure
|
|
from bottle import route, error, run, get, ServerAdapter, static_file
|
|
|
|
class Server(ServerAdapter):
|
|
server = None
|
|
def run(self, handler):
|
|
if self.quiet:
|
|
class QuietHandler(WSGIRequestHandler):
|
|
def log_request(self, *args, **kw):
|
|
pass
|
|
self.options['handler_class'] = QuietHandler
|
|
self.server = make_server(self.host, self.port, handler, **self.options)
|
|
self.server.serve_forever()
|
|
def stop(self):
|
|
self.server.shutdown()
|
|
|
|
class API:
|
|
|
|
def __init__(self, main):
|
|
|
|
@route('/departures/<stop_id>')
|
|
def departures(stop_id: str):
|
|
resp = []
|
|
for d in Departure.get(stop_id):
|
|
resp.append(d.json())
|
|
resp.sort(key=itemgetter("departure"))
|
|
return {'departures': resp}
|
|
|
|
@route('/stops')
|
|
def stop():
|
|
return {'stops': main.config["stops"]}
|
|
|
|
@get("/static/<file>")
|
|
def index(file: str):
|
|
return static_file(file, root="static")
|
|
|
|
@get("/")
|
|
def static():
|
|
return static_file("index.html", root="static")
|
|
|
|
@error(404)
|
|
def error404(err):
|
|
return ''
|
|
|
|
self.server = Server(host=main.config["http"]["host"], port=main.config["http"]["port"])
|
|
self.thread = threading.Thread(target=self.start)
|
|
self.thread.start()
|
|
|
|
def start(self):
|
|
run(quiet=True, server=self.server) |