Python帶參數的函數裝飾器

 

 

# -*- coding: utf-8 -*-
# author：baoshan
# 帶參數的函數裝飾器


def say_hello(country):
    def wrapper(func):
        def deco(*args, **kwargs):
            if country == 'china':
                print('你好！')
            elif country == 'america':
                print('hello')
            else:
                return
            func(*args, **kwargs)
        return deco
    return wrapper


@say_hello('china')
def chinese():
    print('我來自中國。')


@say_hello('america')
def america():
    print('I am from America.')


america()
print('-'*20)
chinese()

 

輸出結果：

hello
I am from America.
--------------------
你好！
我來自中國。

 

帶參數的函數裝飾器

參考自：https://zhuanlan.zhihu.com/p/65968462