python中带有参数的装饰器

1. 带有参数的装饰器介绍

        带有参数的装饰器就是使用装饰器装饰函数的时候可以传入指定参数，语法格式: @装饰器(参数,...)

错误写法:

def decorator(fn, flag):
    def inner(num1, num2):
        if flag == "+":
            print("--正在努力加法计算--")
        elif flag == "-":
            print("--正在努力减法计算--")
        result = fn(num1, num2)
        return result
    return inner


@decorator('+')
def add(a, b):
    result = a + b
    return result

result = add(1, 3)
print(result)

执行结果:

代码说明:

• 装饰器只能接收一个参数，并且还是函数类型。

正确写法:

        在装饰器外面再包裹上一个函数，让最外面的函数接收参数，返回的是装饰器，因为@符号后面必须是装饰器实例。

示例代码1：

# 添加输出日志的功能
def logging(flag):

    def decorator(fn):
        def inner(num1, num2):
            if flag == "+":
                print("--正在努力加法计算--")
            elif flag == "-":
                print("--正在努力减法计算--")
            result = fn(num1, num2)
            return result
        return inner

    # 返回装饰器
    return decorator


# 使用装饰器装饰函数
@logging("+")
def add(a, b):
    result = a + b
    return result


@logging("-")
def sub(a, b):
    result = a - b
    return result

result = add(1, 2)
print(result)

result = sub(1, 2)
print(result)

运行结果： 

示例代码2：

# !/usr/bin/env python
# -*- coding:utf-8 -*-


# 带参数的装饰器
def wrapper_out(parameter):
    print(parameter)

    def wrapper(func):
        def inner(*args, **kwargs):
            ret = func(*args, **kwargs)
            return ret

        return inner

    return wrapper


@wrapper_out('QQ')
def qq():
    print('成功登录QQ！')


qq()
"""
@wrapper_out('QQ') 分析解读：
当函数执行到带参数装饰器 @wrapper_out('QQ') 这句时，分两步执行：
    1.执行wrapper_out('QQ')这个函数，把相应的参数'QQ'传给parameter，并且得到返回值wrapper函数名
    2.将@与wrapper结合，得到之前熟悉的标准版的装饰器，按照装饰器的执行流程执行。
"""

运行结果：

示例代码3：

# 带参数装饰器
def wrapper_out(parameter):
    print(f"请登录{parameter}平台")

    def wrapper(func):
        def inner(*args, **kwargs):
            if parameter == 'wechat':
                username = input('请输入用户名：').strip()
                password = input('请输入密码：').strip()
                with open('wechat.txt', encoding='utf-8', mode='r') as f:
                    for line in f:
                        user, pwd = line.strip().split('|')
                        if user == username and pwd == password:
                            ret = func(*args, **kwargs)
                            return ret
                    return False
            elif parameter == 'taobao':
                username = input('请输入用户名：').strip()
                password = input('请输入密码：').strip()
                with open('taobao.txt', encoding='utf-8', mode='r') as f:
                    for line in f:
                        user, pwd = line.strip().split('|')
                        if user == username and pwd == password:
                            ret = func(*args, **kwargs)
                            return ret
                    return False

        return inner

    return wrapper


@wrapper_out('wechat')
def wechat():
    print('成功登录微信！')


wechat()


@wrapper_out('taobao')
def taobao():
    print("成功登录淘宝！！！")


taobao()

"""
    两个不同软件微信和淘宝，有不同的账户和密码，
    根据不同软件的登录，写个装饰器验证登录功能
    账户文件名：wechat，wechat|123
    账户文件名：taobao，taobao|456
"""

运行结果：

 示例代码4：

# 带参数装饰器应用：优化版
def wrapper_out(parameter):
    print(f"请登录{parameter}平台：")

    def wrapper(func):
        def inner(*args, **kwargs):
            username = input('请输入用户名：').strip()
            password = input('请输入密码：').strip()
            # 注意文件名需要与装饰器传入参数一致，增强耦合性思想
            with open(f"{parameter}.txt", encoding='utf-8', mode='r') as f:
                for line in f:
                    user, pwd = line.strip().split('|')
                    if user == username and pwd == password:
                        ret = func(*args, **kwargs)
                        return ret
                return False

        return inner

    return wrapper


@wrapper_out('wechat')
def wechat():
    print('成功登录微信！')


wechat()


@wrapper_out('taobao')
def taobao():
    print("成功登录淘宝！！！")


taobao()
"""
    两个不同软件微信和淘宝，有不同的账户和密码，
    根据不同软件的登录，写个装饰器验证登录功能
    账户文件名：wechat，wechat|123
    账户文件名：taobao，taobao|456
"""

运行结果：

​​​​​​​

2. 小结

• 使用带有参数的装饰器，其实是在装饰器外面又包裹了一个函数，使用该函数接收参数，返回是装饰器，因为 @ 符号需要配合装饰器实例使用