| |
| import argparse |
| import sys |
|
|
| IS_MEMBER, IS_PARENTHESIS_ROOT = 5, 3 |
|
|
| if __name__ == "__main__": |
| parser = argparse.ArgumentParser() |
| parser.add_argument("paths", default=[], nargs="*", help="Input paths") |
| parser.add_argument("--is_member", default=False, action="store_true", help="Add is_member") |
| parser.add_argument("--is_parenthesis_root", default=False, action="store_true", help="Add is_parenthesis_root") |
| args = parser.parse_args() |
|
|
| for path in args.paths: |
| with open(path, "r", encoding="utf-8") as conllu_file: |
| block = [] |
| for line in conllu_file: |
| line = line.rstrip("\n") |
| if not line: |
| assert block |
| |
| heads, deps = {}, {} |
| for entry in block: |
| columns = entry.split("\t") |
| if len(columns) == 10: |
| assert int(columns[0]) == len(heads) + 1 |
| deps[len(heads) + 1] = columns[7] |
| heads[len(heads) + 1] = columns[6] |
|
|
| for i in range(len(block)): |
| columns = block[i].split("\t") |
| if len(columns) == 10: |
| if args.is_member and columns[IS_MEMBER] == "1": |
| parent = int(columns[0]) |
| while parent and deps[parent] not in ("Apos", "Coord"): |
| parent = int(heads[parent]) |
|
|
| if deps[parent] == "Apos": |
| columns[7] += "_Ap" |
| elif deps[parent] == "Coord": |
| columns[7] += "_Co" |
| else: |
| print("Did not find correct parent of IsMember {} in the below sentence".format(block[i]), *block, file=sys.stderr, sep="\n") |
| columns[IS_MEMBER] = "_" |
|
|
| if args.is_parenthesis_root and columns[IS_PARENTHESIS_ROOT] == "1": |
| columns[7] += "_Pa" |
| columns[IS_PARENTHESIS_ROOT] = "_" |
|
|
| block[i] = "\t".join(columns) |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
|
|
| print(*block, sep="\n", end="\n\n") |
| block = [] |
| else: |
| block.append(line) |
| assert not block |
|
|
