view g_pos.py @ 3:0df72a8ab095 draft default tip

planemo upload for repository https://github.com/Alveo/alveo-galaxy-tools commit f2432aaedd36ae7662873623d8861d0982dffdd2

from __future__ import print_function, unicode_literals
import nltk
import argparse
import io

nltk.download('averaged_perceptron_tagger', quiet=True)
nltk.download('punkt', quiet=True)


def arguments():
    parser = argparse.ArgumentParser(description="tokenize a text")
    parser.add_argument('--input', required=True, action="store", type=str, help="input text file")
    parser.add_argument('--output', required=True, action="store", type=str, help="output file path")
    return parser.parse_args()


def postag(in_file, out_file):
    """Input: a text file with one token per line
    Output: a version of the text with Part of Speech tags written as word/TAG
    """
    with open(in_file, 'rb') as fd:
        text = fd.read()
        text = text.decode('utf-8')

    sentences = nltk.sent_tokenize(text)

    with io.open(out_file, 'w') as output:
        for sentence in sentences:
            tokens = nltk.word_tokenize(sentence)
            postags = nltk.pos_tag(tokens)
            for postag in postags:
                # print postag
                p = "%s/%s " % postag
                output.write(p)
        output.write('\n')


if __name__ == '__main__':
    args = arguments()
    postag(args.input, args.output)