server.py

from . import routes
from . import worker
from . import settings
from aiohttp import web


def prepare_app(app, pool):
    app.router.add_get("/", routes.index)
    app.router.add_get("/api/domains", routes.domains)
    app.router.add_post("/api/domains", routes.domains)
    app.router.add_get("/api/domains/stats", routes.stats)
    app.add_routes([web.static("/static", routes.STATIC_DIR)])
    app["pool"] = pool


async def on_shutdown(app):
    if app["pool"]:
        app["pool"].close()
        await app["pool"].wait_closed()


def start(port=None):
    app = web.Application(middlewares=settings.MIDDLEWARES)
    prepare_app(app, None)
    app.on_shutdown.append(on_shutdown)
    web.run_app(app, port=port)


if __name__ == "__main__":
    start()