Python编程之基础知识练习_002

练习内容：

 (1)函数装饰器。
 (2)使用魔术方法__call__,将一个类实现为装饰器。
 (3)使用魔术方法__enter__与__exit__,实现一个类，使其能为函数提供上下文管理功能。

1.函数装饰器

__author__ = 'Orcsir'
from functools import wraps, update_wrapper
import time


# 将一个函数实现为装饰器
def timeit(func):
    @wraps(func)
    def _wrapper(*args, **kwargs):
        print("Inject some code before func run.")

        ret = func(*args, **kwargs)

        print("Inject some code after func run.")

        return ret

    return _wrapper


@timeit
def func(*args, **kwargs):
    print("func start processing.")
    time.sleep(2)  # 模拟处理时间
    print("func end processing.")
    return None

# Test
func(1,2)

 2.使用魔术方法__call__,将一个类实现为装饰器

__author__ = 'Orcsir'
from functools import wraps, update_wrapper
import time

# 将一个类实现为装饰器
class TimeIt:
    def __init__(self, wrapped):
        self.__wrapped = wrapped
        wraps(wrapped)(self)

    def __call__(self, *args, **kwargs):
        print("Inject some code before func run.")
        ret = self.__wrapped(*args, **kwargs)
        print("Inject some code after func run.")
        return ret


@TimeIt
def func(*args, **kwargs):
    print("func start processing.")
    time.sleep(2)  # 模拟处理时间
    print("func end processing.")
    return None

# Test
func()

3.使用魔术方法__enter__与__exit__,实现一个类，使其能为函数提供上下文管理功能

__author__ = 'Orcsir'
from functools import wraps, update_wrapper
import time

# 实现一个类，使其为函数提供上下文管理功能
class ContextMage4Func:
    def __init__(self, func):
        self.__func = func

    def __enter__(self):
        print("为func函数的运行准备环境.")
        return self.__func

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("func函数退出或出现异常时，做一些清理工作.")


def func(*args, **kwargs):
    print("func start processing.")
    time.sleep(2)  # 模拟处理时间
    print("func end processing.")
    return None

# Test
with ContextMage4Func(func):
    func(1, 2)
print("~~~~~~~~~~~~~~~~~~~~~~")
with ContextMage4Func(func) as f:
    f(1, 2)