python中如何自定义查询菜单

在Python中自定义查询菜单有许多方法，使用标准库中的curses模块、利用第三方库如prompt_toolkit、创建基于tkinter的图形用户界面（GUI），这些都是有效的方法。下面，我们将详细介绍其中一种方法，使用curses模块来创建一个自定义查询菜单。

一、CURSES模块

1、安装和导入CURSES模块

curses模块是Python标准库的一部分，因此无需额外安装。你可以直接导入它：

import curses

2、初始化CURSES窗口

首先，我们需要初始化curses窗口。以下是基本的初始化和清理代码：

import curses
def main(stdscr):
    # Clear screen
    stdscr.clear()
    # Your code here
curses.wrapper(main)

curses.wrapper函数可以帮助我们处理初始化和清理工作。

3、创建菜单选项

我们需要定义菜单选项，并创建一个函数来显示这些选项：

def display_menu(stdscr, menu_items, current_row):
    stdscr.clear()
    h, w = stdscr.getmaxyx()
    for idx, row in enumerate(menu_items):
        x = w // 2 - len(row) // 2
        y = h // 2 - len(menu_items) // 2 + idx
        if idx == current_row:
            stdscr.attron(curses.color_pair(1))
            stdscr.addstr(y, x, row)
            stdscr.attroff(curses.color_pair(1))
        else:
            stdscr.addstr(y, x, row)
    stdscr.refresh()

在这个函数中，我们遍历menu_items并将它们显示在窗口的中央。当前选中的行会有不同的颜色。

4、添加颜色对

我们需要初始化颜色对，以便在显示菜单时使用：

def main(stdscr):
    # Initialize curses
    curses.curs_set(0)
    curses.start_color()
    curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
    # Define menu items
    menu_items = ['Option 1', 'Option 2', 'Option 3', 'Exit']
    current_row = 0
    display_menu(stdscr, menu_items, current_row)

5、处理用户输入

我们需要处理用户输入，以便用户可以在菜单中进行导航：

def main(stdscr):
    # Initialize curses
    curses.curs_set(0)
    curses.start_color()
    curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
    # Define menu items
    menu_items = ['Option 1', 'Option 2', 'Option 3', 'Exit']
    current_row = 0
    display_menu(stdscr, menu_items, current_row)
    while True:
        key = stdscr.getch()
        if key == curses.KEY_UP and current_row > 0:
            current_row -= 1
        elif key == curses.KEY_DOWN and current_row < len(menu_items) - 1:
            current_row += 1
        elif key == curses.KEY_ENTER or key in [10, 13]:
            if current_row == len(menu_items) - 1:
                break
        display_menu(stdscr, menu_items, current_row)

在这个代码中，我们使用箭头键来改变current_row的值，并使用Enter键来选择菜单项。

二、PROMPT_TOOLKIT模块

prompt_toolkit是一个用于构建交互式命令行应用程序的第三方库。

1、安装PROMPT_TOOLKIT

你可以使用pip来安装：

pip install prompt_toolkit

2、创建菜单

我们可以使用prompt_toolkit.shortcuts中的函数来创建一个简单的菜单：

from prompt_toolkit import prompt
from prompt_toolkit.shortcuts import CompleteStyle, PromptSession
menu_items = ['Option 1', 'Option 2', 'Option 3', 'Exit']
def get_menu_choice():
    session = PromptSession()
    while True:
        text = session.prompt('Choose an option: ', completer=WordCompleter(menu_items, ignore_case=True), complete_style=CompleteStyle.READLINE_LIKE)
        if text in menu_items:
            return text
choice = get_menu_choice()
print(f'You selected: {choice}')

三、TKINTER模块

tkinter是Python的标准GUI库。

1、安装和导入TKINTER

tkinter也是Python标准库的一部分，因此无需额外安装：

import tkinter as tk
from tkinter import simpledialog

2、创建简单的GUI菜单

你可以使用tkinter创建一个简单的菜单：

class MenuApp:
    def __init__(self, root):
        self.root = root
        self.root.title("Menu")
        self.menu_options = ['Option 1', 'Option 2', 'Option 3', 'Exit']
        self.create_widgets()
    def create_widgets(self):
        self.label = tk.Label(self.root, text="Choose an option:")
        self.label.pack()
        self.menu_var = tk.StringVar(self.root)
        self.menu_var.set(self.menu_options[0])
        self.menu = tk.OptionMenu(self.root, self.menu_var, *self.menu_options)
        self.menu.pack()
        self.button = tk.Button(self.root, text="Select", command=self.select_option)
        self.button.pack()
    def select_option(self):
        choice = self.menu_var.get()
        print(f'You selected: {choice}')
        if choice == 'Exit':
            self.root.quit()
root = tk.Tk()
app = MenuApp(root)
root.mainloop()