Python函数-装饰器

Posted 2020-10-23 Sch01aR#

如果多个函数想通过一个装饰器来实现不同的功能的话，可以给装饰器传入参数，让装饰器里的函数对参数进行判断，来实现不同的功能

# -*- coding:utf-8 -*-
__author__ = "MuT6 Sch01aR"

def deco(info_type):
    def outer(func): #这在传入参数名
        def inner(*args,**kwargs):
            if info_type == \'name\': #对装饰器的参数进行判断
                func(*args, **kwargs)
                name_info = input("your name:")
                print("your name is %s"%name_info)
            if info_type == \'age\':
                func(*args, **kwargs)
                age_info = input("your age:")
                print("your age is %s"%age_info)
        return inner
    return outer

@deco(info_type=\'name\') #给装饰器传参数
def name():
    print("input your name")

@deco(info_type=\'age\')
def age():
    print("input your age")

if __name__ == \'__main__\':
    name()
    age()

运行结果