from marshmallow import Schema, fields

# This file contains the schema for penal charge operations
class PenalChargeSchema(Schema):
    channel = fields.Str(allow_none=True)
    transactionId = fields.Str(allow_none=True)
    fbnTransactionId = fields.Str(allow_none=True)
    debtId = fields.Str(allow_none=True)
    accountId = fields.Str(allow_none=True)
    penalCharge = fields.Float(required=True)
    customerId = fields.Str(allow_none=True)
    lienAmount = fields.Float(required=True)
    comment = fields.Str(allow_none=True)
    countryId = fields.Str(allow_none=True)

#represents the response schema for penal charge
class CollectPenalFeeResponseSchema(Schema):
    customerId = fields.Str(allow_none=True)
    transactionId = fields.Str(allow_none=True)
    amountCollected = fields.Float(required=True)
    accountId = fields.Str(allow_none=True)
    responseCode = fields.Str(allow_none=True)
    responseMessage = fields.Str(allow_none=True)