[REF] refactor statement reader in order to support multi statement in one file

Author: sebastienbeau

README.rst

@@ -24,9 +24,10 @@ You can read a statement like this::
 
     >>> from os.path import join
     >>> statement_file = open(join('cfonb', 'tests', 'bank_statement.cfo'))
-    >>> from cfonb.parser.statement import Statement
-    >>> statement = Statement()
-    >>> statement.parse(statement_file)
+    >>> from cfonb import StatementReader
+    >>> reader = StatementReader()
+    >>> result = reader.parse(statement_file)
+    >>> statement = result[0]
 
 The statement has a header and a footer, which are both statement rows::
 

cfonb/__init__.py

@@ -1 +1,2 @@
 #
+from parser.statement import StatementReader

cfonb/parser/statement.py

@@ -2,32 +2,39 @@
 from cfonb.parser import Row, ParsingError
 
 
-class Statement(object):
-    """Satement file parser and container. Parse file object to corresponding
+
+
+class StatementReader(object):
+    """Statement file parser. Parse file object to corresponding
     row objects for further use. Offers useful method for reading, writing and
     comparing issues.
     """
 
-    def __init__(self):
-        self.header = None
-        self.footer   = None
-        self.lines  = list()
-
     def parse(self, file_obj):
-        file_lines  = file_obj.readlines()
-        # header and footer
-        self.header = Row(file_lines.pop(0))
-        self.footer = Row(file_lines.pop())
+        """ Parser a file object and return the list of bank statement
+        extracted from the file"""
 
-        if file_lines[0][0:2] != '04':
-            raise ParserError('the first line after the header must be a'
-                    '04 line. This line is invalid %s'%file_lines[0])
+        file_lines  = file_obj.readlines()
         # content
+        result = {}
         for index, line in enumerate(file_lines):
-            # parse line
-            if line[0:2] == '04':
+
+            if line[0:2] == '01':
+                row = Row(line)
+                
+                #If a statement already exist for the same account
+                #we updated it else we create a new own
+                if result.get(row.account_nb):
+                    statement = result[row.account_nb]
+                else:
+                    statement = Statement()
+                    result[row.account_nb] = statement
+                    statement.header = row
+
+            elif line[0:2] == '04':
                 row = Row(line)
-                self.lines.append(row)
+                statement.lines.append(row)
+            
             elif line[0:2] == '05':
                 new_row = Row(line)
                 if new_row.get('label') and row.get('label'):
@@ -39,8 +46,21 @@ def parse(self, file_obj):
                             break
                 else:
                     row.update(new_row)
+
+            elif line[0:2] == '07':
+                statement.footer = Row(line)
             else:
-                if line[0:2] in ['01', '07']:#we don't take care of subtotal
-                    continue
-                else:
-                    raise ParsingError('line %s is invalid: "%s"' % (index, line))
+                raise ParsingError('line %s is invalid: "%s"' % (index, line))
+        
+
+
+        return [result[key] for key in result]
+
+class Statement(object):
+
+    def __init__(self):
+        self.header = None
+        self.footer   = None
+        self.lines  = list()
+
+