装饰器的简单实用:通过装饰器在函数运行前判断当前代码运行的环境从而执行相应的业务逻辑
import platform
def judge_platform(func):
def add_env_parameter(a, b):
if platform.system() == "Windows":
env = "Windows"
else:
env = "Linux"
return func(env, a, b)
return add_env_parameter
def test(env, a, b):
if env == "Windows":
print(a + b)
elif env == "Linux":
print(a - b)
if __name__ == '__main__':
c = 10
d = 5
foo = judge_platform(test)
foo(c, d)
import platform
def judge_platform(func):
def add_env_parameter(a, b):
if platform.system() == "Windows":
env = "Windows"
else:
env = "Linux"
return func(env, a, b)
return add_env_parameter
@judge_platform # 相当于运行judge_platform(test)
def test(env, a, b):
if env == "Windows":
print(a + b)
elif env == "Linux":
print(a - b)
if __name__ == '__main__':
c = 10
d = 5
test(c, d)
通过上面的代码比较,我们可以总结出被装饰器装饰的函数执行的过程为:先执行装饰函数,在装饰函数内部调用我们被装饰的函数。