ticketfrei3/kibicara/kibicara.py

66 lines
2 KiB
Python
Raw Normal View History

2020-07-01 19:39:13 +00:00
# Copyright (C) 2020 by Thomas Lindner <tom@dl6tom.de>
# Copyright (C) 2020 by Cathy Hu <cathy.hu@fau.de>
#
# SPDX-License-Identifier: 0BSD
2020-07-11 10:54:07 +00:00
""" Entrypoint of Kibicara. """
2020-07-01 19:39:13 +00:00
from asyncio import run as asyncio_run
from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from hypercorn.config import Config
from hypercorn.asyncio import serve
2020-07-15 14:46:13 +00:00
from kibicara.config import args, config
2020-07-01 19:39:13 +00:00
from kibicara.model import Mapping
from kibicara.platformapi import Spawner
from kibicara.webapi import router
2020-07-15 14:46:13 +00:00
from logging import basicConfig, DEBUG, getLogger, INFO, WARNING
2020-07-01 19:39:13 +00:00
logger = getLogger(__name__)
class Main:
2020-07-11 10:54:07 +00:00
""" Entrypoint for Kibicara.
Initializes the platform bots and starts the hypercorn webserver serving the
Kibicara application and the specified frontend on port 8000.
"""
2020-07-01 19:39:13 +00:00
def __init__(self):
2020-07-11 10:54:07 +00:00
asyncio_run(self.__run())
2020-07-01 19:39:13 +00:00
2020-07-11 10:54:07 +00:00
async def __run(self):
2020-07-15 14:46:13 +00:00
LOGLEVELS = {
None: WARNING,
1: INFO,
2: DEBUG,
}
basicConfig(
level=LOGLEVELS.get(args.verbose, DEBUG),
format="%(asctime)s %(name)s %(message)s",
)
2020-07-06 12:50:52 +00:00
getLogger('aiosqlite').setLevel(WARNING)
2020-07-01 19:39:13 +00:00
Mapping.create_all()
await Spawner.init_all()
2020-07-11 10:54:07 +00:00
await self.__start_webserver()
2020-07-01 19:39:13 +00:00
2020-07-11 10:54:07 +00:00
async def __start_webserver(self):
2020-07-01 19:39:13 +00:00
class SinglePageApplication(StaticFiles):
async def get_response(self, path, scope):
response = await super().get_response(path, scope)
if response.status_code == 404:
response = await super().get_response('.', scope)
return response
2020-07-01 19:39:13 +00:00
app = FastAPI()
server_config = Config()
server_config.accesslog = '-'
app.include_router(router, prefix='/api')
if config['frontend_path'] is not None:
app.mount(
'/',
app=SinglePageApplication(directory=config['frontend_path'], html=True),
)
2020-07-01 19:39:13 +00:00
await serve(app, server_config)