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
|
"""Parser for the statement of Poštovní spořitelna"""
from ofxstatement.parser import StatementParser
from ofxstatement.plugin import Plugin
from ofxstatement.statement import Statement, StatementLine
import re
import datetime
import logging
log = logging.getLogger('PS')
RecordParseRE = re.compile(r"\n{2,}", re.MULTILINE)
class PSTextFormatParser(StatementParser):
# these keys mean in order: amount, date of transaction being recorded in
# books, transaction type for statistical purposes, currency, name of
# the other account, type of the transaction, memo, other account no.,
# possible number of other sub-account, variable symbol (for identification
# of the transaction), balance of the account after transaction
#
# note, that variable symbol is not guaranteed to be unique (it is quite
# often number of the invoice or something which allows accountants to
# identify the transaction on the account statement).
_keys = set(('částka', 'datum zaúčtování', 'konstantní symbol', 'měna',
'název protiúčtu', 'označení operace', 'poznámka',
'protiúčet', 'specifický symbol', 'variabilní symbol',
'zůstatek',))
statement = None
currentLine = 0
def __init__(self, fname):
StatementParser.__init__(self)
# bank_id=None, account_id=None, currency=None
self.statement = Statement(currency="CZK")
self.statement.start_date = datetime.date.fromtimestamp(0)
self.statement.end_date = datetime.date.today()
self.fin = fname
def split_records(self):
# uses self.fin
return RecordParseRE.split(self.fin.read().strip())[1:]
def parse_record(self, rec):
MultilineRecRE = re.compile(r"\n\s+")
SplitItemRE = re.compile(r":\s+")
# join together multiline items
rec = MultilineRecRE.sub(" ", rec)
log.debug('rec = {}'.format(rec))
res_iter = [SplitItemRE.split(line.strip(), maxsplit=1)
for line in rec.split("\n")]
res_dict = dict((item for item in res_iter if len(item) > 1))
res_key_set = set(res_dict.keys())
log.debug('res_dict = {}'.format(res_dict))
log.debug('res_key_set = {}'.format(res_key_set))
# Insurance against unknown keys in the statement
assert res_key_set <= self._keys, \
"Unknown keys in the transaction: %s" % (res_key_set - self._keys)
# no idea, how to make this unique and monotonically increasing
try:
trans_ID = res_dict['datum zaúčtování'] + res_dict['částka']
if 'název protiúčtu' in res_dict:
trans_ID += res_dict['název protiúčtu']
if 'variabilní symbol' in res_dict:
trans_ID += res_dict['variabilní symbol']
date_str = res_dict['datum zaúčtování'].split('.')
except KeyError:
log.error('res_dict:\n{}'.format(res_dict))
raise
memo = ''
if 'poznámka' in res_dict:
memo = "%s\n" % res_dict['poznámka']
if 'označení operace' in res_dict:
memo += "%s\n" % res_dict['označení operace']
stat_line = StatementLine(
id=trans_ID,
date=datetime.date(
int(date_str[2]), int(date_str[1]), int(date_str[0])),
memo=memo,
amount=float(res_dict['částka']))
if 'název protiúčtu' in res_dict:
stat_line.payee = res_dict['název protiúčtu']
# According to OFX spec this could be
# "check (or other reference) no."
# I don't see any requirement on monotonicity or uniqueness, but
# I wonder how GnuCash understands that
if 'variabilní symbol' in res_dict:
stat_line.check_no = res_dict['variabilní symbol']
return stat_line
class PSPlugin(Plugin):
def get_parser(self, fin):
encoding = self.settings.get('charset', 'utf-8-sig')
inf = open(fin, encoding=encoding)
return PSTextFormatParser(inf)
|
