���Z� � @ sn d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d d l
m Z d d l m
Z
m Z m Z m Z m Z d d l m Z d' Z d( Z Gd d � d e � Z d d � Z d d � Z Gd d
� d
� Z Gd d � d � Z Gd d � d e � Z Gd d � d e � Z Gd d
� d
e � Z Gd! d � d e � Z Gd" d � d e � Z Gd# d � d e � Z! Gd$ d � d e � Z" Gd% d � d e � Z# Gd& d � d e � Z$ e � Z% e% j& e e' e( e) f � e% j& e e* � e% j& e e j+ � e% j& e! e j, � e% j& e" e j- � e% j& e# e j. e j/ f � e% j& e e j0 � d S))� N)�ABC�abstractmethod)� CIMultiDict� )�hdrs)�PY_36�content_disposition_header�guess_filename�parse_mimetype�sentinel)�
DEFAULT_LIMIT�PAYLOAD_REGISTRY� get_payload� payload_type�Payload� BytesPayload�
StringPayload�
IOBasePayload�BytesIOPayload�BufferedReaderPayload�
TextIOPayload�StringIOPayload� JsonPayload� � c @ s e Z d Z d S)� LookupErrorN)�__name__�
__module__� __qualname__� r r � /payload.pyr s r c O s t j | | | � S)N)r
�get)�data�args�kwargsr r r r s c C s t j | | � d S)N)r
�register)�factory�typer r r �register_payload s r( c @ s( e Z d Z d d � Z d d � Z d S)r c C s
| | _ d S)N)r' )�selfr' r r r �__init__& s zpayload_type.__init__c C s t | | j � | S)N)r( r' )r) r& r r r �__call__) s zpayload_type.__call__N)r r r r* r+ r r r r r $ s c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �PayloadRegistryzZPayload registry.
note: we need zope.interface for more efficient adapter search
c C s
g | _ d S)N)� _registry)r) r r r r* 4 s zPayloadRegistry.__init__c O sY t | t � r | Sx6 | j D]+ \ } } t | | � r | | | | � Sq Wt � � d S)N)�
isinstancer r- r )r) r" r# r$ r& r' r r r r! 7 s zPayloadRegistry.getc C s | j j | | f � d S)N)r- �append)r) r& r' r r r r% @ s zPayloadRegistry.registerN)r r r �__doc__r* r! r% r r r r r, . s r, c
@ s� e Z d Z d Z d Z d Z d d d e d d d d d d � Z e d d
� � Z e d d � � Z
e d
d � � Z e d d � � Z e d d � � Z
d d d � Z e d d � � Z d S)r Nzapplication/octet-stream�headers� content_type�filename�encodingc K s� | | _ | | _ | | _ | d k rd t | � | _ | t k rd t j | j k rd | j t j } | t k rv d } | | _ d S)N) �_value� _encoding� _filenamer �_headersr r Z CONTENT_TYPE�
_content_type)r) �valuer1 r2 r3 r4 r$ r r r r* J s zPayload.__init__c C s | j S)zSize of the payload.)�_size)r) r r r �sizeY s z Payload.sizec C s | j S)zFilename of the payload.)r7 )r) r r r r3 ^ s zPayload.filenamec C s | j S)zCustom item headers)r8 )r) r r r r1 c s zPayload.headersc C s | j S)zPayload encoding)r6 )r) r r r r4 h s zPayload.encodingc C sZ | j d k r | j S| j d k rO t j | j � d } | d k rK d S| St j Sd S)z Content typeNr zapplication/octet-stream)r9 r7 � mimetypesZ
guess_typer )r) Zmimer r r r2 m s zPayload.content_typeTc K s> | j d k r t � | _ t | d | | �| j t j <d S)z$Sets ``Content-Disposition`` header.N� quote_fields)r8 r r r ZCONTENT_DISPOSITION)r) Zdisptyper> Zparamsr r r �set_content_dispositionx s z Payload.set_content_dispositionc � s d S)zLWrite payload.
writer is an AbstractStreamWriter instance:
Nr )r) �writerr r r �write� s z
Payload.write)r r r r; r8 r9 r r* �propertyr< r3 r1 r4 r2 r? r rA r r r r r D s c s. e Z d Z � f d d � Z d d � Z � S)r c s| d | k r d | d <t � j | | | � t | � | _ | j t k rx t r_ d | i } n i } t j d t | � d S)Nr2 zapplication/octet-stream�sourcez}Sending a large body directly with raw bytes might lock the event loop. You should probably pass an io.BytesIO object instead) �superr* �lenr; �TOO_LARGE_BYTES_BODYr �warnings�warn�ResourceWarning)r) r: r# r$ )� __class__r r r* � s
zBytesPayload.__init__c � s | j | j � Id Hd S)N)rA r5 )r) r@ r r r rA � s zBytesPayload.write)r r r r* rA r r )rJ r r � s c s. e Z d Z d d d d � f d d � Z � S)r r4 Nr2 c s� | d k rK | d k r' d } d } qa t | � } | j j d d � } n | d k ra d | } t � j | j | � | d | d | | �d S)Nzutf-8ztext/plain; charset=utf-8�charsetztext/plain; charset=%sr4 r2 )r
�
parametersr! rD r* �encode)r) r: r4 r2 r# r$ �mimetype)rJ r r r* � s
zStringPayload.__init__)r r r r* r r )rJ r r � s c s" e Z d Z � f d d � Z � S)r c s t � j | j � | | � d S)N)rD r* �read)r) r: r# r$ )rJ r r r* � s zStringIOPayload.__init__)r r r r* r r )rJ r r � s c s1 e Z d Z d � f d d � Z d d � Z � S)r Z
attachmentc sg d | k r t | � | d <t � j | | | � | j d k rc | d k rc | j | d | j �d S)Nr3 )r rD r* r7 r? )r) r: Z dispositionr# r$ )rJ r r r* � s
zIOBasePayload.__init__c
� s\ zG | j j t � } x. | rE | j | � Id H| j j t � } q WWd | j j � Xd S)N)r5 rO r rA �close)r) r@ �chunkr r r rA � s zIOBasePayload.write)r r r r* rA r r )rJ r r � s c sL e Z d Z d d d d � f d d � Z e d d � � Z d d � Z � S)
r r4 Nr2 c s� | d k rK | d k r' d } d } qa t | � } | j j d d � } n | d k ra d | } t � j | | d | d | | �d S)Nzutf-8ztext/plain; charset=utf-8rK ztext/plain; charset=%sr2 r4 )r
rL r! rD r* )r) r: r4 r2 r# r$ rN )rJ r r r* � s
zTextIOPayload.__init__c C sG y* t j | j j � � j | j j � SWn t k
rB d SYn Xd S)N)�os�fstatr5 �fileno�st_size�tell�OSError)r) r r r r< � s *
zTextIOPayload.sizec
� sh zS | j j t � } x: | rQ | j | j | j � � Id H| j j t � } q WWd | j j � Xd S)N)r5 rO r rA rM r6 rP )r) r@ rQ r r r rA � s zTextIOPayload.write)r r r r* rB r< rA r r )rJ r r � s c @ s" e Z d Z e d d � � Z d S)r c C s? | j j � } | j j d t j � } | j j | � | | S)Nr )r5 rV �seekrR �SEEK_END)r) Zposition�endr r r r< � s zBytesIOPayload.sizeN)r r r rB r< r r r r r � s c @ s" e Z d Z e d d � � Z d S)r c C sG y* t j | j j � � j | j j � SWn t k
rB d SYn Xd S)N)rR rS r5 rT rU rV rW )r) r r r r< s *
zBufferedReaderPayload.sizeN)r r r rB r< r r r r r s c s. e Z d Z d d e j � f d d � Z � S)r zutf-8zapplication/jsonc s5 t � j | | � j | � | d | d | | �d S)Nr2 r4 )rD r* rM )r) r: r4 r2 �dumpsr# r$ )rJ r r r* s zJsonPayload.__init__)r r r �jsonr[ r* r r )rJ r r s ) r
r r r r r r r r r r r i )1�ior\ r= rR rG �abcr r Z multidictr � r Zhelpersr r r r
r Zstreamsr �__all__rF � Exceptionr r r( r r, r r r r r r r r r r
r% �bytes� bytearray�
memoryview�str�StringIO�
TextIOBase�BytesIO�BufferedReader�BufferedRandom�IOBaser r r r �<module> sJ (
D%