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
|
from ofxstatement.parser import CsvStatementParser
from ofxstatement.plugin import Plugin
from ofxstatement import statement
DETAILS_FIELD = 5
class DanskeCsvStatementParser(CsvStatementParser):
mappings = {"date": 0,
"memo": 4,
"amount": 8
}
date_format = "%Y:%m:%d"
def parse(self):
stmt = super(DanskeCsvStatementParser, self).parse()
statement.recalculate_balance(stmt)
return stmt
def parse_record(self, line):
if self.cur_record == 1:
return None
# fill statement line according to mappings
sl = super(DanskeCsvStatementParser, self).parse_record(line)
# generate transaction id out of available data
sl.id = statement.generate_transaction_id(sl)
if not sl.memo:
sl.memo = line[DETAILS_FIELD]
return sl
def use_details_for_memo(self):
self.mappings['memo'] = DETAILS_FIELD
class DanskePlugin(Plugin):
"""Lithuanian Danske bank CSV
"""
def get_parser(self, fin):
encoding = self.settings.get('charset', 'utf-8')
f = open(fin, 'r', encoding=encoding)
parser = DanskeCsvStatementParser(f)
parser.statement.currency = self.settings['currency']
parser.statement.account_id = self.settings['account']
parser.statement.bank_id = self.settings.get('bank', 'Danske')
if 'use-details-for-memo' in self.settings:
parser.use_details_for_memo()
return parser
|
