练习内容：
(1)函数装饰器。
(2)使用魔术方法__call__,将一个类实现为装饰器。 (3)使用魔术方法__enter__与__exit__,实现一个类，使其能为函数提供上下文管理功能。1.函数装饰器
1 __author__ = 'Orcsir' 2 from functools import wraps, update_wrapper 3 import time 4 5 6 # 将一个函数实现为装饰器 7 def timeit(func): 8 @wraps(func) 9 def _wrapper(*args, **kwargs):10 print("Inject some code before func run.")11 12 ret = func(*args, **kwargs)13 14 print("Inject some code after func run.")15 16 return ret17 18 return _wrapper19 20 21 @timeit22 def func(*args, **kwargs):23 print("func start processing.")24 time.sleep(2) # 模拟处理时间25 print("func end processing.")26 return None27 28 # Test29 func(1,2) 2.使用魔术方法__call__,将一个类实现为装饰器
1 __author__ = 'Orcsir' 2 from functools import wraps, update_wrapper 3 import time 4 5 # 将一个类实现为装饰器 6 class TimeIt: 7 def __init__(self, wrapped): 8 self.__wrapped = wrapped 9 wraps(wrapped)(self)10 11 def __call__(self, *args, **kwargs):12 print("Inject some code before func run.")13 ret = self.__wrapped(*args, **kwargs)14 print("Inject some code after func run.")15 return ret16 17 18 @TimeIt19 def func(*args, **kwargs):20 print("func start processing.")21 time.sleep(2) # 模拟处理时间22 print("func end processing.")23 return None24 25 # Test26 func() 3.使用魔术方法__enter__与__exit__,实现一个类，使其能为函数提供上下文管理功能
1 __author__ = 'Orcsir' 2 from functools import wraps, update_wrapper 3 import time 4 5 # 实现一个类，使其为函数提供上下文管理功能 6 class ContextMage4Func: 7 def __init__(self, func): 8 self.__func = func 9 10 def __enter__(self):11 print("为func函数的运行准备环境.")12 return self.__func13 14 def __exit__(self, exc_type, exc_val, exc_tb):15 print("func函数退出或出现异常时，做一些清理工作.")16 17 18 def func(*args, **kwargs):19 print("func start processing.")20 time.sleep(2) # 模拟处理时间21 print("func end processing.")22 return None23 24 # Test25 with ContextMage4Func(func):26 func(1, 2)27 print("~~~~~~~~~~~~~~~~~~~~~~")28 with ContextMage4Func(func) as f:29 f(1, 2)