from marshmallow import Schema, fields

# Disbursement Schema
class DisbursementSchema(Schema):
    requestId = fields.Str(required=True, data_key="requestId")
    debtId = fields.Str(required=True, data_key="debtId")
    transactionId = fields.Str(required=True, data_key="transactionId")
    customerId = fields.Str(required=True, data_key="customerId")
    accountId = fields.Str(required=True, data_key="accountId")
    productId = fields.Str(required=True, data_key="productId")
    provideAmount = fields.Float(required=True, data_key="provideAmount")
    collectAmountInterest = fields.Float(required=False, data_key="collectAmountInterest")  # Optional
    collectAmountMgtFee = fields.Float(required=True, data_key="collectAmountMgtFee")
    collectAmountInsurance = fields.Float(required=True, data_key="collectAmountInsurance")
    collectAmountVAT = fields.Float(required=True, data_key="collectAmountVAT")
    countryId = fields.Str(required=True, data_key="countryId")
    comment = fields.Str(required=False, data_key="comment")  # Optional