39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from fastapi import APIRouter
|
|
from Events.Engine.abstract_class import CategoryCluster, MethodToEvent
|
|
|
|
|
|
class CreateRouterFromCluster:
|
|
|
|
def __init__(self, **kwargs):
|
|
self.prefix = kwargs.get("prefix")
|
|
self.tags = kwargs.get("tags")
|
|
self.include_in_schema = bool(kwargs.get("include_in_schema", True))
|
|
self.router = APIRouter(
|
|
prefix=self.prefix, tags=self.tags, include_in_schema=self.include_in_schema
|
|
)
|
|
|
|
|
|
class CreateEndpointFromCluster:
|
|
|
|
def __init__(self, **kwargs):
|
|
self.router: CategoryCluster = kwargs.get("router")
|
|
self.method_endpoint: MethodToEvent = kwargs.get("method_endpoint")
|
|
self.attach_router()
|
|
|
|
def attach_router(self):
|
|
method = getattr(self.router, self.method_endpoint.METHOD.lower())
|
|
|
|
# Create a unique operation ID based on the endpoint path, method, and a unique identifier
|
|
kwargs = {
|
|
"path": self.method_endpoint.URL,
|
|
"summary": self.method_endpoint.SUMMARY,
|
|
"description": self.method_endpoint.DESCRIPTION,
|
|
}
|
|
if (
|
|
hasattr(self.method_endpoint, "RESPONSE_MODEL")
|
|
and self.method_endpoint.RESPONSE_MODEL is not None
|
|
):
|
|
kwargs["response_model"] = self.method_endpoint.RESPONSE_MODEL
|
|
|
|
method(**kwargs)(self.method_endpoint.endpoint_callable)
|