Datalife/python-retrofix
1
0
Fork 0
mirror of https://gitlab.com/datalifeit/python-retrofix synced 2023-12-14 06:13:03 +01:00
Code Issues Projects Releases Wiki Activity

Do not try to import unexistent DETAIL_RECORD and fix aeat340 reader.

Browse source
This commit is contained in:
Albert Cervera i Areny 2018-01-20 12:15:32 +01:00
parent 6d8241737a
commit 3e5d5d5aa5
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
retrofix/aeat340.py
View file

@ -21,7 +21,7 @@
from .exception import RetrofixException
from .fields import Const, Number, Char, Integer, Numeric, Date
from .fields import SIGN_N_BLANK, SIGN_POSITIVE, DETAIL_RECORD
from .fields import SIGN_N_BLANK, SIGN_POSITIVE
from .record import Record
@ -198,8 +198,14 @@ def read(data):
current_line = lines.pop(0)
while lines:
if Record.valid(current_line, DETAIL_RECORD):
record = Record.extract(current_line, DETAIL_RECORD)
if Record.valid(current_line, ISSUED_RECORD):
record = Record.extract(current_line, ISSUED_RECORD)
elif Record.valid(current_line, RECEIVED_RECORD):
record = Record.extract(current_line, RECEIVED_RECORD)
elif Record.valid(current_line, INVESTMENT_RECORD):
record = Record.extract(current_line, INVESTMENT_RECORD)
elif Record.valid(current_line, INTRACOMMUNITY_RECORD):
record = Record.extract(current_line, INTRACOMMUNITY_RECORD)
else:
raise RetrofixException('Invalid record: %s' % current_line)
records.append(record)