broccoli_words_file_to_context_file.py

import argparse
import sys
from broccoli_ontology_txt_to_nt import handleSubject

__author__ = 'buchholb'

parser = argparse.ArgumentParser()

parser.add_argument('--wordsfile',
                    type=str,
                    help='Broccoli Wordsfile.',
                    required=True)

def writeContextFileToStdout(wordsfile):
    for line in open(wordsfile):
        cols = line.strip('\n').split('\t')
        if cols[0].startswith(':e:'):
            cols[0] = handleSubject(cols[0][3:])
            entityFlag = '1'
        else:
            entityFlag = '0'
        print('\t'.join([cols[0], entityFlag, cols[1], cols[2]]))


def main():
    args = vars(parser.parse_args())
    wordsfile = args['wordsfile']
    writeContextFileToStdout(wordsfile)


if __name__ == '__main__':
    main()