@@ -130,23 +130,6 @@ def feed():
130130 yield event
131131
132132
133- class aclosing (contextlib .AbstractAsyncContextManager ):
134- """Async context manager for safely finalizing an asynchronously cleaned-up
135- resource such as an async generator, calling its ``aclose()`` method.
136-
137- This is a copy of contextlib.aclosing needed for python 3.9 compatibility
138- """
139-
140- def __init__ (self , thing ):
141- self .thing = thing
142-
143- async def __aenter__ (self ):
144- return self .thing
145-
146- async def __aexit__ (self , * exc_info ):
147- await self .thing .aclose ()
148-
149-
150133def Selector (fds : Collection [FDLike ], events = selectors .EVENT_READ ) -> selectors .DefaultSelector :
151134 """A selectors.DefaultSelector with given fds registered"""
152135 selector = selectors .DefaultSelector ()
@@ -183,7 +166,7 @@ def event_stream(fds: Collection[FDLike], read: Callable[[FDLike], T], timeout:
183166
184167async def async_selector_stream (selector : selectors .BaseSelector ) -> AsyncIterator [SelectorEvent ]:
185168 """An asyncronous infinite stream of selector read events"""
186- async with aclosing (astream (selector , selector .select )) as stream :
169+ async with contextlib . aclosing (astream (selector , selector .select )) as stream :
187170 async for events in stream :
188171 for event in events :
189172 yield event
@@ -192,22 +175,22 @@ async def async_selector_stream(selector: selectors.BaseSelector) -> AsyncIterat
192175async def async_selector_file_stream (fds : Collection [FDLike ]) -> AsyncIterator [SelectorEvent ]:
193176 """An asyncronous infinite stream of selector read events"""
194177 selector = Selector (fds )
195- async with aclosing (async_selector_stream (selector )) as stream :
178+ async with contextlib . aclosing (async_selector_stream (selector )) as stream :
196179 async for event in stream :
197180 yield event
198181
199182
200183async def async_file_stream (fds : Collection [FDLike ]) -> AsyncIterator [FDLike ]:
201184 """An asyncronous infinite stream of read ready files"""
202- async with aclosing (async_selector_file_stream (fds )) as stream :
185+ async with contextlib . aclosing (async_selector_file_stream (fds )) as stream :
203186 async for key , _ in stream :
204187 yield key .fileobj
205188
206189
207190async def async_event_stream (fds : Collection [FDLike ], read : Callable [[FDLike ], T ]):
208191 """An asyncronous stream of events. The given read callable is called for each file
209192 that is reported as ready"""
210- async with aclosing (async_file_stream (fds )) as stream :
193+ async with contextlib . aclosing (async_file_stream (fds )) as stream :
211194 async for fd in stream :
212195 yield read (fd )
213196
0 commit comments