在python中将Callable键入特定的方法

Question

我试图对输入进行非常严格的操作，但是我只想用几种非常特定的方法来键入Callable时遇到一个问题！

我有一堂课：

class Complex:

    def __add__(self, other):
        return Complex()

    def __sub__(self, other):
        return Complex()

    def __div__(self, other):
        return Complex()

在另一个文件中，我想编写一个接受Callable的方法，但是该方法的参数只能是add函数或sub函数。 这意味着如果我尝试将div作为函数传递给test_add_sub（），我希望linter引发错误。

下面的代码似乎不起作用:(当我将div函数传递给test_add_sub时，lint或编译器都会抱怨！也无法编写Complex.complex_func（）。

    add_sub_type = Complex.__add__ or Complex.__sub__
    add_sub_type2 = Callable[[Complex.__add__ or Complex.__sub__], None]


    def test_add_sub(complex_func: add_sub_type) -> None:
        print(complex_func)
        Complex.complex_func() <-- 'Class Complex has no complex_func member'

预先非常感谢大家。

Answer 1

typing不能做到这一点。 静态伪类型系统不允许您定义具有任意成员集的类型。 （此外， Complex.complex_func()也不是您如何调用complex_func的方法。）

如果您确实希望对此进行基于类型的静态检查，则可以使用枚举代替方法：

class ComplexAddSub(enum.Enum):
    add = Complex.__add__
    sub = Complex.__sub__

    def __call__(self, left: Complex, right: Complex):
        return self.value(left, right)

def whatever(func: ComplexAddSub):
    func(Complex(), Complex())

whatever(ComplexAddSub.add)

Answer 2

运行时检测

def test_add_sub(complex_func):
    tester = {Complex.__addr__: False, Complex.__sub__: False}
    if (tester.get(complex_func, True)):
        raise RuntimeError('Invalid argument!')

    #complex_func(some_complex_object, another_complex_object)

在此处阅读有关实例方法的更多信息。