Python 中怎么函数功能进行扩展，很多新手对此不是很清楚，为了帮助大家解决这个难题，下面小编将为大家详细讲解，有这方面需求的人可以来学习下，希望你能有所收获。
装饰器可以对原函数进行功能扩展，但还不需要修改原函数的内容(开闭原则)，也不需要修改原函数的调用。
demo.py（装饰器，@）：
计算
安全
数据库
网络和加速
企业服务
# 闭包
def w1(func):
def inner():
# 对原函数进行功能扩展
print("功能扩展")
func()
# return func() # 如果原函数需要返回值,可以return
return inner # 闭包
@w1
# 相当于 f1 = w1(f1)
def f1():
print('f1') # 原函数不需要修改
f1() # 原函数的调用也不需要修改demo.py（装饰器通用格式,对不定长参数并且有返回值的函数进行装饰）：
def set_func(func):
def call_func(*args, **kwargs):
print("装饰器扩展的功能")
return func(*args, **kwargs) # 这里的*和*表示拆包。 不管有没有返回值,return都没问题。
return call_func
@set_func # 相当于 test1 = set_func(test1)
# 对含有不定长参数并且有返回值的函数进行装饰。
def test1(num, *args, **kwargs):
print("-----test1----%d" % num)
return "ok"
ret = test1(100)
print(ret)demo.py（多个装饰器的装饰顺序）：
def add_1(func):
def call_func(*args, **kwargs):
print("装饰器1 扩展的功能")
return func(*args, **kwargs)
return call_func
def add_2(func):
def call_func(*args, **kwargs):
print("装饰器2 扩展的功能")
return func(*args, **kwargs)
return call_func
@add_2
@add_1
# 先装饰add_1，再装饰add_2
def test1():
print("------test1------")
test1() # 在调用函数之前就已经装饰好了。
# 装饰器2 扩展的功能
# 装饰器1 扩展的功能
# ------test1------demo.py（用类充当装饰器）：
# 用类充当装饰器
class Test(object):
def __init__(self, func):
self.func = func
def __call__(self, *args, **kwargs):
print("这里是装饰器添加的功能.....")
return self.func(*args, **kwargs)
@Test # 相当于get_str = Test(get_str) # 实例化对象,调用__init__方法。
def get_str():
return "haha"
print(get_str()) # 实例对象(),会自动调用对象的__call__方法。@functools.wraps修饰装饰器的内层函数。(修饰内层函数后,被装饰器装饰的函数的__name__、__doc__不会被装饰器改变)
demo.py（@functools.wraps修饰装饰器的内层函数）：
# coding:utf-8
import functools # 导入
# 自定义的装饰器
def login_required(func):
@functools.wraps(func)
# 装饰器的内层函数,一般要加@functools.wraps装饰器
def wrapper(*arg, **kwargs):
"""wrapper的说明文档"""
# 。。。
return func(*arg, **kwargs)
return wrapper
# 使用自定义的装饰器
@login_required
def demofunc():
"""demofunc的说明文档"""
pass
print(demofunc.__name__) # 不加@functools.wraps装饰器时："wrapper"。 加装饰器时："demofunc"
print(demofunc.__doc__) # 不加@functools.wraps装饰器时："wrapper的说明文档"。 加装饰器时："demofunc的说明文档"看完上述内容是否对您有帮助呢？如果还想对相关知识有进一步的了解或阅读更多相关文章，请关注亿速云行业资讯频道，感谢您对亿速云的支持。
亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
