Merge pull request #9 from thaichat04/add-comment

Add comment in result parser

Author: Hong-Thai Nguyen

cfonb/parser/statement.py

@@ -45,10 +45,13 @@ def parse(self, file_obj):
                     while True:
                         index += 1
                         if not row.get('label_%s'%index):
-                            row['label_%s'%index] = new_row.label
+                            row['label_%s'%index] = new_row.get('label')
                             break
                 else:
                     row.update(new_row)
+                comment = line.split('   ')[2].strip()
+                if comment:
+                    row['comment'] = row.get('comment', '') + (BR_LINE if row.get('comment', '') else '') + comment
             elif line[0:2] == '07':
                 statement.footer = Row(line)
             else:

cfonb/tests/test_statement.py

@@ -50,6 +50,23 @@ def test_parse_with_original_content(self):
         assert_true(result[0].lines[0].get('origin').endswith('0515589    29701EUR2E01711467640B1230620     LIBPRELEVEMENTS SEPA DOMICILIES  '))
         assert_true(len(result[0].lines[0].get('origin').splitlines()) == 5)
 
+    def test_parse_with_comment_content(self):
+        content = '''0115589    29701EUR2 01711467640  220620GAEC DE STANG KERBAIL                             0000000352431O0000            
+0415589    29701EUR2E01711467640B1230620  220620PRLV ORANGE                             0 0000000000315}VOTRE ABONNEMENT
+0515589    29701EUR2E01711467640B1230620     NBEORANGE                                                                  
+0515589    29701EUR2E01711467640B1230620     LCCVOTRE ABONNEMENT FIXE  02XXXXX896 (FACTURE: XXXXX8960E2) - P            
+0515589    29701EUR2E01711467640B1230620     RCN298576896 985768960E235E                                                
+0515589    29701EUR2E01711467640B1230620     LIBPRELEVEMENTS SEPA DOMICILIES  
+0715589    29701EUR2 01711467640  230620                                                  0000000352746O                '''
+        reader = StatementReader()
+        result = reader.parse(StringIO(content))
+        assert_equal(len(result[0].lines), 1)
+        assert_true(result[0].lines[0].get('comment').startswith(
+            'NBEORANGE'))
+        assert_true(result[0].lines[0].get('comment').endswith(
+            'LIBPRELEVEMENTS SEPA DOMICILIES'))
+        assert_true(len(result[0].lines[0].get('comment').splitlines()) == 4)
+
 
 def suite():
     suite = unittest.TestSuite()

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-version = '2.3'
+version = '2.4'
 
 long_description = open('README.rst').read()
 
@@ -14,7 +14,7 @@
       author='Dhatim',
       author_email='[email protected]',
       url='https://github.com/dhatim/python-cfonb',
-      download_url = 'https://github.com/dhatim/python-cfonb/archive/2.3.tar.gz',
+      download_url = 'https://github.com/dhatim/python-cfonb/archive/2.4.tar.gz',
       license='LGPL',
       packages=find_packages(exclude=['ez_setup', 'examples', 'tests']),
       include_package_data=True,