本文主要是介绍python装饰器、with和contextmanager,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
1,装饰器:
装饰器是一种简单的记号而已,有如下等价的转换规则(带不带参数的装饰器都一样,只要是一个可调用对象即可)。
@foo
def bar:
pass
等价于:
x = foo
bar = x(bar)
'----------------'
@foo('abc')
def bar:
pass
等价于:
x = foo('abc')
bar = x(bar)
'----------------'
@foo.X('abc').Y
def bar:
pass
等价于:
x = foo.X('abc').Y
bar = x(bar)
真正起作用的是"函数式编程"方面的知识,包括闭包,LEGB访问规则,自由变量等。
装饰器的用处:返回一个增强版的函数,主要是对函数的上文和下文,都可以进行处理。
比如:参数检查,缓存数据(建立输入与输出的字典),代理,上下文提供者
from threading import RLock
lock = RLock()
def locker(function):def _locker(*args,**kw):lock.acquire()try:return function(*args,**kw)finally:lock.release()return _locker@locker
def thread_safe():pass
以上装饰器的定义都没有问题,但还差最后一步!!
因为我们讲了函数也是对象,它有__name__等属性,但经过decorator装饰之后的函数,
它们的__name__已经从原来的'thread_safe'变成了'_locker'。所以不完美。
不需要直接编写wrapper.__name__ = func.__name__这样的代码,Python内置的functools.wraps就是干这个事的。
import functools
def locker(function):@functools.wraps(function)def _locker(*args,**kw):lock.acquire()try:return function(*args,**kw)finally:lock.release()return _locker
2,with语句:
任何实现了 __enter__和__exit__的类都可以和 with使用。class Context(object):def __enter__(self):print 'entering the zone'def __exit__(self, exception_type, exception_value, exception_traceback):print 'leaving the zone'if exception_type is None:print 'with no error'else:print 'with an error (%s)' % exception_value'异常是会被新的异常覆盖的raise TypeError('i am the second bug')
try:with Context():print 'i am the buggy zone'raise TypeError('i am the bug')
except Exception,s:print 'I got :',Exception,s
3,contextmanager装饰器:
一种和yield结合使用的方便的上下文管理方式,切记yield是协程技术。
Typical usage: # @contextmanager def some_generator(<arguments>): <setup> try: yield <value> finally: <cleanup> This makes this: with some_generator(<arguments>) as <variable>: <body> equivalent to this: <setup> try: <variable> = <value> <body> finally: <cleanup>
'---------自己实现contextmanager装饰器---------------'
class MyGeneratorContextManager(object):def __init__(self, gen):print("__init__ called")self.gen = gendef __enter__(self):print("__enter__ called")return self.gen.next()def __exit__(self, exc_type, exc_val, exc_tb):print("__exit__called exc_type = %s, exc_val = %s, exc_tb = %s"\% (exc_type, exc_val, exc_tb))try:' 即便不调用next也是没有关系的,只是不会做清理的处理,不会执行print("end foo"),但是finally还是程序结束前总会被执行的'return self.gen.next() except StopIteration: '是不会捕获with中的异常,但是这里抛出的异常可以覆盖with中抛出的异常!'print 'nothing to do with StopIteration'def MyContextManager(func):def _deco(*args,**kwargs):print("func info:", func)return MyGeneratorContextManager(func(*args,**kwargs))return _deco@MyContextManager
def foo(*args,**kwargs):try: # 尝试用老方法捕捉错误print("start foo:", args,kwargs)yield [1, 2, 3]print("end foo") # 需要调用self.gen.next()才能输出except (Exception, AssertionError):print("EXCEPTION ENCOUNTERED!")finally:print("FINALLY")with foo("oh no!") as tmp: # tmp的值就是__enter__返回的值print("START WITH")print 'tmp:',tmpassert 1>2# 出错之后直接从with中跳出去,下面不可能被执行print("END WITH")
4,多个装饰器
#coding:utf-8
def decorator1(func): print 1def wrapper(): print 2func()print 3print 4 return wrapper def decorator2(func): print 5def wrapper(): print 6 func()print 7 print 8 return wrapper @decorator1
@decorator2
def test(): print 'hello python!' test() # 等价于下面两行代码
# test = decorator2(test)
# test = decorator1(test)
这篇关于python装饰器、with和contextmanager的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!