Mercurial > hg > wordstream
view wordstream/main.py @ 21:c05704d14958 default tip
py3
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Wed, 02 Feb 2022 16:15:36 +0000 |
parents | 0c2b1fe8f53f |
children |
line wrap: on
line source
#!/usr/bin/env python import sys import urllib2 from optparse import OptionParser from pprint import pprint from wordstream.api import Corpus def main(args=sys.argv[1:]): parser = OptionParser('%prog [options] <path_or_url> <path_or_url> <...>') parser.add_option('--print-corpus', dest='print_corpus', default=False, action='store_true', help="print the given corpus of associations") parser.add_option('-n', type='int', dest='n', default=1, help='number of words to eat per munch') options, args = parser.parse_args() corpus = Corpus() corpus.feed_stuff(*args) if options.print_corpus: pprint(corpus) sys.exit(0) n = 1 while 1: stream = raw_input() splitstream = stream.split() if splitstream: for i in range(n): word = corpus.eat(splitstream[-1]) if not word: break print word if __name__ == '__main__': main()