python async with和async for的使用
网上asyncwith和asyncfor的中文资料比较少,我把PEP492中的官方陈述翻译一下。
异步上下文管理器”asyncwith”
异步上下文管理器指的是在enter和exit方法处能够暂停执行的上下文管理器。
为了实现这样的功能,需要加入两个新的方法:__aenter__和__aexit__。这两个方法都要返回一个awaitable类型的值。
异步上下文管理器的一种使用方法是:
classAsyncContextManager: asyncdef__aenter__(self): awaitlog('enteringcontext') asyncdef__aexit__(self,exc_type,exc,tb): awaitlog('exitingcontext')
新语法
异步上下文管理器使用一种新的语法:
asyncwithEXPRasVAR: BLOCK
这段代码在语义上等同于:
mgr=(EXPR) aexit=type(mgr).__aexit__ aenter=type(mgr).__aenter__(mgr) exc=True VAR=awaitaenter try: BLOCK except: ifnotawaitaexit(mgr,*sys.exc_info()): raise else: awaitaexit(mgr,None,None,None)
和常规的with表达式一样,可以在一个asyncwith表达式中指定多个上下文管理器。
如果向asyncwith表达式传入的上下文管理器中没有__aenter__和__aexit__方法,这将引起一个错误。如果在asyncdef函数外面使用asyncwith,将引起一个SyntaxError(语法错误)。
例子
使用asyncwith能够很容易地实现一个数据库事务管理器。
asyncdefcommit(session,data): ... asyncwithsession.transaction(): ... awaitsession.update(data) ...
需要使用锁的代码也很简单:
asyncwithlock: ...
而不是:
with(yieldfromlock): ...
异步迭代器“asyncfor”
一个异步可迭代对象(asynchronousiterable)能够在迭代过程中调用异步代码,而异步迭代器就是能够在next方法中调用异步代码。为了支持异步迭代:
1、一个对象必须实现__aiter__方法,该方法返回一个异步迭代器(asynchronousiterator)对象。
2、一个异步迭代器对象必须实现__anext__方法,该方法返回一个awaitable类型的值。
3、为了停止迭代,__anext__必须抛出一个StopAsyncIteration异常。
异步迭代的一个例子如下:
classAsyncIterable: def__aiter__(self): returnself asyncdef__anext__(self): data=awaitself.fetch_data() ifdata: returndata else: raiseStopAsyncIteration asyncdeffetch_data(self): ...
新语法
通过异步迭代器实现的一个新的迭代语法如下:
asyncforTARGETinITER: BLOCK else: BLOCK2
这在语义上等同于:
iter=(ITER) iter=type(iter).__aiter__(iter) running=True whilerunning: try: TARGET=awaittype(iter).__anext__(iter) exceptStopAsyncIteration: running=False else: BLOCK else: BLOCK2
把一个没有__aiter__方法的迭代对象传递给asyncfor将引起TypeError。如果在asyncdef函数外面使用asyncwith,将引起一个SyntaxError(语法错误)。
和常规的for表达式一样,asyncfor也有一个可选的else分句。.
例子1
使用异步迭代器能够在迭代过程中异步地缓存数据:
asyncfordataincursor: ...
这里的cursor是一个异步迭代器,能够从一个数据库中每经过N次迭代预取N行数据。
下面的语法展示了这种新的异步迭代协议的用法:
classCursor: def__init__(self): self.buffer=collections.deque() asyncdef_prefetch(self): ... def__aiter__(self): returnself asyncdef__anext__(self): ifnotself.buffer: self.buffer=awaitself._prefetch() ifnotself.buffer: raiseStopAsyncIteration returnself.buffer.popleft()
接下来这个Cursor类可以这样使用:
asyncforrowinCursor(): print(row) whichwouldbeequivalenttothefollowingcode: i=Cursor().__aiter__() whileTrue: try: row=awaiti.__anext__() exceptStopAsyncIteration: break else: print(row)
例子2
下面的代码可以将常规的迭代对象变成异步迭代对象。尽管这不是一个非常有用的东西,但这段代码说明了常规迭代器和异步迭代器之间的关系。
classAsyncIteratorWrapper: def__init__(self,obj): self._it=iter(obj) def__aiter__(self): returnself asyncdef__anext__(self): try: value=next(self._it) exceptStopIteration: raiseStopAsyncIteration returnvalue asyncforletterinAsyncIteratorWrapper("abc"): print(letter)
为什么要抛出StopAsyncIteration?
协程(Coroutines)内部仍然是基于生成器的。因此在PEP479之前,下面两种写法没有本质的区别:
defg1(): yieldfromfut return'spam'
和
defg2(): yieldfromfut raiseStopIteration('spam')
自从PEP479得到接受并成为协程的默认实现,下面这个例子将StopIteration包装成一个RuntimeError。
asyncdefa1(): awaitfut raiseStopIteration('spam')
告知外围代码迭代已经结束的唯一方法就是抛出StopIteration。因此加入了一个新的异常类StopAsyncIteration。
由PEP479的规定,所有协程中抛出的StopIteration异常都被包装在RuntimeError中。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。