标签：cto blank highlight ora 函数调用 而且 如何 single set
上一章：函数装饰器和闭包（三）
单分派函数
假设我们现在要开发一个函数，这个函数可以传入一个元素，函数要判断元素的类型，再将其打印出来
from collections.abc import MutableSequence, MutableMapping
def print_item(item):
if isinstance(item, int):
print("int:", item)
elif isinstance(item, MutableSequence):
print("iter:", item)
elif isinstance(item, MutableMapping):
print("map:", item)
else:
print(item)
运行结果：
>>> print_item(1)
int: 1
>>> print_item([1, 2, 3])
sequence: [1, 2, 3]
>>> print_item({"a": 1, "b": 2})
map: {‘a‘: 1, ‘b‘: 2}
这个函数很好理解，没有什么好解释的，但是我们要通过这个函数来展示如何使用functools.singledispatch装饰器让Python实现“重载”的效果
from collections.abc import MutableSequence, MutableMapping
from functools import singledispatch
@singledispatch # <1>
def print_item(item):
print(item)
@print_item.register(int) # <2>
@print_item.register(float)
def _(item):
print("number:", item)
@print_item.register(str) # <3>
def _(item):
print("str:", item)
@print_item.register(MutableSequence) # <4>
def _(item):
print("sequence:", item)
@print_item.register(MutableMapping) # <5>
def _(item):
print("map:", item)
我们用@singledispatch装饰了print_item函数，把这个函数作为基函数，然后再用@print_item.register装饰其他函数，当我们调用print_item，会根据item的类型查找应该执行的函数
运行结果：
>>> print_item(1)
number: 1
>>> print_item(1.1)
number: 1.1
>>> print_item("hello")
str: hello
>>> print_item([1, 2, 3])
sequence: [1, 2, 3]
>>> print_item({"a": 1, "b": 2})
map: {‘a‘: 1, ‘b‘: 2}
参数化装饰器
Python在解释装饰器时，把被装饰的函数当做第一个参数传递给装饰器函数，但怎么样才能让装饰器接收其他参数呢？答案是：创建一个装饰器工厂函数，把参数传给它，返回一个装饰器，然后再把它应用到要装饰的函数上。可能这样的解释听得云里雾里，没关系，我们来看下面一个例子
registry = set() # <1>
def register(active=True): # <2>
def decorate(func): # <3>
print(‘running register(active=%s)->decorate(%s)‘ % (active, func))
if active: # <4>
registry.add(func)
else:
registry.discard(func) # <5>
return func # <6>
return decorate # <7>
@register(active=False) # <8>
def f1():
print(‘running f1()‘)
@register() # <9>
def f2():
print(‘running f2()‘)
def f3():
print(‘running f3()‘)
我们导入这个文件：
>>> from registration_param import *
running register(active=False)->decorate(<function f1 at 0x00000072D8DA4620>)
running register(active=True)->decorate(<function f2 at 0x00000072D8DA46A8>)
>>> registry
{<function f2 at 0x00000072D8DA46A8>}
可以看到，当把register作为装饰器工厂函数，返回的decorate函数得到了执行，由于f1传入的active为False，所以f1没有注册到registry集合对象里面，只有f2被注册到集合对象，我们打印registry，可以看到，里面确实只有f2函数对象
标签：cto blank highlight ora 函数调用 而且 如何 single set
原文地址：https://www.cnblogs.com/beiluowuzheng/p/9310005.html
