"""Payment Entity Handler."""

from fastapi import APIRouter

from moneyhub.logic import payment_entity as logic
from moneyhub.schemas.payment_entity import PaymentEntityVatDataActivity

payment_entity_router = APIRouter(prefix='/payment-entity', tags=['payment_entity'])


@payment_entity_router.get(
    '/{payment_entity_id}/statement-period/{statement_period_id}/vat',
    status_code=200
)
def get_vat_data_by_payment_entity_and_statement_period(
    payment_entity_id: int,
    statement_period_id: int
) -> PaymentEntityVatDataActivity:
    """Check vat data by payment entity and statement period.

    Args:
        payment_entity_id (int): id of the payment entity
        statement_period_id (int): id of the statement period
    Returns:
        PaymentEntityVatDataActivity: vat data activity

    """
    return logic.get_vat_data_by_payment_entity_and_statement_period(
        payment_entity_id, statement_period_id)