python async with和async for的使用
網(wǎng)上async with和async for的中文資料比較少,我把PEP 492中的官方陳述翻譯一下。
異步上下文管理器”async with”
異步上下文管理器指的是在enter和exit方法處能夠暫停執(zhí)行的上下文管理器。
為了實現(xiàn)這樣的功能,需要加入兩個新的方法:__aenter__ 和__aexit__。這兩個方法都要返回一個 awaitable類型的值。
異步上下文管理器的一種使用方法是:
class AsyncContextManager: async def __aenter__(self): await log('entering context') async def __aexit__(self, exc_type, exc, tb): await log('exiting context')
新語法
異步上下文管理器使用一種新的語法:
async with EXPR as VAR: BLOCK
這段代碼在語義上等同于:
mgr = (EXPR) aexit = type(mgr).__aexit__ aenter = type(mgr).__aenter__(mgr) exc = True VAR = await aenter try: BLOCK except: if not await aexit(mgr, *sys.exc_info()): raise else: await aexit(mgr, None, None, None)
和常規(guī)的with表達式一樣,可以在一個async with表達式中指定多個上下文管理器。
如果向async with表達式傳入的上下文管理器中沒有__aenter__ 和__aexit__方法,這將引起一個錯誤 。如果在async def函數(shù)外面使用async with,將引起一個SyntaxError(語法錯誤)。
例子
使用async with能夠很容易地實現(xiàn)一個數(shù)據(jù)庫事務(wù)管理器。
async def commit(session, data): ... async with session.transaction(): ... await session.update(data) ...
需要使用鎖的代碼也很簡單:
async with lock: ...
而不是:
with (yield from lock): ...
異步迭代器 “async for”
一個異步可迭代對象(asynchronous iterable)能夠在迭代過程中調(diào)用異步代碼,而異步迭代器就是能夠在next方法中調(diào)用異步代碼。為了支持異步迭代:
1、一個對象必須實現(xiàn)__aiter__方法,該方法返回一個異步迭代器(asynchronous iterator)對象。
2、一個異步迭代器對象必須實現(xiàn)__anext__方法,該方法返回一個awaitable類型的值。
3、為了停止迭代,__anext__必須拋出一個StopAsyncIteration異常。
異步迭代的一個例子如下:
class AsyncIterable: def __aiter__(self): return self async def __anext__(self): data = await self.fetch_data() if data: return data else: raise StopAsyncIteration async def fetch_data(self): ...
新語法
通過異步迭代器實現(xiàn)的一個新的迭代語法如下:
async for TARGET in ITER: BLOCK else: BLOCK2
這在語義上等同于:
iter = (ITER) iter = type(iter).__aiter__(iter) running = True while running: try: TARGET = await type(iter).__anext__(iter) except StopAsyncIteration: running = False else: BLOCK else: BLOCK2
把一個沒有__aiter__方法的迭代對象傳遞給 async for將引起TypeError。如果在async def函數(shù)外面使用async with,將引起一個SyntaxError(語法錯誤)。
和常規(guī)的for表達式一樣, async for也有一個可選的else 分句。.
例子1
使用異步迭代器能夠在迭代過程中異步地緩存數(shù)據(jù):
async for data in cursor: ...
這里的cursor是一個異步迭代器,能夠從一個數(shù)據(jù)庫中每經(jīng)過N次迭代預(yù)取N行數(shù)據(jù)。
下面的語法展示了這種新的異步迭代協(xié)議的用法:
class Cursor: def __init__(self): self.buffer = collections.deque() async def _prefetch(self): ... def __aiter__(self): return self async def __anext__(self): if not self.buffer: self.buffer = await self._prefetch() if not self.buffer: raise StopAsyncIteration return self.buffer.popleft()
接下來這個Cursor 類可以這樣使用:
async for row in Cursor(): print(row) which would be equivalent to the following code: i = Cursor().__aiter__() while True: try: row = await i.__anext__() except StopAsyncIteration: break else: print(row)
例子2
下面的代碼可以將常規(guī)的迭代對象變成異步迭代對象。盡管這不是一個非常有用的東西,但這段代碼說明了常規(guī)迭代器和異步迭代器之間的關(guān)系。
class AsyncIteratorWrapper: def __init__(self, obj): self._it = iter(obj) def __aiter__(self): return self async def __anext__(self): try: value = next(self._it) except StopIteration: raise StopAsyncIteration return value async for letter in AsyncIteratorWrapper("abc"): print(letter)
為什么要拋出StopAsyncIteration?
協(xié)程(Coroutines)內(nèi)部仍然是基于生成器的。因此在PEP 479之前,下面兩種寫法沒有本質(zhì)的區(qū)別:
def g1(): yield from fut return 'spam'
和
def g2(): yield from fut raise StopIteration('spam')
自從 PEP 479 得到接受并成為協(xié)程 的默認實現(xiàn),下面這個例子將StopIteration包裝成一個RuntimeError。
async def a1(): await fut raise StopIteration('spam')
告知外圍代碼迭代已經(jīng)結(jié)束的唯一方法就是拋出StopIteration。因此加入了一個新的異常類StopAsyncIteration。
由PEP 479的規(guī)定 , 所有協(xié)程中拋出的StopIteration異常都被包裝在RuntimeError中。
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
使用Python編程分析火爆全網(wǎng)的魷魚游戲豆瓣影評
本文來為大家介紹如何使用Python爬取影評的操作,主要是爬取《魷魚游戲》在豆瓣上的一些影評,對數(shù)據(jù)做一些簡單的分析,用數(shù)據(jù)的角度重新審視下這部劇,有需要的朋友可以借鑒參考下2021-10-10python用線性回歸預(yù)測股票價格的實現(xiàn)代碼
這篇文章主要介紹了python用線性回歸預(yù)測股票價格的實現(xiàn)代碼,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2019-09-0920行Python代碼實現(xiàn)一款永久免費PDF編輯工具
本文主要介紹了Python代碼實現(xiàn)一款永久免費PDF編輯工具,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2022-07-07Python爬蟲實戰(zhàn)案例之爬取喜馬拉雅音頻數(shù)據(jù)詳解
這篇文章主要介紹了Python爬蟲實戰(zhàn)案例之取喜馬拉雅音頻數(shù)據(jù)詳解,本文通過實例代碼給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-12-12python使用 multiprocessing 多進程處理批量數(shù)據(jù)的示例代碼
這篇文章主要介紹了使用 multiprocessing 多進程處理批量數(shù)據(jù)的示例代碼,本文通過示例代碼給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2023-09-09