装饰器
装饰本质上就是一个闭包函数，它可以对我们已有函数进行额外的功能拓展，装饰器符合了 开发中的封闭开放原则
装饰器的作用
在不改变原有函数的源代码的情况下，给函数增加新的功能
def decorator(fn): # fn: 需要装饰的函数def inner():'''函数执行之前需要执行的代码'''fn() # 执行被装饰的目标函数'''函数执行之前需要执行的代码'''return inner装饰器的使用
# 定义一个装饰器
def login(fn):def inner():print('登录')fn()return inner# 定义需要被装饰的函数
def comment():print('评论')comment = login(comment)
comment()执行结果
装饰器语法糖
如果每次编写代码都要写comment = login(comment)，那台麻烦了，所以就有了更简洁的写法，语法糖
格式：@装饰器名字
def login(fn):def inner():print('登录')fn()return inner@login
def comment():print('评论')comment()执行结果
简单小案例
打印数字1-1000需要花费多少时间？
import timedef decorator(func):def inner():start_time = time.time()func()end_time = time.time()print(f"打印数字1-1000需要花费{end_time - start_time}")return inner@decorator
def func1():for i in range(1000):print(i)func1()执行结果
装饰带有参数的函数
def decorator(fn):def inner(a, b):fn(a, b)return inner@decorator
def sum_num(a, b):result = a + bprint(result)sum_num(1, 2)装饰的函数带有返回值
def decorator(fn):def inner(a, b):return fn(a, b)return inner@decorator
def sum_num(a, b):result = a + breturn resultresult = sum_num(1, 2)
print(result)装饰的函数带有不定长参数
def decorator(fn):def inner(*args, **kwargs):fn(*args, **kwargs)return inner@decorator
def sum_num(*args, **kwargs):print(args, kwargs)sum_num(1, 2, 3, name='张三', age='李四')通用装饰器
def decorator(fn):def inner(*args, **kwargs):result = fn(*args, **kwargs)return resultreturn inner@decorator
def sum_num(*args, **kwargs):return args, kwargsprint(sum_num(1, 2, 3, name='张三'))多个装饰器的使用
多个装饰器装饰一个函数
def check1(fn):def inner():print('登录一')fn()return innerdef check2(fn):def inner():print('登录二')fn()return inner@check1
@check2
def comment():print('评论')comment()执行结果
带有参数的装饰器
def check(flag): # 这里起到了函数里面的所有函数都能够用到flagdef decorator(fn):def inner(num1, num2):if flag == '+':print('正在进行加法运算')return fn(num1, num2)return innerreturn decorator@check('+') # 第一层函数只是为了接受参数
def add(a, b):result = a + breturn resultprint(add(1, 5))执行结果
类装饰器
一个类里面一旦实现了__call__方法，那么这个类实例化出来的对象就可以像函数一样进行调用
class Check():num = 1def __call__(self):print('call方法被调用')print(self.num)result = Check()
result()执行结果
使用类装饰器
class Check(object):def __init__(self, fn):self.__fn = fn # fn = commentdef __call__(self, *args, **kwargs):print('登录')self.__fn()@Check
def comment():print('评论')comment()执行结果
property属性
property属性可以把类中的一个方法当作属性来进行使用
定义property属性有两种方式
- 装饰器方式
- 类属性方式
装饰器方式
@property 把方法当作属性使用，当获取属性时会执行下面修饰的方法
@方法名.setter 把方法当作属性使用，当设置属性时会执行下面修饰的方法
装饰器方法 的property属性修饰的方法名一定要一样
class Person(object):def __init__(self):self.__age = 18@propertydef age(self):return self.__age# 获取属性
p = Person()
print(p.age)class Person(object):def __init__(self):self.__age = 18@propertydef age(self):return self.__age@age.setterdef age(self, new_age):self.__age = new_age# 获取属性
p = Person()
print(p.age)# 修改属性
p.age = 100
print(p.age)类属性装饰器
property的参数说明：
第一个参数是获取属性时要执行的方法
第二个参数时设置属性时要执行的方法
class Person(object):def __init__(self):self.__age = 18def get_age(self):return self.__agedef set_age(self, new_age):self.__age = new_ageage = property(get_age, set_age)# 获取属性
p = Person()
print(p.age)# 修改属性
p.age = 100
print(p.age)
