"""Payee model."""

from enum import Enum

from pydantic import BaseModel


class PayeeEntityType(Enum):
    account_payee = 'account_payee'
    payee = 'payee'

    def __str__(self) -> str:
        return self.value


class Payee(BaseModel):
    """Payee model."""

    payee_type: PayeeEntityType
    payee_id: str

    @property
    def url_basename(self) -> str:
        return str(self.payee_type).replace('_', '-')