diff --git a/trustgraph-cli/scripts/tg-dump-msgpack b/trustgraph-cli/scripts/tg-dump-msgpack index 16fe2325..9f91394f 100755 --- a/trustgraph-cli/scripts/tg-dump-msgpack +++ b/trustgraph-cli/scripts/tg-dump-msgpack @@ -2,8 +2,33 @@ import msgpack import sys +import argparse -unpacker = msgpack.Unpacker(sys.stdin.buffer, raw=False) -for unpacked in unpacker: - print(unpacked) +def run(input_file): + + with open(input_file, 'rb') as f: + + unpacker = msgpack.Unpacker(f, raw=False) + + for unpacked in unpacker: + print(unpacked) + +def main(): + + parser = argparse.ArgumentParser( + prog='tg-load-pdf', + description=__doc__, + ) + + parser.add_argument( + '-i', '--input-file', + required=True, + help=f'Input file' + ) + + args = parser.parse_args() + + run(**vars(args)) + +main() diff --git a/trustgraph-cli/setup.py b/trustgraph-cli/setup.py index af2e9418..f4c4e92d 100644 --- a/trustgraph-cli/setup.py +++ b/trustgraph-cli/setup.py @@ -39,6 +39,7 @@ setuptools.setup( "pulsar-client", "rdflib", "tabulate", + "msgpack", ], scripts=[ "scripts/tg-graph-show",