from marshmallow import Schema, fields

class DisbursementSchema(Schema):
    transactionId = fields.Str(required=False, allow_none=True)
    FbnTransactionId = fields.Str(required=False, allow_none=True)
    debtId = fields.Str(required=False, allow_none=True)
    customerId = fields.Str(required=False, allow_none=True)
    accountId = fields.Str(required=False, allow_none=True)
    productId = fields.Str(required=False, allow_none=True)
    provideAmount = fields.Float(required=True)
    collectAmountInterest = fields.Float(required=True)
    collectAmountMgtFee = fields.Float(required=True)
    collectAmountInsurance = fields.Float(required=True)
    collectAmountVAT = fields.Float(required=True)
    countryId = fields.Str(required=False, allow_none=True)
    comment = fields.Str(required=False, allow_none=True)


class DisburseLoanResponseSchema(Schema):
    transactionId = fields.Str(allow_none=True)
    fbnTransactionId = fields.Str(allow_none=True)
    debtId = fields.Str(allow_none=True)
    customerId = fields.Str(allow_none=True)
    accountId = fields.Str(allow_none=True)
    productId = fields.Str(allow_none=True)
    provideAmount = fields.Float(required=True)
    collectAmountInterest = fields.Float(required=True)
    collectAmountMgtFee = fields.Float(required=True)
    collectAmountInsurance = fields.Float(required=True)
    collectAmountVAT = fields.Float(required=True)
    countryId = fields.Str(allow_none=True)
    responseCode = fields.Str(allow_none=True)
    responseMessage = fields.Str(allow_none=True)