diff options
| -rwxr-xr-x | wikiquery | 10 |
1 files changed, 8 insertions, 2 deletions
@@ -1,12 +1,13 @@ #! /usr/bin/env python +import shlex import optparse import sys import wikipedia from functools import partial -version = "0.0.2" +version = "0.1.0" def stringify(o): if isinstance(o, list): @@ -22,13 +23,18 @@ def display(o): print stringify(o) def parse_args(): + args = sys.argv[1:] + + if not args: + args = sys.stdin.read().splitlines() + parser = optparse.OptionParser(usage="!wiki <topic> [--search|--url]") parser.add_option("--search", action="store_true") parser.add_option("--url", action="store_true") parser.add_option("-v", "--version", action="store_true") - return parser.parse_args() + return parser.parse_args(args) def main(): options, args = parse_args() |