mrs.py

#!/usr/bin/env python3

import sys
import argparse
from delphin.mrs import simplemrs, mrx, dmrx, eds, simpledmrs

mrsformats = {
    'simplemrs': simplemrs,
    'mrx': mrx,
    'dmrx': dmrx,
    'eds': eds,
    'simpledmrs': simpledmrs
}

parser = argparse.ArgumentParser(description="Utility for manipulating MRSs")
subparsers = parser.add_subparsers(dest='command')

convert_parser = subparsers.add_parser('convert', aliases=['c'])
convert_parser.add_argument('--from', '-f', dest='srcfmt',
                            choices=list(mrsformats.keys()))
convert_parser.add_argument('--to', '-t', dest='tgtfmt',
                            choices=list(mrsformats.keys()))
convert_parser.add_argument('--pretty-print', '-p', action='store_true')
convert_parser.add_argument('--color', '-c', action='store_true')
convert_parser.add_argument('infile', metavar='PATH', nargs='?')

path_parser = subparsers.add_parser('paths', aliases=['p'])
path_parser.add_argument('--format', '-f', choices=list(mrsformats.keys()))
path_parser.add_argument('--depth', '-d', default=-1)
path_parser.add_argument('infile', metavar='PATH', nargs='?')

args = parser.parse_args()
if args.command in ('convert', 'c'):
    srcfmt = mrsformats[args.srcfmt]
    if args.infile is not None:
        ms = srcfmt.load(open(args.infile, 'r'))
    else:
        ms = srcfmt.loads(sys.stdin.read())
    output = mrsformats[args.tgtfmt].dumps(ms, pretty_print=args.pretty_print,
                                           color=args.color)
    print(output)

elif args.command in ('paths', 'p'):
    from delphin.mrs import path as mrspath
    if args.infile is not None:
        instream = open(args.infile, 'r')
    else:
        instream = sys.stdin
    outstream = sys.stdout
    ms = mrsformats[args.format].load(instream)
    for m in ms:
        paths = list(mrspath.get_paths(m, max_depth=int(args.depth)))
        print('\t'.join(paths))