
    g                         d Z ddlZddlZddlmZmZmZmZmZ ddl	m
Z
 ddlmZmZ dZ G d d	      Z G d
 d      Z ee       G d de             Z ee       G d de             Zy)am  
Payload implementation 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)Any	AwaitableCallableDictTuple   )AbstractStreamWriter)Payloadpayload_type)streamerc                   X    e Zd Zdeded   f   deedf   deeef   ddfdZ	de
ddfd	Zy)
_stream_wrappercoro.Nargskwargsreturnc                 T    t        j                  |      | _        || _        || _        y N)types	coroutiner   r   r   )selfr   r   r   s       M/var/www/openai/venv/lib/python3.12/site-packages/aiohttp/payload_streamer.py__init__z_stream_wrapper.__init__$   s"     OOD)		    writerc                 t   K    | j                   |g| j                  i | j                   d {    y 7 wr   )r   r   r   r   r   s     r   __call__z_stream_wrapper.__call__.   s,     dii::dkk:::s   .868)__name__
__module____qualname__r   r   r   r   r   strr   r	   r    r   r   r   r   #   s_    sIdO+, CHo S#X	
 
;%9 ;d ;r   r   c                   @    e Zd Zdeded   f   ddfdZdededefdZy)	r   r   .Nr   c                 J    t        j                  dt        d       || _        y )Nz5@streamer is deprecated, use async generators instead   )
stacklevel)warningswarnDeprecationWarningr   )r   r   s     r   r   zstreamer.__init__3   s     C	

 	r   r   r   c                 0    t        | j                  ||      S r   )r   r   )r   r   r   s      r   r   zstreamer.__call__;   s    tyy$77r   )	r   r    r!   r   r   r   r   r   r   r#   r   r   r   r   2   s=    Xc9T?&:;  8c 8S 8_ 8r   r   c                   2    e Zd ZdeddfdZddededefdZy)	StreamWrapperPayloadr   r   Nc                 B   K   | j                  |       d {    y 7 wr   _valuer   s     r   writezStreamWrapperPayload.writeA        kk&!!!   encodingerrorsc                     t        d      )NzUnable to decode.)	TypeError)r   r4   r5   s      r   decodezStreamWrapperPayload.decodeD   s    +,,r   )zutf-8strict)r   r    r!   r	   r1   r"   r8   r#   r   r   r-   r-   ?   s0    ""6 "4 "-s -c - -r   r-   c                   @     e Zd Zdedededdf fdZdeddfdZ xZS )	StreamPayloadvaluer   r   r   Nc                 6    t        |    |       g|i | y r   )superr   )r   r<   r   r   	__class__s       r   r   zStreamPayload.__init__J   s    24262r   r   c                 B   K   | j                  |       d {    y 7 wr   r/   r   s     r   r1   zStreamPayload.writeM   r2   r3   )r   r    r!   r   r   r	   r1   __classcell__)r?   s   @r   r;   r;   H   s8    3c 3# 3 3 3""6 "4 "r   r;   )__doc__r   r(   typingr   r   r   r   r   abcr	   payloadr
   r   __all__r   r   r-   r;   r#   r   r   <module>rG      sx   0   8 8 % *
; ;
8 
8 o-7 - - h"( " "r   