1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
#!/usr/bin/env python3
# This file is part of ofxstatement-austrian.
# See README.rst for more information.
import csv
import re
from ofxstatement import statement
from ofxstatement.parser import CsvStatementParser
from ofxstatement.plugin import Plugin
from ofxstatement.statement import generate_transaction_id
from ofxstatement.plugins.utils \
import clean_multiple_whitespaces, fix_amount_string
class EasybankCsvParser(CsvStatementParser):
"""The csv parser for Easybank (base)."""
date_format = "%d.%m.%Y"
def split_records(self):
"""Split records using a custom dialect."""
return csv.reader(self.fin, delimiter=";")
class EasybankCreditCardCsvParser(EasybankCsvParser):
"""The csv parser for Easybank (credit card)."""
mappings = {
"memo": 1,
"id": 2,
"date": 3,
"amount": 5,
}
def parse(self):
"""Parse."""
stmt = super(EasybankCreditCardCsvParser, self).parse()
statement.recalculate_balance(stmt)
return stmt
def parse_record(self, line):
"""Parse a single record."""
# Split the description into two parts and save it to the line list.
parts = line[1].split('|')
# 3 parts: Description, foreign language, transaction id
# 2 parts: Description, transaction id
if len(parts) == 3:
line[1] = "{} ({})".format(parts[0], parts[1])
else:
line[1] = parts[0]
line.insert(2, parts[-1])
# Account id
if not self.statement.account_id:
self.statement.account_id = line[0]
# Currency
if not self.statement.currency:
self.statement.currency = line[6]
# Cleanup amount
line[5] = fix_amount_string(line[5])
line[1] = clean_multiple_whitespaces(line[1])
# Create statement and fixup missing parts
stmtline = super(EasybankCreditCardCsvParser, self).parse_record(line)
stmtline.trntype = 'DEBIT' if stmtline.amount < 0 else 'CREDIT'
return stmtline
class EasybankGiroCsvParser(EasybankCsvParser):
"""The csv parser for Easybank (giro)."""
mappings = {
"check_no": 1,
"memo": 2,
"payee": 3,
"date": 4,
"amount": 6,
}
reg_description = re.compile(r'[A-Z]{2}/000[0-9]{6}')
reg_iban = re.compile(
r'([A-Z]{6}[A-Z0-9]{2}[^\s]*)?\s?([A-Z]{2}[0-9]{10,34})\s(.*)')
reg_legacy = re.compile(r'(.*)([0-9]{5,})\s([0-9]{6,})(.*)')
def extract_check_no(self, description):
'''Try to extract the statement check_no.'''
result = ''
mo = self.reg_description.search(description)
if mo:
result = str(int(mo.group(0).split('/')[1]))
return result
def extract_description(self, description):
'''Cleanup description from a giro account.'''
# extract iban/bic, account number, ...
parts = [x.strip() for x in self.reg_description.split(description)]
# parts: memo, transaction
if not parts[1]:
return parts[0], parts[0]
# parts: memo, transaction, banking information
else:
# extract iban, bic and text
iban_bic = self.reg_iban.search(parts[1])
if iban_bic:
# iban, bic and text
if iban_bic.group(1):
result = '{0} ({1} {2})'.format(iban_bic.group(3),
iban_bic.group(2),
iban_bic.group(1))
# iban only
else:
result = '{0} ({1})'.format(
iban_bic.group(3), iban_bic.group(2))
return parts[0], result
# extract legacy banking number
account_number = self.reg_legacy.search(parts[1])
if account_number:
if account_number.group(1):
text = account_number.group(1).strip()
else:
text = account_number.group(4).strip()
return parts[0], '{0} ({1} {2})'.format(
text, account_number.group(3), account_number.group(2))
# Could not extract anything useful, return parts as is.
return parts[0], parts[1]
def parse(self):
"""Parse."""
stmt = super(EasybankGiroCsvParser, self).parse()
statement.recalculate_balance(stmt)
return stmt
def parse_record(self, line):
"""Parse a single record."""
# Extract check_no/id
description = line[1]
del line[1]
# Get check_no from description
line.insert(1, self.extract_check_no(description))
# Get memo and payee from description
tt = self.extract_description(description)
line.insert(2, tt[0])
line.insert(3, tt[1])
# line.insert(2, self.extract_description(description))
# Account id
if not self.statement.account_id:
self.statement.account_id = line[0]
# Currency
if not self.statement.currency:
self.statement.currency = line[7]
# Cleanup parts
line[6] = fix_amount_string(line[6])
line[2] = clean_multiple_whitespaces(line[2])
line[3] = clean_multiple_whitespaces(line[3])
# Create statement and fixup missing parts
stmtline = super(EasybankGiroCsvParser, self).parse_record(line)
stmtline.trntype = 'DEBIT' if stmtline.amount < 0 else 'CREDIT'
stmtline.id = generate_transaction_id(stmtline)
return stmtline
class EasybankPlugin(Plugin):
"""Easybank (CSV)"""
def determine_parser(self, fp):
"""Determine the parser to use based on the first booking line."""
description = fp.readline().split(";")[1]
fp.seek(0) # reset pointer
if '|' in description:
return EasybankCreditCardCsvParser(fp)
else:
return EasybankGiroCsvParser(fp)
def get_parser(self, filename):
"""Get a parser instance."""
encoding = self.settings.get('charset', 'cp1252')
f = open(filename, 'r', encoding=encoding)
parser = self.determine_parser(f)
parser.statement.bank_id = self.settings.get('bank', 'Easybank')
return parser
# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4 smartindent autoindent
|
