���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_type�streamerc @ s( e Z d Z d d � Z d d � Z d S)�_stream_wrapperc C s( t j | � | _ | | _ | | _ d S)N)�asyncio� coroutine�coro�args�kwargs)�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