server.py

import arq

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


def prepare_app(app, pool):
    app.router.add_get("/api/domains", routes.domains)
    app.router.add_post("/api/domains", routes.domains)

    app["pool"] = pool
    app["crawler"] = worker.Crawler(
        redis_settings=arq.RedisSettings(**settings.REDIS_CONFIG)
    )


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()