Question

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

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

def main():
    print fun(1e6)

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

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

E.g。这段伪代码的内容：

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

Answer 1

首先，你不应该真的这样做（即使有些日子我也想要......）。 Python不是静态类型的，所以你不应该真正引入cludgy静态类型。

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的语法以允许您将类型放在那里。

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

2 个答案: