[ENH] Better separation of different MT940 types.

This commit is contained in:
Ronald Portier (Therp BV)
2015-06-10 14:07:57 +02:00
parent 4807de4d6e
commit 449c00e4ef
3 changed files with 10 additions and 4 deletions

View File

@@ -101,7 +101,7 @@ class MT940(object):
At least, you should override handle_tag_61 and handle_tag_86.
Don't forget to call super.
handle_tag_* functions receive the remainder of the the line (that is,
without ':XX:') and are supposed to write into self.current_transaction
"""
@@ -110,6 +110,7 @@ class MT940(object):
"""Initialize parser - override at least header_regex.
This in fact uses the ING syntax, override in others."""
self.mt940_type = 'General'
self.header_lines = 3 # Number of lines to skip
self.header_regex = '^{1:[0-9A-Z]{25,25}}' # Start of relevant data
self.footer_regex = '^-}$|^-XXX$' # Stop processing on seeing this
@@ -123,8 +124,8 @@ class MT940(object):
if not bool(re.match(self.header_regex, line)):
raise ValueError(
'File starting with %s does not seem to be a'
' MT940 format bank statement.' %
data[:12]
' valid %s MT940 format bank statement.' %
(data[:12], self.mt940_type)
)
def parse(self, data):

View File

@@ -33,6 +33,11 @@ class MT940Parser(MT940):
r'(?P<reference>\w{1,50})'
)
def __init__(self):
"""Initialize parser - override at least header_regex."""
super(MT940Parser, self).__init__()
self.mt940_type = 'ING'
def handle_tag_61(self, data):
"""get transaction values"""
super(MT940Parser, self).handle_tag_61(data)

View File

@@ -38,8 +38,8 @@ class MT940Parser(MT940):
def __init__(self):
"""Initialize parser - override at least header_regex."""
super(MT940Parser, self).__init__()
self.mt940_type = 'RABO'
self.header_lines = 1 # Number of lines to skip
# Do not user $ for end of string below: line contains much
# more data than just the first line.
self.header_regex = '^:940:' # Start of relevant data