python 超简单的命令行文本菜单，没有外部依赖

Posted 2021-05-10

import sys

if sys.version_info.major < 3:
    input = raw_input


class Menu:

    def __init__(self, items, title=None, prompt=None, exit_on_quit=True):
        self.items = items
        self.title = title
        self.prompt = prompt
        self.exit = exit_on_quit

    def process_response(self, response):
        try:
            item_num = int(response)
            if item_num < 1 or item_num > len(self.items):
                raise ValueError
            return item_num, self.items[item_num - 1]
        except ValueError:
            print('Invalid Response. Renenter...')

    def select(self):
        item_num = None
        while item_num is None:
            if self.title:
                print("\n{}".format(self.title))
            for item_num, item in enumerate(self.items):
                print('{:2}. {}'.format(item_num + 1, item))
            if self.prompt is None:
                self.prompt = 'Select option (1-{})? '.format(len(self.items))
            response = input(self.prompt)
            if self.exit and response.lower() == 'q':
                sys.exit(0)
            item_num, item = self.process_response(response)
        return item_num, item

if __name__ == '__main__':
    items = ['apple', 'banana', 'cantalope']
    item = Menu(items, quit=True).select()
    print(item)