Python干货-装饰器与偏函数

装饰器

装饰器本质上是一个Python函数，它可以让其他函数在不需要做任何代码变动的前提下增加额外功能，装饰器的返回值也是一个函数对象。它经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计，有了装饰器，我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。

概括的讲，装饰器的作用就是为已经存在的函数或对象添加额外的功能。

举个简单的栗子：

# 定义一个记录日志用来调试的函数
def debug(func):
    def wrapper():
        print("[DEBUG]: enter {}()".format(func.__name__))
        return func()
    return wrapper

# 定义一个say_hello函数
def say_hello():
    print("hello!")

# 调用debug函数并将say_hello函数当作参数传入，返回的是wrapper函数
hello = debug(say_hello)
# 执行wrapper函数
hello()

[DEBUG]: enter say_hello()
hello!

debug函数对say_hello函数进行了包装，也就是增加了更多功能，实际上，debug函数已经是一个装饰器了

更加简单的写法是使用语法糖语法@debug，示例如下：

# 定义一个记录日志用来调试的函数
def debug(func):
    def wrapper():
        print("[DEBUG]: enter {}()".format(func.__name__))
        return func()
    return wrapper

@debug
# 定义一个say_hello函数
def say_hello():
    print("hello!")

say_hello()

[DEBUG]: enter say_hello()
hello!

@debug的效果与hello = debug(say_hello)是一样的，是不是很惊艳！

关于参数问题

定装饰器函数wrapper中调用了原函数say_hello，但如果原函数有参数，或者有很多参数的时候，可以利用python多参数形式，如下：

# 定义一个记录日志用来调试的函数
def debug(func):
    def wrapper(*args, **kwargs):
        print("[DEBUG]: enter {}()".format(func.__name__))
        return func(*args, **kwargs)
    return wrapper

@debug
# 定义一个say_hello函数
def say_hello(name, world):
    print(world, name)

say_hello('Jack', 'hello')

[DEBUG]: enter say_hello()
hello Jack

通过使用*args, **kwargs这得传参数形式，支持元组形式或字典形式的传值结构，不管是有参数还是多个参数，都可以很好的应对

带参数的装饰器

定义一个可以指定日志输出级别的“日志输出装饰器函数”

def logging(level):
    def wrapper(func):
        def inner_wrapper(*args, **kwargs):
            print("[{level}]: enter function {func}()".format(
                level=level,
                func=func.__name__))
            return func(*args, **kwargs)
        return inner_wrapper
    return wrapper

@logging(level='INFO')
def say(something):
    print("say {}!".format(something))

# 如果没有使用@语法，等同于
# say = logging(level='INFO')(say)

@logging(level='DEBUG')
def do(something):
    print("do {}...".format(something))

say('hello')
do("my work")

[INFO]: enter function say()
say hello!
[DEBUG]: enter function do()
do my work...

解析：

当带参数的装饰器被打在某个函数上时，比如@logging(level=‘DEBUG’)，它其实是一个函数，会马上被执行

而被执行的logging函数返回的结果，才是一个装饰器函数O(∩_∩)O哈哈~

偏函数

偏函数可以设定函数的默认行为，示例如下：

num = int('12345')
print(num)

12345

在默认情况下，int()函数会将字符串数字转换成十进制的数据，便也可以通过设定参数来让int()函数将其转换成八进制数字，如下：

num = int('12345', base=8)
print(num)

5349

那如何改变int()函数的默认行为呢？

# 首先导入 functools 模块
import functools

# 定义一个默认转换为二进制的int函数
int2 = functools.partial(int, base=2)
# 调用 
int2('10110101')

181

functools.partial的作用就是将固定一个函数的默认行为，从而简化之后的使用

来看看关于它的描述：

Init signature: functools.partial(self, /, *args, **kwargs)
Docstring:     
partial(func, *args, **keywords) - new function with partial application
of the given arguments and keywords.

所以，这个方法可以接收函数、*args、**kwargs这些对象

这里的base=2是字典结构的参数

max2 = functools.partial(max, 10)
max2(5, 6, 7)  # 相当于max2(10, 5, 6, 7)

10

小结
当函数的参数个数太多，需要简化时，使用functools.partial可以创建一个新的函数，这个新函数可以固定住原函数的部分参数，从而在调用时更简单。