Refactored sample file generation scripts

Author: matheuscferraz

sample-charge-payment.py

@@ -2,7 +2,8 @@
 # coding: utf-8
 
 from febraban.cnab240.itau.sispag import ChargePayment, File
-from febraban.cnab240.libs.barCode import BarCode
+from febraban.cnab240.itau.sispag.file.lot import Lot
+from febraban.cnab240.libs.barCode import BarCodeJ
 from febraban.cnab240.user import User, UserAddress, UserBank
 
 
@@ -23,24 +24,41 @@
     )
 )
 
-barCode = BarCode("34112345600000012311090000250501234123454000")
-taxId = "12345678901234"
+barCodes = [
+    "34112318100000000011090027345767456112367000",
+    "34112318100000000021090027345687456112367000",
+    "34112318100000000031090027345507456112367000",
+]
+
+taxIds = [
+    "12345678901234",
+    "12345678901234",
+    "12345678901234",
+]
 
 file = File()
-file.setHeaderLotType(
-     kind="98",   # Tipo de pagamento - Diversos
-     method="30"  # Pagamento de Boleto mesmo banco = 30; Pagamento de Boleto outro banco = 31
-)
 file.setSender(sender)
 
-payment = ChargePayment()
-payment.setSender(sender)
-payment.setScheduleDate("08062019")
-payment.setIdentifier("ID1234567890")
-payment.setBarCode(barCode)
-payment.setReceiverTaxId(taxId)
+lot = Lot()
+sender.name = "SENDER NAME"
+lot.setSender(sender)
+lot.setHeaderLotType(
+    kind="98",
+    method="30"
+)
+
+for barCodeString, taxId in zip(barCodes, taxIds):
+    barCode = BarCodeJ(barCodeString)
+
+    payment = ChargePayment()
+    payment.setSender(sender)
+    payment.setScheduleDate("26022020")
+    payment.setIdentifier("ID1234567890")
+    payment.setBarCode(barCode)
+    payment.setReceiverTaxId(taxId)
 
-file.add(register=payment)
+    lot.add(register=payment)
 
-file.output(fileName="output3.REM", path="/../../")
+file.addLot(lot)
+file.output(fileName="output.REM", path="/../../")
 

sample-transfer.py

@@ -2,6 +2,7 @@
 # coding: utf-8
 
 from febraban.cnab240.itau.sispag import Transfer, File
+from febraban.cnab240.itau.sispag.file.lot import Lot
 from febraban.cnab240.user import User, UserAddress, UserBank
 
 
@@ -22,34 +23,53 @@
     )
 )
 
-receiver = User(
+receiver1 = User(
     name="RECEIVER NAME HERE",
-    identifier="12345678901",
+    identifier="01234567890",
     bank=UserBank(
-        bankId="033",
+        bankId="341",
         branchCode="1234",
         accountNumber="123456",
-        accountVerifier="7"
+        accountVerifier="9"
     )
 )
 
-file = File()
-file.setHeaderLotType(
-    kind="20",  #Tipo de pagamento - Diversos
-    method="01" #TED - Outra titularidade
+receiver2 = User(
+    name="RECEIVER NAME HERE",
+    identifier="01234567890",
+    bank=UserBank(
+        bankId="341",
+        branchCode="1234",
+        accountNumber="123456",
+        accountVerifier="9"
+    )
 )
+
+receivers = [receiver1, receiver2]
+
+file = File()
 file.setSender(sender)
 
-payment = Transfer()
-payment.setSender(sender)
-payment.setReceiver(receiver)
-payment.setAmountInCents("12000")
-payment.setScheduleDate("12102017")
-payment.setInfo(
-    reason="10"  #Crédito em Conta Corrente
+lot = Lot()
+sender.name = "SENDER NAME"
+lot.setSender(sender)
+lot.setHeaderLotType(
+    kind="20",  #Tipo de pagamento - Fornecedores
+    method="01" #TED - Outra titularidade
 )
-payment.setIdentifier("ID1234567890")
-file.add(register=payment)
 
+for receiver in receivers:
+    payment = Transfer()
+    payment.setSender(sender)
+    payment.setReceiver(receiver)
+    payment.setAmountInCents("10000")
+    payment.setScheduleDate("06052020")
+    payment.setInfo(
+        reason="10"  #Crédito em Conta Corrente
+    )
+    payment.setIdentifier("ID1234567890")
+    lot.add(register=payment)
+
+file.addLot(lot)
 file.output(fileName="output.REM", path="/../../")
 

sample-utility-payment.py

@@ -2,6 +2,7 @@
 # coding: utf-8
 
 from febraban.cnab240.itau.sispag import UtilityPayment, File
+from febraban.cnab240.itau.sispag.file.lot import Lot
 from febraban.cnab240.libs.barCode import LineNumberO
 from febraban.cnab240.user import User, UserAddress, UserBank
 
@@ -23,24 +24,37 @@
     )
 )
 
-lineNumber = LineNumberO("846800000012349701090114004112370844901232603900")
+lineNumbers = [
+    "836000000015522801380034710717288116000212965610",
+    "836400000011079401380070610530261110080671594119",
+    "836100000022434201380079213410922115000104587191",
+]
 
 file = File()
-file.setHeaderLotType(
-     kind="98",   # Tipo de pagamento - Diversos
-     method="13"  # Concessionarias
-)
 file.setSender(sender)
 
-payment = UtilityPayment()
-payment.setPayment(
-    sender=sender,
-    scheduleDate="18032020",
-    identifier="ID1234567890",
-    lineNumber=lineNumber
+lot = Lot()
+lot.setSender(sender)
+lot.setHeaderLotType(
+    kind="98",
+    method="13"
 )
 
-file.add(register=payment)
+for lineNumber in lineNumbers:
+    print lineNumber
+
+    lineNumber = LineNumberO(lineNumber)
+
+    payment = UtilityPayment()
+    payment.setPayment(
+        sender=sender,
+        scheduleDate="20032020",
+        identifier="ID1234567890",
+        lineNumber=lineNumber
+    )
+
+    lot.add(register=payment)
 
+file.addLot(lot)
 file.output(fileName="output.REM", path="/../../")