From 35b2b28b4aa21494519d30322bf8e8ee0720181d Mon Sep 17 00:00:00 2001
From: Jonathon Reinhart <Jonathon.Reinhart@gmail.com>
Date: Sun, 29 Jan 2017 17:52:11 -0500
Subject: [PATCH] Add hl7dump.py

This script is used to dump out HL7 messages in a verbose format.
---
 utils/hl7dump.py | 86 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100755 utils/hl7dump.py

diff --git a/utils/hl7dump.py b/utils/hl7dump.py
new file mode 100755
index 0000000..2420744
--- /dev/null
+++ b/utils/hl7dump.py
@@ -0,0 +1,86 @@
+#!/usr/bin/env python
+from __future__ import print_function
+from hl7apy.core import Message, Segment, Field, Component, SubComponent
+
+INDENT=4
+
+def __dump(thing, with_value=False):
+    p = '{:<60}'.format(thing)
+    if with_value:
+        v = thing.value
+
+        # For objects of base_datatypes, e.g. SI, ST, IS, ID, TN,
+        # we need to get their .value
+        try:
+            p += '     Value: ({}) "{}"'.format(type(v).__name__, v.value)
+        except AttributeError:
+            p += '     Value: ' + str(v)
+
+    print(p)
+    return True
+
+def _dump_default(thing):
+    return __dump(thing, False)
+
+def _dump_default_with_value(thing):
+    return __dump(thing, True)
+
+
+def _dump_field_or_component(thing):
+    __dump(thing, True)
+
+    # Don't bother showing Components or SubComponents for these simple types
+    if (not args.verbose) and thing.datatype in ('NM', 'ST', 'TS', 'ID', 'IS'):
+        return False
+
+    return True
+
+
+def dump_thing(thing, indent=0):
+
+    dumper = {
+        Message:        _dump_default,
+        Segment:        _dump_default,
+        Field:          _dump_field_or_component,
+        Component:      _dump_field_or_component,
+        #SubComponent:   _dump_subcomponent,
+    }.get(type(thing),  _dump_default_with_value)
+
+    print(' '*indent, end='')
+    if not dumper(thing):
+        # Don't dump children
+        return
+
+    try:
+        children = thing.children
+    except AttributeError:
+        return
+
+    for c in children:
+        dump_thing(c, indent + INDENT)
+
+
+if __name__ == '__main__':
+    from hl7apy.parser import parse_message, parse_segment
+    from hl7apy.consts import VALIDATION_LEVEL
+    import hl7apy
+    import argparse
+
+    ap = argparse.ArgumentParser()
+    ap.add_argument('file', type=argparse.FileType('r'))
+    ap.add_argument('-s', '--strict', action='store_true')
+    ap.add_argument('-v', '--verbose', action='store_true')
+    global args
+    args = ap.parse_args()
+
+    if args.strict:
+        hl7apy.set_default_validation_level(VALIDATION_LEVEL.STRICT)
+
+
+    raw = args.file.read()
+    raw = raw.replace('\r\n', '\r')
+    raw = raw.replace('\n', '\r')
+
+
+    msg = parse_message(raw)
+    dump_thing(msg)