from polyfactory.factories.pydantic_factory import ModelFactory

from src.models import (
    AbacusState,
    Account,
    AccountPayableContract,
    AccountPayee,
    AccountPaymentDetails,
    AccountPaymentHold,
    AccountPaymentTerm,
    AccountPaymentTermDataloaderItem,
    ContractCloseBalance,
    Event,
    GetAccountsResponse,
    PayableBalanceAfterTax,
    PayableDetails,
    PayableDetailsResponse,
    PaymentAccount,
    PaymentAccountInstance,
    PaymentAllocationFlowthrough,
    PaymentAllocationFlowthroughResponse,
    PaymentAllocationFlowthroughUpdate,
    PaymentEntity,
    PaymentGroup,
    PaymentGroupPayment,
    PaymentMethodMinimum,
)


class EventFactory(ModelFactory[Event]):
    __model__ = Event


class PaymentGroupPaymentFactory(ModelFactory[PaymentGroupPayment]):
    __model__ = PaymentGroupPayment


class ContractCloseBalanceFactory(ModelFactory[ContractCloseBalance]):
    __model__ = ContractCloseBalance


class PayableBalanceAfterTaxFactory(ModelFactory[PayableBalanceAfterTax]):
    __model__ = PayableBalanceAfterTax


class AccountFactory(ModelFactory[Account]):
    __model__ = Account

    @classmethod
    def currency_code(cls) -> str:
        return cls.__faker__.currency_code()


class PaymentEntityFactory(ModelFactory[PaymentEntity]):
    __model__ = PaymentEntity


class PaymentMethodMinimumFactory(ModelFactory[PaymentMethodMinimum]):
    __model__ = PaymentMethodMinimum


class PaymentAccountFactory(ModelFactory[PaymentAccount]):
    __model__ = PaymentAccount


class AbacusStateFactory(ModelFactory[AbacusState]):
    __model__ = AbacusState


class AccountPaymentDetailsFactory(ModelFactory[AccountPaymentDetails]):
    __model__ = AccountPaymentDetails


class AccountPayableContractFactory(ModelFactory[AccountPayableContract]):
    __model__ = AccountPayableContract


class PaymentGroupFactory(ModelFactory[PaymentGroup]):
    """PaymentGroup factory."""

    __model__ = PaymentGroup


class GetAccountsResponseFactory(ModelFactory[GetAccountsResponse]):
    """GetAccountsResponse factory."""

    __model__ = GetAccountsResponse


class AccountPaymentHoldFactory(ModelFactory[AccountPaymentHold]):
    """AccountPaymentHold factory."""

    __model__ = AccountPaymentHold


class AccountPaymentTermFactory(ModelFactory[AccountPaymentTerm]):
    """AccountPaymentTerm factory."""

    __model__ = AccountPaymentTerm


class AccountPaymentTermDataloaderItemFactory(
    ModelFactory[AccountPaymentTermDataloaderItem]
):
    """AccountPaymentTermDataloaderItem factory."""

    __model__ = AccountPaymentTermDataloaderItem


class PayableDetailsResponseFactory(ModelFactory[PayableDetailsResponse]):
    """PayableDetailsResponse factory."""

    __model__ = PayableDetailsResponse


class PaymentAllocationFlowthroughFactory(ModelFactory[PaymentAllocationFlowthrough]):
    """PaymentAllocationFlowthrough factory."""

    __model__ = PaymentAllocationFlowthrough


class PaymentAllocationFlowthroughResponseFactory(
    ModelFactory[PaymentAllocationFlowthroughResponse]
):
    """PaymentAllocationFlowthroughResponse factory."""

    __model__ = PaymentAllocationFlowthroughResponse


class PaymentAllocationFlowthroughUpdateBodyFactory(
    ModelFactory[PaymentAllocationFlowthroughUpdate]
):
    """PaymentAllocationFlowthroughUpdateBody factory."""

    __model__ = PaymentAllocationFlowthroughUpdate


class PaymentAccountInstanceFactory(ModelFactory[PaymentAccountInstance]):
    """PaymentAccount factory."""

    __model__ = PaymentAccountInstance


class PayableDetailsFactory(ModelFactory[PayableDetails]):
    """PayableDetails factory."""

    __model__ = PayableDetails


class AccountPayeeFactory(ModelFactory[AccountPayee]):
    """AccountPayee factory."""

    __model__ = AccountPayee