Python编程之基础知识练习_002

Posted 2020-11-04 orcsir

练习内容：

 (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 ret
17 
18     return _wrapper
19 
20 
21 @timeit
22 def func(*args, **kwargs):
23     print("func start processing.")
24     time.sleep(2)  # 模拟处理时间
25     print("func end processing.")
26     return None
27 
28 # Test
29 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 ret
16 
17 
18 @TimeIt
19 def func(*args, **kwargs):
20     print("func start processing.")
21     time.sleep(2)  # 模拟处理时间
22     print("func end processing.")
23     return None
24 
25 # Test
26 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.__func
13 
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 None
23 
24 # Test
25 with ContextMage4Func(func):
26     func(1, 2)
27 print("~~~~~~~~~~~~~~~~~~~~~~")
28 with ContextMage4Func(func) as f:
29     f(1, 2)