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
|
from enum import Enum
from .formals import SegmentSequence
from .segments.base import FinTS3Segment
from .segments.dialog import HIRMS2
class MessageDirection(Enum):
FROM_CUSTOMER = 1
FROM_INSTITUTE = 2
class FinTSMessage(SegmentSequence):
DIRECTION = None
# Auto-Numbering, dialog relation, security base
def __init__(self, dialog=None, *args, **kwargs):
self.dialog = dialog
self.next_segment_number = 1
super().__init__(*args, **kwargs)
def __iadd__(self, segment: FinTS3Segment):
if not isinstance(segment, FinTS3Segment):
raise TypeError("Can only append FinTS3Segment instances, not {!r}".format(segment))
segment.header.number = self.next_segment_number
self.next_segment_number += 1
self.segments.append(segment)
return self
def response_segments(self, ref, *args, **kwargs):
for segment in self.find_segments(*args, **kwargs):
if segment.header.reference == ref.header.number:
yield segment
def responses(self, ref, code=None):
for segment in self.response_segments(ref, HIRMS2):
for response in segment.responses:
if code is None or response.code == code:
yield response
class FinTSCustomerMessage(FinTSMessage):
DIRECTION = MessageDirection.FROM_CUSTOMER
# Identification, authentication
class FinTSInstituteMessage(FinTSMessage):
DIRECTION = MessageDirection.FROM_INSTITUTE
|
