Z @ s d Z d d l Z d d l m Z m Z d
Z Gd d d Z Gd d d Z e e Gd d
d
e Z e e Gd d d e Z d S)al Payload implemenation for coroutines as data provider.
As a simple case, you can upload data from file::
@aiohttp.streamer
async def file_sender(writer, file_name=None):
with open(file_name, 'rb') as f:
chunk = f.read(2**16)
while chunk:
await writer.write(chunk)
chunk = f.read(2**16)
Then you can use `file_sender` like this:
async with session.post('http://httpbin.org/post',
data=file_sender(file_name='huge_file')) as resp:
print(await resp.text())
..note:: Coroutine must accept `writer` as first argument
N )Payload payload_typestreamerc @ s( e Z d Z d d Z d d Z d S)_stream_wrapperc C s( t j | | _ | | _ | | _ d S)N)asyncio coroutinecoroargskwargs)selfr r
r r
/payload_streamer.py__init__" s z_stream_wrapper.__init__c s" | j | | j | j Id Hd S)N)r r
r )r writerr
r
r __call__' s z_stream_wrapper.__call__N)__name__
__module__ __qualname__r r r
r
r
r r s r c @ s( e Z d Z d d Z d d Z d S)r c C s
| | _ d S)N)r )r r r
r
r r - s zstreamer.__init__c O s t | j | | S)N)r r )r r
r r
r
r r 0 s zstreamer.__call__N)r r r r r r
r
r
r r + s c @ s e Z d Z d d Z d S)StreamWrapperPayloadc s | j | Id Hd S)N)_value)r r r
r
r write7 s zStreamWrapperPayload.writeN)r r r r r
r
r
r r 4 s r c s. e Z d Z f d d Z d d Z S)
StreamPayloadc s t j | | | d S)N)superr )r valuer
r ) __class__r
r r > s zStreamPayload.__init__c s | j | Id Hd S)N)r )r r r
r
r r A s zStreamPayload.write)r r r r r r
r
)r r r ; s r )r )
__doc__r Zpayloadr r __all__r r r r r
r
r
r <module> s