Python readline模块能否将按键绑定至自定义Python函数？

无需编写C扩展即可实现你要的功能，下面是具体方案和兼容editline的注意事项：

核心实现思路：用ctypes调用底层库 #

Python的ctypes模块可以直接对接系统的readline或editline库，绕过Python标准readline模块的限制，将Python函数包装成C兼容的回调绑定到按键。

针对readline的示例代码 #

import ctypes
import readline

# 加载系统readline库
libreadline = ctypes.CDLL("libreadline.so")

# 定义符合readline要求的回调函数类型
RLHookFunc = ctypes.CFUNCTYPE(ctypes.c_int, ctypes.c_int, ctypes.c_int)

def custom_key_handler(key, count):
    # 记录按键操作
    print(f"触发按键：{key}")
    # 修改提示符
    readline.set_prompt("[新提示符] ")
    # 强制刷新显示
    libreadline.rl_redisplay()
    # 返回0表示已处理该按键，不传递给readline默认逻辑
    return 0

# 将Python函数转为C兼容的回调
c_handler = RLHookFunc(custom_key_handler)

# 绑定F5键（键码63，可根据需要替换为其他键码）
libreadline.rl_bind_key(63, c_handler)

# 测试输入
input("初始提示符：")

现成封装库 #

如果不想自己写ctypes逻辑，可以用第三方库简化：

• readline_callback：专门为Python的readline模块提供按键回调支持，内部基于ctypes实现，无需编译。
• prompt_toolkit：虽然是独立的输入处理库，但原生支持按键绑定到Python函数，且能适配readline/editline等底层终端库，若能接受替换input()，这是更省心的选择。

兼容editline的适配方案 #

editline的API与readline存在差异，需要编写兼容层：

1. 先检测底层库类型，加载对应的动态库（libedit.so）。
2. 适配editline的回调函数签名和绑定逻辑。
3. 注意键码或按键序列的差异（editline常用ANSI序列绑定按键，而非数字键码）。

示例兼容代码：

import ctypes
import readline

def load_terminal_lib():
    try:
        return ("readline", ctypes.CDLL("libreadline.so"))
    except OSError:
        try:
            return ("editline", ctypes.CDLL("libedit.so"))
        except OSError:
            raise RuntimeError("未找到readline或editline库")

lib_type, lib = load_terminal_lib()

if lib_type == "readline":
    RLHookFunc = ctypes.CFUNCTYPE(ctypes.c_int, ctypes.c_int, ctypes.c_int)
    def handler(key, count):
        print(f"按键触发：{key}")
        readline.set_prompt("[READLINE模式] ")
        lib.rl_redisplay()
        return 0
    c_handler = RLHookFunc(handler)
    lib.rl_bind_key(63, c_handler)
elif lib_type == "editline":
    # 定义editline回调类型
    ELHookFunc = ctypes.CFUNCTYPE(ctypes.c_int, ctypes.c_void_p, ctypes.c_int)
    # 获取已初始化的EditLine实例（Python的readline模块已完成初始化）
    el_inst = lib.el_get(None, ctypes.c_int(0))
    def handler(el_ptr, key):
        print(f"按键触发：{key}")
        readline.set_prompt("[EDITLINE模式] ")
        lib.el_redisplay(el_ptr)
        return 0
    c_handler = ELHookFunc(handler)
    # 绑定F5键的ANSI序列（不同终端可能有差异）
    lib.el_bind(el_inst, b"^[[15~", c_handler, None)

# 测试
input("测试提示符：")

注意：editline的按键序列需要根据你的终端实际输出调整，可以用cat -v查看按键对应的转义序列。

内容的提问来源于stack exchange，提问作者user1806566