Python中如何指定函数参数类型并限制传入函数范围

你提到的需求完全可行，下面分场景给出具体实现方式：

1. 限制为指定的几个函数之一 #

要让IDE和静态类型检查工具（如mypy、PyCharm）识别限制，可以使用Literal类型（Python 3.8+，需从typing导入），直接把允许的函数作为Literal的参数：

from typing import Literal

def func_as_param():
    print("func_as_param called")

def other_func():
    print("other_func called")

# 仅允许传入func_as_param或other_func
def func_with_func_arg(func: Literal[func_as_param, other_func]):
    print("func_with_func_arg called")
    func()

def test_func_with_func_arg():
    func_with_func_arg(func_as_param)  # 合法
    func_with_func_arg(other_func)     # 合法
    # func_with_func_arg(lambda: None)  # IDE/类型检查工具会报错，运行时不强制

如果需要运行时强制限制，可以在函数内部添加判断逻辑：

def func_with_func_arg(func):
    allowed_funcs = {func_as_param, other_func}
    if func not in allowed_funcs:
        raise ValueError(f"仅允许传入{allowed_funcs}中的函数")
    print("func_with_func_arg called")
    func()

2. 限制为特定模块中的函数 #

运行时强制限制 #

直接检查函数的__module__属性，判断其所属模块：

# 假设要限制的目标模块是my_custom_module
import my_custom_module

def func_with_module_func_arg(func):
    if func.__module__ != my_custom_module.__name__:
        raise ValueError("只能传入my_custom_module模块中的函数")
    func()

静态检查层面可以结合Callable注解提示IDE，更严谨的约束可通过mypy等工具配置规则实现。

3. 限制为特定签名的函数 #

用Callable定义基础签名 #

适合简单的参数、返回值约束：

from typing import Callable

# 限制参数为「无输入、返回None」的函数
def func_with_sig_arg(func: Callable[[], None]):
    func()

def valid_func():
    pass

def invalid_func(a: int):
    pass

func_with_sig_arg(valid_func)  # 合法
# func_with_sig_arg(invalid_func)  # IDE/类型检查工具会报错

用Protocol实现灵活签名约束 #

支持复杂签名（如关键字参数、默认值等）：

from typing import Protocol

# 定义符合要求的函数签名协议
class RequiredFunc(Protocol):
    def __call__(self, username: str, age: int = 18) -> str:
        ...

def user_info(username: str, age: int = 18) -> str:
    return f"{username}, {age}岁"

def invalid_func(username: str):
    return username

def func_with_protocol_arg(func: RequiredFunc):
    print(func("Bob"))

func_with_protocol_arg(user_info)  # 合法
# func_with_protocol_arg(invalid_func)  # 类型检查报错

你的原代码为什么不生效 #

你写的func: func_as_param是把func_as_param这个函数实例当作类型注解，Python会自动将其解析为该函数的类型（即function类），所有函数都属于这个类型，因此没有限制。type(func_as_param)得到的也是function类，同样无法区分不同的函数实例。

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