routes.py

# pylint: disable=unused-import
# flake8: noqa: F401
import inspect
import sys
from resource.webapp.get_index import GetIndex
from resource.account.login import Login
from resource.engine.add_worker import AddWorker
from resource.engine.get_thread_count import GetThreadCount
from resource.engine.get_worker_count import GetWorkerCount
from resource.healthz import Healthz
from resource.log.get_cron_log import GetCronLog
from resource.module.get_modules import GetModules
from resource.module.run_module import RunModule


def set_routes(*args):

    plugins = args[0]

    # Create the webapp endpoints

    plugins["api"].add_resource(
        GetIndex,
        "/",
        resource_class_kwargs={"db": plugins["db"], "engine": plugins["engine"]}
    )

    # Build the endpoints imported in this file

    classes = inspect.getmembers(sys.modules[__name__], inspect.isclass)
    classes = [c for c in classes if c[0] != "GetIndex" and c[0] != "GetStatic"]

    for res in classes:

        # Build the endpoint

        endpoint = res[1].__module__.replace("resource.", ".", 1).replace(".", "/")

        functions = inspect.getmembers(res[1], predicate=inspect.isfunction)
        http_functions = [f for n, f in functions if n in ["get", "post"]]

        if len(http_functions) == 0:
            raise Exception(f"No http function found on resource {res[1].__module__}")
        if len(http_functions) > 1:
            raise Exception(f"Too much http functions found on resource {res[1].__module__}")

        function_args = [a for a in inspect.signature(http_functions[0]).parameters if a != "self" and a != "kwargs"]

        if len(function_args) > 1:
            raise Exception(f"Too much args for http function on resource {res[1].__module__}")

        if len(function_args) == 1:
            endpoint += f"/<{function_args[0]}>"

        # Build the args

        class_args = {a: plugins[a] for a in inspect.getfullargspec(res[1]).args if a != "self"}

        # Add the resource

        plugins["api"].add_resource(res[1], endpoint, resource_class_kwargs=class_args)

        # Add the resource in the doc

        plugins["docs"].register(res[1], resource_class_kwargs=class_args)