from fastapi import APIRouter

from app.models.link import LinkBudgetRequest, LinkBudgetResponse
from app.routers.errors import as_501
from app.services import link as link_service

router = APIRouter()


@router.post("/budget", response_model=LinkBudgetResponse)
def budget(request: LinkBudgetRequest) -> LinkBudgetResponse:
    return as_501(lambda: link_service.link_budget(request))