def test_empty_rgroups(self, spec): if 'FIX.4.4' not in spec.version and 'FIX5.' not in spec.version: # only relevant for fix 4.4 or above return codec = Codec(spec=spec, decode_as='UTF-8') msg = b'35=AJ;17807=11;232=2;233=bli;234=blu;' \ b'233=blih;234=bluh;555=0;10=000;' msg = codec.parse(msg, separator=';') assert {35: 'AJ', 17807: '11', 232: [ {233: 'bli', 234: 'blu'}, {233: 'blih', 234: 'bluh'} ], 555: [], 10: '000' } == msg lhs = tuple(codec._unmap(msg)) assert lhs == ((35, 'AJ'), (232, 2), (233, 'bli'), (234, 'blu'), (233, 'blih'), (234, 'bluh'), (555, 0), (17807, '11'), (10, '000') ) serialised = '35=AJ;232=2;233=bli;234=blu;233=blih;234=bluh;' \ '555=0;17807=11;10=000;'.replace(';', chr(1)).encode('UTF-8') assert serialised == codec.serialise(msg)
def test_nested_rgroup(self, spec): codec = Codec(spec=spec, decode_as='UTF-8') msg = b'35=AE;555=1;687=AA;683=2;688=1;689=1;' \ b'688=2;689=2;17807=11;10=000;' msg = codec.parse(msg, separator=';') assert { 35: 'AE', 555: [ dict(((687, 'AA'), (683, [ dict(((688, '1'), (689, '1'))), dict(((688, '2'), (689, '2'))) ]))) ], 17807: '11', 10: '000' } == msg lhs = tuple(codec._unmap(msg)) assert lhs == ((35, 'AE'), (555, 1), (683, 2), (688, '1'), (689, '1'), (688, '2'), (689, '2'), (687, 'AA'), (17807, '11'), (10, '000')) serialised = '35=AE;555=1;683=2;688=1;689=1;' \ '688=2;689=2;687=AA;17807=11;10=000;'.replace(';', chr(1)).encode('UTF-8') assert serialised == codec.serialise(msg)
def test_consecutive_rgroups(self, spec): codec = Codec(spec=spec, decode_as='UTF-8') msg = b'35=B;215=1;216=1;' \ b'146=2;55=EURUSD;55=EURGBP;10=000;' msg = codec.parse(msg, separator=';') assert {35: 'B', 215: [{216 : '1'}], 146: [{55 : 'EURUSD'}, {55 : 'EURGBP'}], 10: '000' } == msg lhs = tuple(codec._unmap(msg)) assert lhs == ((35, 'B'), (215, 1), (216, '1'), (146, 2), (55, 'EURUSD'), (55, 'EURGBP'), (10, '000') ) serialised = '35=B;215=1;216=1;' \ '146=2;55=EURUSD;55=EURGBP;10=000;'.replace(';', chr(1)).encode('UTF-8') assert serialised == codec.serialise(msg)
def test_codec(self, spec): codec = Codec(spec=spec, decode_as='UTF-8') msg = (b'8=FIX.4.2;35=D;49=BLA;56=BLA;57=DEST;143=LN;11=eleven;18=1;21=2;54=2;40=2;59=0;55=PROD;' b'38=10;44=1;52=20110215-02:20:52.675;10=000;') res = codec.parse(msg, separator=';') assert {8: u'FIX.4.2', 11: u'eleven', 18: u'1', 21: u'2', 35: u'D', 38: u'10', 40: u'2', 44: u'1', 49: u'BLA', 52: u'20110215-02:20:52.675', 54: u'2', 55: u'PROD', 56: u'BLA', 57: u'DEST', 59: u'0', 10: u'000', 143: u'LN'} == res codec = Codec(spec=spec) msg = (b'8=FIX.4.2;35=D;49=BLA;56=BLA;57=DEST;143=LN;11=eleven;18=1;21=2;54=2;40=2;59=0;55=PROD;' b'38=10;44=1;52=20110215-02:20:52.675;10=000;') res = codec.parse(msg, separator=';') assert {8: 'FIX.4.2', 11: 'eleven', 18: '1', 21: '2', 35: 'D', 38: '10', 40: '2', 44: '1', 49: 'BLA', 52: '20110215-02:20:52.675', 54: '2', 55: 'PROD', 56: 'BLA', 57: 'DEST', 59: '0', 10: '000', 143: 'LN'} == res codec = Codec(spec=spec, decode_all_as_347=True) res = codec.parse(msg, separator=';') assert {8: 'FIX.4.2', 11: 'eleven', 18: '1', 21: '2', 35: 'D', 38: '10', 40: '2', 44: '1', 49: 'BLA', 52: '20110215-02:20:52.675', 54: '2', 55: 'PROD', 56: 'BLA', 57: 'DEST', 59: '0', 10: '000', 143: 'LN'} == res msg = (b'8=FIX.4.2;35=D;49=BLA;56=BLA;57=DEST;347=UTF-8;143=LN;11=eleven;18=1;21=2;54=2;40=2;59=0;55=PROD;' b'38=10;44=1;52=20110215-02:20:52.675;10=000;') codec = Codec(spec=spec, decode_all_as_347=True) res = codec.parse(msg, separator=';') assert {8: u'FIX.4.2', 11: u'eleven', 18: u'1', 21: u'2', 35: u'D', 38: u'10', 40: u'2', 44: u'1', 49: u'BLA', 52: u'20110215-02:20:52.675', 54: u'2', 55: u'PROD', 56: u'BLA', 57: u'DEST', 59: u'0', 10: u'000', 143: u'LN', 347: u'UTF-8'} == res msg = (b'8=FIX.4.2;35=8;49=BLA;56=BLA;57=DEST;143=LN;11=eleven;18=1;21=2;54=2;40=2;59=0;55=PROD;' b'38=10;44=1;52=20110215-02:20:52.675;' b'382=2;' b'375=A;337=B;' b'375=B;437=B;' b'10=000;') codec = Codec(spec=spec) res = codec.parse(msg, separator=';') assert {8: 'FIX.4.2', 11: 'eleven', 382: [dict(((375, 'A'), (337, 'B'))), dict(((375, 'B'), (437, 'B')))], 18: '1', 21: '2', 35: '8', 38: '10', 40: '2', 44: '1', 49: 'BLA', 52: '20110215-02:20:52.675', 54: '2', 55: 'PROD', 56: 'BLA', 57: 'DEST', 59: '0', 143: 'LN', 10: '000'} == res # make sure that with a group finishing the message it still works msg = (b'8=FIX.4.2;35=8;49=BLA;56=BLA;57=DEST;143=LN;11=eleven;18=1;21=2;54=2;40=2;59=0;55=PROD;' b'38=10;44=1;52=20110215-02:20:52.675;' b'382=2;' b'375=A;337=B;' b'375=B;437=B;') res = codec.parse(msg, separator=';') assert {8: 'FIX.4.2', 11: 'eleven', 382: [dict(((375, 'A'), (337, 'B'))), dict(((375, 'B'), (437, 'B')))], 18: '1', 21: '2', 35: '8', 38: '10', 40: '2', 44: '1', 49: 'BLA', 52: '20110215-02:20:52.675', 54: '2', 55: 'PROD', 56: 'BLA', 57: 'DEST', 59: '0', 143: 'LN', } == res
class FixMessage(FixFragment): # pylint: disable=R0904 # too many public methods. Needed for compatibility and functionality """ Simple dictionary-like object, for use with FIX raw messages. Note that the tags are converted (when possible) to integers, and that the values are kept as strings. The default separator is ``;``, but can be specified. Check the definition of :py:meth:`~pyfixmsg.FixMessage.load_fix` for details. Example: >>> fix = FixMessage() >>> fix.load_fix(line) >>> #print(fix) {6: '0', 8: 'FIX.4.2', 10: '100', 10481: 'A', 14: '0', 15: 'EUR', 10641: 'blabla', 18: '1', 21: '2', 22: '5', 151: '1', 11951: 'HOOF7M0f4BGJ0rkaNTkkeAA', .... FixMessage also have a ``time`` attribute, a ``direction`` attribute (inbound : 0, outbound : 1) and a ``recipient`` which is rather where it's been received from or sent to. FixMessages sort by default on time, and will be considered equal if the dictionary values are the same AND the time is the same. This FixMessage is eager : it will parse the whole fix and store it locally. It is significantly faster in most usage patterns that we observed. useful shortcut methods : :: fix.tag_exact(tag, value) fix.tag_iexact(tag, value) fix.tag_contains(tag, value) fix.tag_icontains(tag, value) fix.tag_match_regex(tag, value) Note : the tag_* methods don't support repeating groups """ # Class type of FIX message fragments FragmentType = FixFragment @classmethod def from_dict(cls, tags_dict): """ Create a FixMessage from a dictionary. :param tags_dict: dictionary of FIX tags to values :type tags_dict: ``dict`` of ``int`` to ``str`` or ``int`` or ``float`` or ``long`` :return: a FixMessage object :rtype: ``FixMessage`` """ msg = cls() msg.update(tags_dict) return msg @classmethod def from_buffer(cls, msg_buffer, fix_codec): """ Create a FixMessage from a buffer and a codec :param msg_buffer: a buffer as a string :type msg_buffer: ``str`` :param fix_codec: an object with static encode() and decode() calls :type fix_codec: ``Codec`` :return: a FixMessage object :rtype: ``FixMessage`` """ msg = cls() msg.codec = fix_codec msg.from_wire(msg_buffer, fix_codec) return msg def __lt__(self, other): return self.time < other.time def __gt__(self, other): return self.time > other.time def __eq__(self, other): if other is None: return None for msg in (self, other): if 9 in msg: del msg[9] if 10 in msg: del msg[10] if super(FixMessage, self).__eq__(other): if (hasattr(other, 'time') and (other.time == self.time) and (other.recipient == self.recipient)): return True return False def __ne__(self, other): if super(FixMessage, self).__ne__(other) or other.time != self.time: return True return False def __le__(self, other): return self.time <= other.time def __ge__(self, other): return self.time >= other.time def __copy__(self): """ copy module support. This copies the message by serialising it and parsing the serialised data back into a new message. This is a lot faster than deepcopy or other techniques. """ new_msg = self.__class__() new_msg.codec = self.codec new_msg.from_wire(self.to_wire()) new_msg.time = self.time new_msg.process = self.process new_msg.recipient = self.recipient new_msg.direction = self.direction new_msg.set_len_and_chksum() return new_msg def copy(self): """ Copy interface without using the copy module""" return self.__copy__() def __init__(self, *args, **kwargs): """ The constructor uses the ``dict()`` signature unmodified. You can set the following manually or through a factory function: * ``self.process`` an opaque value (to store the process that received or processed the message, defaults to empty string). * ``self.separator`` the default separator to use when parsing messages. Defaults to ``';'`` * ``self.time`` the time the message has been created or received. Defaults to ``datetime.utcnow()`` * ``self.recipient`` opaque value (to store for whom the message was intended) * ``self.direction`` Whether the message was received (``0``), sent (``1``) or unknown (``None``) * ``self.typed_values`` Whether the values in the message are typed. Defaults to ``False`` * ``self.codec`` Default :py:class:`~pyfixmsg.codec.stringfix.Codec` to use to parse message. Defaults to a naive codec that doesn't support repeating groups """ self.process = '' self.separator = ';' self.time = datetime.datetime.utcnow() self.recipient = '' self.direction = None self.typed_values = False self.codec = Codec() # Allows maintaining tag order if constructing msg from a FixFragment if args and isinstance(args[0], FixFragment): self.tag_order = getattr(args[0], 'tag_order', None) else: self.tag_order = None super(FixMessage, self).__init__(*args, **kwargs) @property def tags(self): """ Note: this property is there to replace a self reference that existed before. Deprecated. """ warnings.warn("FixMessage.tags is deprecated") return self def set_or_delete(self, tag, value): """ Sets the tag if value is neither None or the empty string. Deletes the tag otherwise. Only works on top-level tags (not inside repeating groups) """ if value is not None and value != "": # don't remove if the tag is 0 self[tag] = value else: if tag in self: del self[tag] def apply(self, update): """ equivalent to :py:meth:`~pyfixmsg.FixMessage.update()` but if any value in the update dictionary is None and the tag is present in the current message, that tag is removed. Note: naive about repeating groups :param update: map of values to update the state with. :type update: ``dict`` """ for tag in update: self.set_or_delete(tag, update[tag]) def load_fix(self, string, process=None, separator=';'): """ Parses a FIX message from a string using default codecs and separators. :param string: the string containing the FIX message to be parsed :type string: ``bytes`` :param process: Optional originator of the FIX message :type process: ``unicode`` :param separator: Character delimiting "tag=val" pairs. Optional. By default this is a ';' character. Specify ``pyfixmsg.SEPARATOR`` when parsing standard FIX. :type separator: ``unicode`` :return: A parsed fix message :rtype: ``FixMessage`` """ fix_msg = self.codec.parse(string.strip(), separator=separator) self.update(fix_msg) self.tag_order = getattr(fix_msg, 'tag_order', None) self.process = process return self @property def fix(self): """ Legacy compatibility, will be removed shortly""" return self.output_fix(self.separator) def output_fix(self, separator=';', calc_checksum=True, remove_length=False): """ ouputs itself as a vanilla FIX message. This forces the output to String fix but tries to reuse the spec from the current codec""" if calc_checksum: self.set_len_and_chksum() if remove_length: del self[9] try: codec = Codec(spec=self.codec.spec) except AttributeError: codec = Codec() return codec.serialise(self, separator, delimiter='=') def to_wire(self, codec=None): """ Return wire representation according to a codec """ codec = codec or self.codec self.set_len_and_chksum() return codec.serialise(self) def from_wire(self, msg, codec=None): """ Extract from a wire representation according to a codec """ codec = codec or self.codec self.update(codec.parse(msg)) self.typed_values = not getattr(codec, 'decoded_values_are_untyped', False) def __str__(self): """ Human-readable representation """ out = self.output_fix() try: out = unicode(out).encode('UTF-8') except (UnicodeDecodeError, NameError): pass return str(out) def calculate_checksum(self): """ calculates the standard fix checksum""" return self.checksum() def checksum(self, value=None): """ FIX checksum """ if value is None: value = pyfixmsg.len_and_chsum(self)[1] % 256 return '{0:03d}'.format(value % 256) def set_len_and_chksum(self): """ Assign length and checksum based on current contents """ length, raw_checksum = pyfixmsg.len_and_chsum(self) self[9] = length self[10] = self.checksum(raw_checksum) def tag_exact(self, tag, value, case_insensitive=False): """ Returns True if self[tag] has the exact value. Returns False if the tag doesnt exist or is not exact """ value = str(value) try: mine = str(self[tag]) except KeyError: return False if case_insensitive: value = value.lower() mine = mine.lower() return mine == value def tag_iexact(self, tag, value): """ Returns True if self[tag] has the exact value (case insensitive). Returns False if the tag doesnt exist or is not exact """ return self.tag_exact(tag, value, case_insensitive=True) def tag_contains(self, tag, value, case_insensitive=False): """ Returns True if self[tag] contains value. Returns False otherwise, or if the tag doesnt exist This is a string string comparison""" value = str(value) try: mine = str(self[tag]) except KeyError: return False if case_insensitive: value = value.lower() mine = mine.lower() return value in mine def tag_icontains(self, tag, value): """ case-insensitive version of tag_contains""" return self.tag_contains(tag, value, case_insensitive=True) def tag_exact_dict(self, dictionary): """ check that all the keys and values of the passed dict are present and identical in the fixmsg""" return all( self.tag_exact(tag, value) for tag, value in list(dictionary.items())) def tag_match_regex(self, tag, regex): """ returns True of self[tag] matches regex, false otherwise or if the tag doesnt exist """ regex = str(regex) try: if re.match(regex, str(self[tag])): return True except KeyError: pass return False def tag_lt(self, tag, value): """ Test tag is smaller than value. Uses decimal comparison if possible. Returns False if tag absent""" if not self.get(tag): return False if not value: return False tag = self.get(tag) try: tag = decimal.Decimal(tag) value = decimal.Decimal(value) except (ValueError, decimal.InvalidOperation): pass return tag < value def tag_le(self, tag, value): """ Test tag is smaller or equal value. Uses decimal comparison if possible. Returns False if tag absent""" if not self.get(tag): return False if not value: return False tag = self.get(tag) try: tag = decimal.Decimal(tag) value = decimal.Decimal(value) except (ValueError, decimal.InvalidOperation): pass return tag <= value def tag_gt(self, tag, value): """ Test tag is greater than value. Uses decimal comparison if possible. Returns False if tag absent""" if not self.get(tag): return False if not value: return False tag = self.get(tag) try: tag = decimal.Decimal(tag) value = decimal.Decimal(value) except (ValueError, decimal.InvalidOperation): pass return tag > value def tag_ge(self, tag, value): """ Test tag is greater or equal to value. Uses decimal comparison if possible. Returns False if tag absent""" if not self.get(tag): return False if not value: return False tag = self.get(tag) try: tag = decimal.Decimal(tag) value = decimal.Decimal(value) except (ValueError, decimal.InvalidOperation): pass return tag >= value def tag_in(self, tag, values): """ returns True of self[tag] is in values, false otherwise or if the tag doesnt exist """ values = [str(i) for i in values] return str(self.get(tag, None)) in values def update_all(self, tag, value): """ this will force a tag (that already exists!) to a value at all appearances """ for path in self.find_all(tag): point = self last_key = None last_point = point for key in path: last_point = point point = point[key] last_key = key if last_key is not None: last_point[last_key] = value