def parse(input_data: Union[RawIOBase, bytearray, bytes],
parse_body: bool = True,
parse_commands={},
**kwargs) -> Dict[str, Any]:
"""
Parses replay
:param (RawIOBase, bytearray, bytes) input_data: data source
:param bool parse_body: define what to parse
"""
reader = ReplayReader(input_data, **kwargs)
result = {
"header": ReplayHeader(reader).to_dict(),
"body_offset": reader.offset(),
}
if parse_body:
body_parser = ReplayBody(reader, True, parse_commands, True, **kwargs)
body_parser.parse()
result["body"] = body_parser.get_body()
result["messages"] = body_parser.get_messages()
result["desync_ticks"] = body_parser.get_desync_ticks()
result["last_tick"] = body_parser.tick
return result
def continuous_parse(
input_data: Union[RawIOBase, bytearray, bytes],
parse_header: bool = False,
**kwargs) -> Iterator[Union[Dict[str, Union[int, Dict]], Tuple]]:
"""
Used for continuous replay body parsing, when header is not needed.
Returns `ReplayBody` parser, that has `continuous_parse`.
Example:
::
>>> reader = ReplayReader(FileIO(input_data))
>>> ReplayHeader(reader) # jump over header
>>> for row in ReplayBody(reader).continuous_parse(): pass
"""
reader = ReplayReader(input_data, **kwargs)
if parse_header:
yield {
"header": ReplayHeader(reader).to_dict(),
"body_offset": reader.offset(),
}
for result in ReplayBody(reader, **kwargs).continuous_parse():
yield result
