Refactory test & add generate statement

Author: Hong-Thai Nguyen

cfonb/tests/test_statement.py

@@ -1,10 +1,9 @@
 # python import - http://docs.python.org/library/unittest.html
 import unittest
-from StringIO import StringIO
+from io import StringIO
 
 # import cfonb module
-from cfonb.parser import Row, ParsingError, statement as p
-
+from cfonb.parser import Row, statement as p
 
 HEAD_LINE = '0130002    00447     0000888899H  160811                                                  0000000132303H                '
 

cfonb/tests/test_transfert.py

@@ -7,7 +7,7 @@
 
 
 def _print(res):
-    print res.replace('\r\n','\\r\\n').replace(' ','.')
+    print(res.replace('\r\n','\\r\\n').replace(' ','.'))
 
 
 class TestTransfert(unittest.TestCase):

cfonb/writer/common.py

@@ -15,4 +15,8 @@ def save(header, body, footer, filename=None):
         f = open(filename,'w')
         f.write(content)
         f.close()
-    return content
+    return content
+
+
+def date_format(date):
+    return date.strftime(format='%d%m') + date.strftime(format='%y')[1:]

cfonb/writer/statement.py

@@ -21,6 +21,92 @@
   - l'entête (01)
   - ligne détail (04)
   - ligne total (facultative)
-  
-Voir <http://segs.free.fr/Fichiers/CFONB%20-%20Structure%20des%20fichiers%20ETEBAC3.pdf>
-"""
+
+"""
+from cfonb.writer.common import write, date_format, save
+
+
+class Statement(object):
+    _header = {}
+    _content = []
+    _footer = {}
+
+    def header(self, bank_code, agency_code, currency, account_number, date, amount):
+        self._header['bank_code'] = bank_code
+        self._header['agency_code'] = agency_code
+        self._header['currency'] = currency
+        self._header['account_number'] = account_number
+        self._header['date'] = date
+        self._header['amount'] = amount
+        return self
+
+    def footer(self, bank_code, agency_code, currency, account_number, date, amount):
+        self._footer['bank_code'] = bank_code
+        self._footer['agency_code'] = agency_code
+        self._footer['currency'] = currency
+        self._footer['account_number'] = account_number
+        self._footer['date'] = date
+        self._footer['amount'] = amount
+        return self
+
+    def add(self, bank_code, operation_code, agency_code, currency,
+            account_number, date, label, amount, reference):
+        line = write('04', 2)
+        line += write(bank_code, 5)
+        line += write(operation_code, 4)
+        line += write(agency_code, 5)
+        line += write(currency, 3)
+        line += write('2', 1) # number of decimal
+        line += write(' ', 1) # SIT code
+        line += write(account_number, 11)
+        line += write('  ', 2) # interbank code
+        line += write(date_format(date), 6)
+        line += write('  ', 2) # rejected code
+        line += write(date_format(date), 6)
+        line += write(label, 31)
+        line += write('  ', 2) # reserve zone
+        line += write('       ', 7) # entry writing code
+        line += write(' ', 1) # exoneration code
+        line += write(' ', 1)  # reserve zone
+        line += write(amount, 14)
+        line += write(reference, 16)
+        self._content.append(line)
+        return self
+
+    def render(self, filename=None):
+        return save(self._header(), '\r\n'.join(self._content), self._footer(), filename)
+
+    def _header(self):
+        if self._header:
+            line = write('01', 2)
+            line += write(self._header['bank_code'], 5)
+            line += write('    ', 4) # reserved zone
+            line += write(self._header['agency_code', 5])
+            line += write(self._header['currency', 3])
+            line += write('2', 1) # number of decimal
+            line += write(' ', 1)  # reserved zone
+            line += write(self._header['account_number'], 11)
+            line += write('  ', 2)  # reserved zone
+            line += write(date_format(self._header['date']), 6)
+            line += write(' ', 50)  # reserved zone
+            line += write(self._header['amount'], 14)
+            line += write(' ', 16)  # reserved zone
+            return line
+        return ''
+
+    def _footer(self):
+        if self._footer:
+            line = write('07', 2)
+            line += write(self._footer['bank_code'], 5)
+            line += write('    ', 4) # reserved zone
+            line += write(self._footer['agency_code', 5])
+            line += write(self._footer['currency', 3])
+            line += write('2', 1) # number of decimal
+            line += write(' ', 1)  # reserved zone
+            line += write(self._footer['account_number'], 11)
+            line += write('  ', 2)  # reserved zone
+            line += write(date_format(self._footer['date']), 6)
+            line += write(' ', 50)  # reserved zone
+            line += write(self._footer['amount'], 14)
+            line += write(' ', 16)  # reserved zone
+            return line

cfonb/writer/transfert.py

@@ -16,7 +16,7 @@
 You should have received a copy of the GNU Lesser General Public License
 along with CFONB.  If not, see <http://www.gnu.org/licenses/>.
 """
-from cfonb.writer.common import write, save
+from cfonb.writer.common import write, save, date_format
 
 """
  Coryright 2011, Stéphane Planquart <[email protected]>
@@ -50,15 +50,15 @@ def __init__(self):
         self.total = 0
         pass
 
-    def setEmeteurInfos(self,num_emeteur,raisonsocial,reference,
-                             guichet,num_compte,banque,datevir):
+    def setEmeteurInfos(self,num_emeteur, raisonsocial, reference,
+                             guichet, num_compte, banque, datevir):
         self._emeteur['num_emetteur']   = str(num_emeteur)
         self._emeteur['raisonsocial']   = raisonsocial
         self._emeteur['reference']      = reference
         self._emeteur['guichet']        = guichet
         self._emeteur['num_compte']     = num_compte
         self._emeteur['banque']         = banque
-        self._datevir                   = datevir.strftime(format='%d%m') + datevir.strftime(format='%y')[1:]
+        self._datevir                   = date_format(datevir)
         return self
 
     def add(self, reference, raisonsocial, domiciliation,
@@ -140,7 +140,7 @@ def _footer(self):
         content += write("", 11)
         #[zone E]Montant : les 16 positions contiennent le montant centimes
         #                  compris (00 s'il y a lieu) cadré à droite , non signé, complété à gauche par des zéros
-        content += write(self.total, 16, rpad=True, caract='0')
+        content += write(self.total, 16, rpad=True, fill_char='0')
         #[zone F]Réservée 31 caractères
         content += write("", 31)
         #[zone G1]Réservée 5 caractères
@@ -179,13 +179,13 @@ def _add(self,reference, raisonsocial,domiciliation,
         #[zone E]Montant : les 16 positions contiennent le montant centimes
         #        compris (00 s'il y a lieu) cadré à droite , non signé, complété à
         #        gauche par des zéros
-        content += write(montant, 16, rpad=True, caract='0')
+        content += write(montant, 16, rpad=True, fill_char='0')
         #[zone F]Libellé : 31 caractères à la disposition du client émetteur
         #        pour indication du motif et des références de l'opération
         content += write(libelle, 31)
         #[zone G1]Code établissement destinataire 5 chiffres
         content += write(etablissement, 5)
         #[zone G2]Zone réservée de 6 caractères
         content += write("", 6)
-        content += "\r\n";
+        content += "\r\n"
         return content