如何在函数定义中强制执行自动参数转换？

Question

考虑这个函数（及其用途）的例子，它接收一个仅作为int 有意义的参数：

def fun(index):
    index = int(index)
    # do something with index

def main():
    print fun(1e6)

我发现我在几个处理整数的函数上不断重复这种模式，因为它确保我确实收到了一些整数（请注意，1e6 是一个浮点实例）。

这有点麻烦，如果忘记它可能会产生问题（有 代码对浮点/整数/字符串/...值的反应不同）。那么这是否可以以某种方式编码在函数签名中？

例如类似于此伪代码的内容：

def fun(int index):
    # do something with index

Answer 1

首先，你真的不应该这样做（尽管有些日子我也想这样做……）。 Python 不是静态类型的，因此您不应该真正引入笨拙的静态类型。

Buuut，回答你的问题。 Decorators:

def ensure_int(fn):
    def wrapper(num):
        return fn(int(num))
    return wrapper

def test(num):
    return num + 2

@ensure_int
def test2(num):
    return num + 2

作品如下：

> test("2") 
ERROR!!!

>test2("2")
4

Answer 2

在 Python 3 中，您可以使用 function annotations，然后编写一个装饰器，将参数转换（或尝试转换）为您指定的类型。

在 Python 2 中，您无法以您想要的方式执行此操作。 Python 2 的语法不支持在函数定义中放入除参数本身之外的任何内容，并且您无法更改 Python 的语法以让您也将类型放入其中。