view g_read_sents.py @ 1:fb617586f4b2 draft

planemo upload commit a81826fe44f09a3710a35c183aa88b745aeec064-dirty

import sys
import os
import nltk
from nltk.corpus import PlaintextCorpusReader
import argparse

def Parser():
    the_parser = argparse.ArgumentParser(description="Segments the text input into separate sentences")
    the_parser.add_argument('--input', required=True, action="store", type=str, help="input text file")
    the_parser.add_argument('--output', required=True, action="store", type=str, help="output file path")

    args = the_parser.parse_args()
    return args

def print_out(outp, text, sentences):
    with open(outp, 'w') as output:
        curr = 0
        for sent in sentences:
            times = count_occurences(sent, sent[-1])
            curr = text.find(sent[0], curr)
            end = find_nth(text, sent[-1], times, curr) + len(sent[-1])
            output.write(text[curr:end] + '\n')
            curr = end

def find_nth(string, sub, n, offset):
    start = string.find(sub, offset)
    while start >= 0 and n > 1:
        start = string.find(sub, start + len(sub))
        n -= 1
    return start

def count_occurences(lst, string):
    count = 0
    for item in lst:
        if string in item:
            count += 1
    return count

def read_sents(inp, outp):
    with open(inp, 'r') as fd:
        i = fd.read()
    corpus = PlaintextCorpusReader(os.path.dirname(inp), os.path.basename(inp))
    sents = corpus.sents()
    print_out(outp, i, sents)

if __name__ == '__main__':
    args = Parser()
    read_sents(args.input, args.output)