"""Exchange rate schemas."""

from abacus_common_logic.marshalling.custom_fields import ma


class ExchangeRateSchema(ma.Schema):
    """Exchange rate display schema."""

    statement_period_id = ma.Integer(required=True)
    exchange_rate_id = ma.Integer(required=True)
    from_currency_code = ma.String(required=True)
    to_currency_code = ma.String(required=True)
    rate = ma.Decimal(required=True, as_string=True)


class ExchangeRatePostSchema(ma.Schema):
    """Exchange rate POST schema."""

    statement_period_id = ma.Integer(required=True)
    exchange_rates = ma.String(required=True)