"""Accounting Period Report marshmallow schema."""

from abacus_common_logic.marshalling.custom_fields import ma

from royalties.constants.constants import REPORT_TYPE


class AccountingPeriodReportPostSchema(ma.Schema):
    """Accounting Period Report schema for POST request."""

    report_type = ma.Enum(options=REPORT_TYPE, data_key='report_type', required=True)
    report_export_url = ma.String(required=True)


class AccountingPeriodReportSchema(AccountingPeriodReportPostSchema):
    """Accounting Period Report details schema."""

    accounting_period_report_id = ma.IntegerId(required=True)
    accounting_period_id = ma.IntegerId(required=True)