315 lines
12 KiB
Python
315 lines
12 KiB
Python
# coding: utf8
|
||
|
||
#######################################################
|
||
# _____ _ _____ _ _ _ #
|
||
# / ____| | |_ _| | | | | | #
|
||
# | | | | | | | |__| | ___| |_ __ ___ _ __ #
|
||
# | | | | | | | __ |/ _ \ | '_ \ / _ \ '__| #
|
||
# | |____| |____ _| |_| | | | __/ | |_) | __/ | #
|
||
# \_____|______|_____|_| |_|\___|_| .__/ \___|_| #
|
||
# | | #
|
||
# |_| #
|
||
#######################################################
|
||
|
||
from __future__ import annotations
|
||
from typing import Callable, TypedDict
|
||
|
||
version = (2, 0, 1, 20231219)
|
||
|
||
|
||
def request_input(prompt_msg: str,
|
||
err_msg: str = "Invalid value.",
|
||
*,
|
||
has_default_val: bool = False,
|
||
default_val: str = "",
|
||
has_quit_val: bool = True,
|
||
quit_val: str = "q",
|
||
check_func: Callable[[str], bool] = None,
|
||
ask_again: bool = True,
|
||
need_confirm: bool = False) -> tuple[str, bool]:
|
||
"""
|
||
请求用户输入
|
||
|
||
:param prompt_msg: 向用户展示的提示信息（不需要加冒号）
|
||
:param err_msg: 当用户输入不符合要求时展示的错误信息
|
||
:param has_default_val: 当前请求是否有默认值
|
||
:param default_val: 当用户没有输入内容时使用的默认值，若当前请求设定为无默认值，则此参数无效
|
||
:param has_quit_val: 当前请求是否有退出值
|
||
:param quit_val: 用户可以输入此值直接退出请求，若当前请求设定为无退出值，则此参数无效
|
||
:param check_func: 用于检查用户输入的值是否符合要求的函数，若为 None 则不进行检查
|
||
:param ask_again: 当用户输入的值不符合要求时是否再次请求输入
|
||
:param need_confirm: 当用户输入后是否请求确认
|
||
:return: 用户输入值和一个布尔值组成的元组，后者标记是否成功获取请求（非退出或者不符合要求）
|
||
"""
|
||
while True:
|
||
print(prompt_msg, end="")
|
||
# 检查默认值
|
||
if has_default_val:
|
||
print(f" (Default [{default_val}]): ", end="")
|
||
input_val = input() or default_val
|
||
else:
|
||
input_val = input(": ")
|
||
# 检查退出值
|
||
if has_quit_val and input_val == quit_val:
|
||
print("Quit.")
|
||
return "", False
|
||
# 确认环节
|
||
if need_confirm:
|
||
# 这里是一个递归调用，所以 need_confirm 不能是 True，否则就会没玩没了地确认了
|
||
result, state = request_input(
|
||
f"Do you confirm the value [{input_val}] (yes/no)",
|
||
has_default_val=False, has_quit_val=False,
|
||
ask_again=True, need_confirm=False,
|
||
check_func=lambda x: x.lower() in ("y", "yes", "n", "no")
|
||
)
|
||
# 确认没通过，重新开始
|
||
if result in ("n", "no"):
|
||
continue
|
||
|
||
# 如果有检查函数，且当前输入不符合要求
|
||
if check_func is not None and not check_func(input_val):
|
||
print(err_msg, end=" ")
|
||
if ask_again:
|
||
print("Please enter again.")
|
||
continue
|
||
else:
|
||
print()
|
||
return "", False
|
||
|
||
return input_val, True
|
||
|
||
|
||
class MenuConfig(TypedDict, total=False):
|
||
left_padding: int
|
||
right_padding: int
|
||
top_padding: int
|
||
bottom_padding: int
|
||
edge_char: str
|
||
wall_char: str
|
||
serial_marker: str
|
||
draw_menu_again: bool
|
||
|
||
|
||
DEFAULT_MENU_CONFIG: MenuConfig = {
|
||
"left_padding": 2,
|
||
"right_padding": 10,
|
||
"top_padding": 1,
|
||
"bottom_padding": 1,
|
||
"edge_char": "#",
|
||
"wall_char": " ",
|
||
"serial_marker": ". ",
|
||
"draw_menu_again": False,
|
||
}
|
||
|
||
|
||
class _OptionNode(object):
|
||
"""选项节点"""
|
||
|
||
# 返回上级菜单的执行函数的返回值
|
||
_RETURN_CODE = 0xbabe
|
||
# 进入下级菜单的执行函数的返回值
|
||
_ENTER_CODE = 0xbeef
|
||
# 退出整个工具的返回值
|
||
_QUIT_CODE = 0xcafe
|
||
|
||
def __init__(self,
|
||
title: str = "Untitled Option",
|
||
exec_func: Callable = None,
|
||
args: tuple = None,
|
||
kwargs: dict = None,
|
||
menu_config: MenuConfig = None,
|
||
_is_root: bool = False):
|
||
"""
|
||
:param title: 选项标题
|
||
:param exec_func: 选择该选项后要执行的函数；对于用户设定的节点，该执行函数必须返回 None
|
||
:param args: 执行函数的位置参数
|
||
:param kwargs: 执行函数的命名参数
|
||
:param menu_config: 绘制子选项菜单的参数；只在根选项设置，子选项继承
|
||
:param _is_root: 标记当前节点是否是根节点；内部使用
|
||
"""
|
||
self.title = title
|
||
self.exec_func = exec_func if exec_func is not None else self._default_func
|
||
self.args = args if args is not None else ()
|
||
self.kwargs = kwargs if kwargs is not None else {}
|
||
self.children = [] # type: list[_OptionNode]
|
||
self._is_root = _is_root
|
||
|
||
self.menu_config = DEFAULT_MENU_CONFIG.copy()
|
||
# 补充没有的设置
|
||
if menu_config is not None:
|
||
self.menu_config.update(menu_config)
|
||
|
||
@staticmethod
|
||
def _default_func():
|
||
"""用户没有指定执行函数时执行的默认函数"""
|
||
print("No function")
|
||
|
||
def _get_max_length_of_option_titles(self) -> int:
|
||
"""
|
||
获取菜单选项中最长的选项标题的长度
|
||
|
||
:return: 菜单选项中最长的选项标题的长度
|
||
"""
|
||
return max([len(c.title) for c in self.children])
|
||
|
||
def _draw_menu(self):
|
||
"""
|
||
绘制菜单选项
|
||
|
||
:return: None
|
||
"""
|
||
left = self.menu_config["left_padding"]
|
||
right = self.menu_config["right_padding"]
|
||
top = self.menu_config["top_padding"]
|
||
bottom = self.menu_config["bottom_padding"]
|
||
ec = self.menu_config["edge_char"]
|
||
wc = self.menu_config["wall_char"]
|
||
sm = self.menu_config["serial_marker"]
|
||
|
||
max_len_option = self._get_max_length_of_option_titles()
|
||
# 获取最大选项序号的位数
|
||
max_len_serial = len(str(len(self.children)))
|
||
# 一个选项的总宽度
|
||
option_width = max_len_serial + len(sm) + max_len_option
|
||
# 菜单的宽度（加上左右两边的空白，但不包括边界）
|
||
menu_width = left + option_width + right
|
||
# 上下边界字符
|
||
top_bottom_edge = ec * (menu_width + len(ec) * 2)
|
||
# 墙是包括边界和空白（单行）
|
||
top_bottom_wall = f"{ec}{wc * menu_width}{ec}"
|
||
left_wall = f"{ec}{wc * left}"
|
||
right_wall = f"{wc * right}{ec}"
|
||
|
||
print("\n".join([top_bottom_edge] + [top_bottom_wall] * top))
|
||
for i, opt in enumerate(self.children):
|
||
# 序号右对齐，选项标题左对齐
|
||
print(f"{left_wall}{i + 1:>{max_len_serial}}{sm}{opt.title:<{max_len_option}}{right_wall}")
|
||
print("\n".join([top_bottom_wall] * bottom + [top_bottom_edge]))
|
||
|
||
def _request_option(self) -> _OptionNode | None:
|
||
"""
|
||
请求用户输入选项序号
|
||
|
||
:return: 用户选择的选项节点或 None
|
||
"""
|
||
choice, state = request_input(
|
||
"\nYour option", "No such option.",
|
||
has_default_val=False, has_quit_val=False,
|
||
check_func=lambda x: x.isdigit() and 1 <= int(x) <= len(self.children),
|
||
ask_again=False, need_confirm=False
|
||
)
|
||
if state is False:
|
||
return None
|
||
|
||
return self.children[int(choice) - 1]
|
||
|
||
def _enter_next_level(self) -> int:
|
||
"""
|
||
进入下一级菜单（执行函数）
|
||
|
||
:return: 状态码
|
||
"""
|
||
state = self._ENTER_CODE
|
||
|
||
while state not in (self._RETURN_CODE, self._QUIT_CODE):
|
||
# 如果上次执行不是返回上级菜单或退出，就继续在当前菜单循环
|
||
# 用户指定是否需要重新绘制菜单，
|
||
# 或者如果上次执行是进入下级菜单，则一定要绘制菜单
|
||
# 这里 state == self._ENTER_CODE 有两种情况，
|
||
# 一种是第一次进入 while 循环的时候，此时表示用户选择进入下级菜单
|
||
# 另一种是用户选择返回上级菜单，结束了 while 循环，
|
||
# 此函数返回 self._ENTER_CODE，被上层的该函数的 while 循环捕获到
|
||
if self.menu_config["draw_menu_again"] or state == self._ENTER_CODE:
|
||
self._draw_menu()
|
||
|
||
# 此处设置为 None，是为了防止用户输入不符合要求的序号时，
|
||
# 重新循环 state 还是 self._ENTER_CODE 导致又绘制了一遍菜单
|
||
state = None
|
||
opt = self._request_option()
|
||
if opt is None:
|
||
continue
|
||
|
||
state = opt.exec_func(*opt.args, **opt.kwargs)
|
||
|
||
# 如果是退出信号，就一路退回到根节点
|
||
if state == self._QUIT_CODE:
|
||
return self._QUIT_CODE
|
||
else:
|
||
return self._ENTER_CODE
|
||
|
||
def _return_previous_level(self) -> int:
|
||
"""
|
||
返回上一级菜单（执行函数）
|
||
|
||
:return: 状态码
|
||
"""
|
||
return self._RETURN_CODE
|
||
|
||
def _quiting_level(self) -> int:
|
||
"""
|
||
退出该工具（执行函数）
|
||
|
||
:return: 状态码
|
||
"""
|
||
return self._QUIT_CODE
|
||
|
||
def add_option(self,
|
||
title: str = "Untitled Option",
|
||
exec_func: Callable = None,
|
||
args: tuple = None,
|
||
kwargs: dict = None) -> _OptionNode:
|
||
"""
|
||
添加选项
|
||
|
||
:param title: 选项标题
|
||
:param exec_func: 选择该选项后要执行的函数；对于用户设定的节点，该执行函数必须返回 None
|
||
:param args: 执行函数的位置参数
|
||
:param kwargs: 执行函数的命名参数
|
||
:return: 选项节点对象
|
||
"""
|
||
# 如果在此父选项下创建了新的选项，该父选项会成为下一级的菜单入口
|
||
# 此处进行后缀标记
|
||
if not self.title.endswith(" [d]"):
|
||
self.title = f"{self.title} [d]"
|
||
|
||
option = _OptionNode(title, exec_func, args, kwargs, self.menu_config, _is_root=False)
|
||
self.children.append(option)
|
||
|
||
# 父选项成为菜单入口之后，其执行函数只能为“进入下一级菜单”
|
||
self.exec_func = self._enter_next_level
|
||
return option
|
||
|
||
def add_return_option(self):
|
||
"""
|
||
添加返回上级菜单的选项
|
||
|
||
:return: None
|
||
"""
|
||
self.add_option(title="[Back]", exec_func=self._return_previous_level)
|
||
|
||
def add_exit_option(self):
|
||
"""
|
||
添加退出菜单选项
|
||
|
||
:return: None
|
||
"""
|
||
self.add_option(title="[Exit]", exec_func=self._quiting_level)
|
||
|
||
def start_loop(self):
|
||
"""开启循环，只在根选项调用"""
|
||
if self._is_root:
|
||
self._enter_next_level()
|
||
print("Exit CliHelper. Bye.")
|
||
else:
|
||
print("Current option is not the root option. Exit.")
|
||
|
||
|
||
class CliHelper(_OptionNode):
|
||
"""命令行辅助工具"""
|
||
|
||
def __init__(self, menu_config: MenuConfig = None, show_version: bool = True):
|
||
super().__init__("Main Menu", menu_config=menu_config, _is_root=True)
|
||
|
||
if show_version:
|
||
print(f"CliHelper v{version[0]}.{version[1]}.{version[2]} ({version[-1]})\n")
|