pyselect.py

# -*- coding: utf-8 -*-

__title__ = "pyselect"
__version__ = "0.2.0"
__author__ = "Matthew Behrens"
__license__ = "MIT"
__copyright__ = "Copyright 2013 Matthew Behrens"

# updated to work with Python 3

import sys


def select(options=None, lam="option"):
    """pass in a list of options, prompt the user to select one, and return the selected option or None"""
    if not options:
        return None
    width = len(str(len(options)))
    for x, option in enumerate(options):
        sys.stdout.write("{:{width}}) {}\n".format(x + 1, eval(lam), width=width))

    sys.stdout.write("{:>{width}} ".format("#?", width=width + 1))
    sys.stdout.flush()
    if sys.stdin.isatty():
        # regular prompt
        try:
            response = input().strip()
        except (EOFError, KeyboardInterrupt):
            # handle ctrl-d, ctrl-c
            response = ""
    else:
        # try connecting to current tty, when using pipes
        sys.stdin = open("/dev/tty")
        try:
            response = ""
            while True:
                response += sys.stdin.read(1)
                if response.endswith("\n"):
                    break
        except (EOFError, KeyboardInterrupt):
            sys.stdout.flush()
            pass
    try:
        response = int(response) - 1
    except ValueError:
        return None
    if response < 0 or response >= len(options):
        return None
    return options[response]


def main(args=None):
    if args is None:
        args = sys.argv[1:]
    response = select(args)
    if response:
        print(response)


if __name__ == "__main__":
    main()