"""Payment Entity logic."""

from moneyhub import models


def get_vat_data_by_payment_entity_and_statement_period(
    payment_entity_id: int,
    statement_period_id: int
) -> dict:
    """Get vat data by payment entity.

    Args:
        payment_entity_id (int): id of the payment entity
        statement_period_id (int): id of the statement period
    Returns:
        dict: vat data activity
    """
    vat_data = bool(models.LedgerVatSummary.get_by_payment_entity_and_activity_statement_period(
        payment_entity_id, statement_period_id))
    if not vat_data:
        vat_data = bool(models.LedgerAccountingRunVat.get_by_payment_entity_and_statement_period(
            payment_entity_id, statement_period_id))
    return {'vat_data': vat_data}