Create ChargePayment setPayment method

Author: daltonfm-stark

.gitignore

@@ -5,4 +5,5 @@
 *.RET
 *.egg-info/*
 dist
+temp
 env/*

febraban/cnab240/itau/sispag/file/lot.py

@@ -24,6 +24,8 @@ def add(self, register):
         register.setPositionInLot(index=self.index)
         self.registers.append(register)
         self.amount += register.amountInCents()
+        if self._isBoletoPayment():
+            self.totalAmount += register.totalAmountInCents()
         if self._isNonBarCodeTax():
             self.otherAmount += register.otherAmountInCents()
             self.additionAmount += register.additionAmountInCents()
@@ -81,6 +83,9 @@ def toString(self):
                 additionSum=self.additionAmount,
                 totalSum=self.totalAmount,
             )
+        elif self._isBoletoPayment():
+            self.trailerLot.setSumOfValues(sum=self.totalAmount)
+
         else:
             self.trailerLot.setSumOfValues(sum=self.amount)
 
@@ -94,6 +99,9 @@ def toString(self):
     def _count(self, cls):
         return len([register for register in self.registers if isinstance(register, cls)])
 
+    def _isBoletoPayment(self):
+        return self.kind == PaymentKind.diverse and self.method in [PaymentMethod.itau, PaymentMethod.other]
+
     def _isNonBarCodeTax(self):
         return self.kind == PaymentKind.tribute and self.method in PaymentMethod.nonBarcodeTaxes()
 

febraban/cnab240/itau/sispag/payment/chargePayment.py

@@ -9,9 +9,21 @@ def __init__(self):
         self.segmentJ52 = SegmentJ52()
         self.amount = 0
         self.discountAmount = 0
-        self.addedAmount = 0
+        self.additionAmount = 0
         self.totalAmount = 0
 
+    def setPayment(self, **kwargs):
+        self.setSender(kwargs["sender"])
+        self.setReceiverTaxId(kwargs["receiverTaxId"])
+        self.setBarCode(kwargs["barCode"])
+        self.setIdentifier(kwargs["identifier"])
+        self.setScheduleDate(kwargs["scheduleDate"].strftime("%d%m%Y"))
+        self.setAmounts(
+            kwargs.get("discountAmount"),
+            kwargs.get("addedAmount"),
+            kwargs.get("totalAmount"),
+        )
+
     def toString(self):
         return "\r\n".join((
             self.segmentJ.content,
@@ -21,6 +33,9 @@ def toString(self):
     def amountInCents(self):
         return self.amount
 
+    def totalAmountInCents(self):
+        return self.totalAmount
+
     def setSender(self, user):
         """Sets the sender for the payment. The sender represents a user, its bank and its address."""
         self.segmentJ.setSenderBank(user.bank)
@@ -46,7 +61,7 @@ def setBarCode(self, barCode):
     def setAmounts(self, discountAmount, addedAmount, totalAmount):
         self.segmentJ.setAmounts(discountAmount, addedAmount, totalAmount)
         self.discountAmount = discountAmount
-        self.addedAmount = addedAmount
+        self.additionAmount = addedAmount
         self.totalAmount = totalAmount
 
     def setPositionInLot(self, index):

febraban/cnab240/itau/sispag/result/parser.py

@@ -1,4 +1,4 @@
-from libs.febraban.cnab240.libs.paymentType import NonBarcodeTaxPayment
+from .....cnab240.libs.paymentType import NonBarcodeTaxPayment
 from .occurrences import occurrences