U
    M8c                     @   s  d Z 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 ddlm	Z	 ddl
mZ ddlmZmZmZmZ dZdZd	ZdddZG dd dZG dd deZG dd deZG dd deZG dd deZG dd deeZG dd deZeeeeedZdS )a  Protocol input serializes.

This module contains classes that implement input serialization
for the various AWS protocol types.

These classes essentially take user input, a model object that
represents what the expected input should look like, and it returns
a dictionary that contains the various parts of a request.  A few
high level design decisions:


* Each protocol type maps to a separate class, all inherit from
  ``Serializer``.
* The return value for ``serialize_to_request`` (the main entry
  point) returns a dictionary that represents a request.  This
  will have keys like ``url_path``, ``query_string``, etc.  This
  is done so that it's a) easy to test and b) not tied to a
  particular HTTP library.  See the ``serialize_to_request`` docstring
  for more details.

Unicode
-------

The input to the serializers should be text (str/unicode), not bytes,
with the exception of blob types.  Those are assumed to be binary,
and if a str/unicode type is passed in, it will be encoded as utf-8.
    N)ElementTree)validate)
formatdate)
has_headeris_json_value_headerparse_to_aware_datetimepercent_encodeiso8601z%Y-%m-%dT%H:%M:%SZz%Y-%m-%dT%H:%M:%S.%fZTc                 C   s&   t |   }|r"t }t||}|S N)SERIALIZERSr   ZParamValidatorZParamValidationDecorator)Zprotocol_nameZinclude_validation
serializerZ	validator r   6/tmp/pip-unpacked-wheel-ozje0y8b/botocore/serialize.pycreate_serializer?   s
    
r   c                   @   sb   e Zd ZdZeZdZdd Zdd Zdd Z	d	d
 Z
dd ZdddZdd Zdd Zdd ZdS )
SerializerPOSTzutf-8c                 C   s   t ddS )a  Serialize parameters into an HTTP request.

        This method takes user provided parameters and a shape
        model and serializes the parameters to an HTTP request.
        More specifically, this method returns information about
        parts of the HTTP request, it does not enforce a particular
        interface or standard for an HTTP request.  It instead returns
        a dictionary of:

            * 'url_path'
            * 'host_prefix'
            * 'query_string'
            * 'headers'
            * 'body'
            * 'method'

        It is then up to consumers to decide how to map this to a Request
        object of their HTTP library of choice.  Below is an example
        return value::

            {'body': {'Action': 'OperationName',
                      'Bar': 'val2',
                      'Foo': 'val1',
                      'Version': '2014-01-01'},
             'headers': {},
             'method': 'POST',
             'query_string': '',
             'host_prefix': 'value.',
             'url_path': '/'}

        :param parameters: The dictionary input parameters for the
            operation (i.e the user input).
        :param operation_model: The OperationModel object that describes
            the operation.
        serialize_to_requestNNotImplementedError)self
parametersoperation_modelr   r   r   r   Q   s    $zSerializer.serialize_to_requestc                 C   s   dd| j i dd}|S )N/     )url_pathquery_stringmethodheadersbody)DEFAULT_METHOD)r   
serializedr   r   r   _create_default_requestw   s    z"Serializer._create_default_requestc                 C   s   |j dkrt}nt}||S )Nr   )microsecondISO8601_MICROISO8601strftime)r   valuetimestamp_formatr   r   r   _timestamp_iso8601   s    
zSerializer._timestamp_iso8601c                 C   s   t t| S r
   )intcalendartimegm	timetupler   r'   r   r   r   _timestamp_unixtimestamp   s    z#Serializer._timestamp_unixtimestampc                 C   s"   t |tjr| |}t|ddS )NT)usegmt)
isinstancedatetimer/   r   r.   r   r   r   _timestamp_rfc822   s    
zSerializer._timestamp_rfc822Nc                 C   s:   |d kr| j }| }t|}t| d| }||}|S )NZ_timestamp_)TIMESTAMP_FORMATlowerr   getattr)r   r'   r(   datetime_obj	converterZfinal_valuer   r   r   _convert_timestamp_to_str   s    z$Serializer._convert_timestamp_to_strc                 C   s   |j d|S Nnameserializationget)r   shapedefault_namer   r   r   _get_serialized_name   s    zSerializer._get_serialized_namec                 C   s,   t |tr|| j}t| | jS r
   )r1   strencodeDEFAULT_ENCODINGbase64	b64encodestripdecoder.   r   r   r   _get_base64   s    
zSerializer._get_base64c                    sZ   |j }|d ksd|krd S |d }|jj}dd | D } fdd|D }|jf |S )NZ
hostPrefixc                 S   s    g | ]\}}|j d r|qS )Z	hostLabelr<   ).0memberr?   r   r   r   
<listcomp>   s   z2Serializer._expand_host_prefix.<locals>.<listcomp>c                    s   i | ]}| | qS r   r   )rJ   r;   r   r   r   
<dictcomp>   s      z2Serializer._expand_host_prefix.<locals>.<dictcomp>)Zendpointinput_shapemembersitemsformat)r   r   r   Zoperation_endpointZhost_prefix_expressionZinput_membersZhost_labelsZformat_kwargsr   rM   r   _expand_host_prefix   s    zSerializer._expand_host_prefix)N)__name__
__module____qualname__r    dictMAP_TYPErD   r   r"   r)   r/   r3   r9   rA   rI   rS   r   r   r   r   r   H   s   &
	r   c                   @   sp   e Zd ZdZdd ZdddZdddZdd	d
ZdddZdddZ	dddZ
dddZdddZdd ZdS ) QuerySerializerr	   c                 C   s   |j }|  }|jd| j|d< ddi|d< |  }|j|d< |jd |d< |d k	rd| ||| ||d< | 	||}|d k	r||d	< |S )
Nr   Content-Typez0application/x-www-form-urlencoded; charset=utf-8r   ActionZ
apiVersionVersionr   host_prefix)
rO   r"   httpr>   r    rX   r;   metadata
_serializerS   )r   r   r   r?   r!   body_paramsr]   r   r   r   r      s&      
z$QuerySerializer.serialize_to_requestr   c                 C   s*   t | d|j | j}|||||d d S )NZ_serialize_type_)prefixr6   	type_name_default_serialize)r   r!   r'   r?   rb   r   r   r   r   r`      s    
zQuerySerializer._serializec           	      C   sR   |j }| D ]>\}}|| }| ||}|r<| d| }| |||| qd S )N.)rP   rQ   rA   r`   )	r   r!   r'   r?   rb   rP   keymember_shapeZmember_prefixr   r   r   _serialize_type_structure   s    z)QuerySerializer._serialize_type_structurec                 C   s   |sd||< d S |  |r\|}|jjdrz| j|jdd}d|dd d |g }n|jjdd}| d| }t|dD ],\}}	| d| }
|j}| ||	||
 qd S )Nr   r;   r@   rf   rK      )	_is_shape_flattenedrK   r=   r>   rA   joinsplit	enumerater`   )r   r!   r'   r?   rb   Zlist_prefixr;   Z	list_nameielementelement_prefixelement_shaper   r   r   _serialize_type_list   s    
 z$QuerySerializer._serialize_type_listc                 C   s   |  |r|}nd| }|d }|j}|j}| j|dd}	| |d}
t|dD ]H\}}|j||	d}|j||
d}| |||| | ||| || qPd S )Nz%s.entryz.{i}.{suffix}rg   rj   r'   rl   )rq   suffix)rm   rg   r'   rA   rp   rR   r`   )r   r!   r'   r?   rb   Zfull_prefixtemplateZ	key_shapeZvalue_shapeZ
key_suffixZvalue_suffixrq   rg   Z
key_prefixZvalue_prefixr   r   r   _serialize_type_map  s    
z#QuerySerializer._serialize_type_mapc                 C   s   |  |||< d S r
   rI   r   r!   r'   r?   rb   r   r   r   _serialize_type_blob  s    z$QuerySerializer._serialize_type_blobc                 C   s   |  ||jd||< d S NtimestampFormatr9   r=   r>   rz   r   r   r   _serialize_type_timestamp  s     
z)QuerySerializer._serialize_type_timestampc                 C   s   |rd||< nd||< d S Ntruefalser   rz   r   r   r   _serialize_type_boolean  s    
z'QuerySerializer._serialize_type_booleanc                 C   s   |||< d S r
   r   rz   r   r   r   re   %  s    z"QuerySerializer._default_serializec                 C   s   |j dS )N	flattenedr<   r   r?   r   r   r   rm   (  s    z#QuerySerializer._is_shape_flattenedN)r   )r   )r   )r   )r   )r   )r   )r   )rT   rU   rV   r4   r   r`   ri   ru   rx   r{   r   r   re   rm   r   r   r   r   rY      s   

	





rY   c                   @   s"   e Zd ZdZdd ZdddZdS )	EC2SerializeraU  EC2 specific customizations to the query protocol serializers.

    The EC2 model is almost, but not exactly, similar to the query protocol
    serializer.  This class encapsulates those differences.  The model
    will have be marked with a ``protocol`` of ``ec2``, so you don't need
    to worry about wiring this class up correctly.

    c                 C   sH   d|j kr|j d S d|j kr@|j d }|d  |dd   S |S d S )NZ	queryNamer;   r   rl   )r=   upper)r   r?   r@   r;   r   r   r   rA   6  s    



z"EC2Serializer._get_serialized_namer   c           	      C   s<   t |dD ],\}}| d| }|j}| |||| q
d S )Nrl   rf   )rp   rK   r`   )	r   r!   r'   r?   rb   rq   rr   rs   rt   r   r   r   ru   C  s    z"EC2Serializer._serialize_type_listN)r   )rT   rU   rV   __doc__rA   ru   r   r   r   r   r   ,  s   	r   c                   @   sR   e Zd ZdZdd ZdddZdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )JSONSerializerZunixtimestampc           	      C   s   d |jd |j}|jd }|  }|jd| j|d< |d| d|d< |  }|j}|d k	rp| 	||| t
|| j|d< | ||}|d k	r||d	< |S )
Nz{}.{}ZtargetPrefixZjsonVersionr   zapplication/x-amz-json-%s)zX-Amz-TargetrZ   r   r   r]   )rR   r_   r;   r"   r^   r>   r    rX   rO   r`   jsondumpsrC   rD   rS   )	r   r   r   targetZjson_versionr!   r   rO   r]   r   r   r   r   M  s,    
 
z#JSONSerializer.serialize_to_requestNc                 C   s&   t | d|j | j}||||| d S Nz_serialize_type_%src   )r   r!   r'   r?   rg   r   r   r   r   r`   g  s    zJSONSerializer._serializec           
      C   st   |j r|||< n`|d k	r,|  }|||< |}|j}| D ]4\}}|| }	d|	jkr^|	jd }| |||	| q:d S r:   )Zis_document_typerX   rP   rQ   r=   r`   )
r   r!   r'   r?   rg   Znew_serializedrP   Z
member_keymember_valuerh   r   r   r   ri   o  s"    


   z(JSONSerializer._serialize_type_structurec                 C   s8   |   }|||< | D ]\}}| |||j| qd S r
   )rX   rQ   r`   r'   )r   r!   r'   r?   rg   Zmap_objZsub_keyZ	sub_valuer   r   r   rx     s    z"JSONSerializer._serialize_type_mapc                 C   s>   g }|||< |D ](}i }|  |||jd ||d  qd S )NZ__current__)r`   rK   append)r   r!   r'   r?   rg   list_objZ	list_itemwrapperr   r   r   ru     s    z#JSONSerializer._serialize_type_listc                 C   s   |||< d S r
   r   r   r!   r'   r?   rg   r   r   r   re     s    z!JSONSerializer._default_serializec                 C   s   |  ||jd||< d S r|   r~   r   r   r   r   r     s     
z(JSONSerializer._serialize_type_timestampc                 C   s   |  |||< d S r
   ry   r   r   r   r   r{     s    z#JSONSerializer._serialize_type_blob)N)rT   rU   rV   r4   r   r`   ri   rx   ru   re   r   r{   r   r   r   r   r   J  s   
r   c                   @   s   e Zd ZdZdZdZddddgZdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd S )!BaseRestSerializera=  Base class for rest protocols.

    The only variance between the various rest protocols is the
    way that the body is serialized.  All other aspects (headers, uri, etc.)
    are the same and logic for serializing those aspects lives here.

    Subclasses must implement the ``_serialize_body_params`` method.

    r	   Zrfc822uriquerystringheaderr   c           
      C   s,  |   }|jd| j|d< |j}|d kr<|jd |d< |S |j}|  |  |  |  d}| D ]"\}}|d krzqh| |||| qh| 	|jd |d |d< d|jkr| 	|jd |d |d< |d |d	< |d
 r|d
 |d
< | 
||||| | ||| | ||}	|	d k	r(|	|d< |S )Nr   Z
requestUrir   )uri_path_kwargsquery_string_kwargsbody_kwargsr   r   ZauthPathZ	auth_pathr   r   r   r]   )r"   r^   r>   r    rO   rP   rX   rQ   _partition_parameters_render_uri_template_serialize_payload_serialize_content_typerS   )
r   r   r   r!   r?   shape_memberspartitioned
param_nameparam_valuer]   r   r   r   r     s`         
    
z'BaseRestSerializer.serialize_to_requestc                 C   sZ   i }t d|D ]<}|dr<t||d d  dd||< qt|| ||< q|jf |S )Nz{(.*?)}+rk   z/~)safe)refindallendswithr   rR   )r   Zuri_templateparamsZencoded_paramsZtemplate_paramr   r   r   r     s    
 
z'BaseRestSerializer._render_uri_templatec           	      C   s   |j d}| ||r8||d}| |}||d< nr|d k	rv||}|d k	rh| ||| |d< q|  |d< n4|d r| |d ||d< n| |r|  |d< d S )Npayloadr   r   r   )r=   r>   _has_streaming_payload_encode_payload_serialize_body_params_serialize_empty_body_requires_empty_body)	r   r   r   r!   r?   r   Zpayload_memberZbody_payloadra   r   r   r   r     s(    


 
 

z%BaseRestSerializer._serialize_payloadc                 C   s   dS )Nr   r   r   r   r   r   r   #  s    z(BaseRestSerializer._serialize_empty_bodyc                 C   s   dS )z
        Some protocols require varied Content-Type headers
        depending on user input. This allows subclasses to apply
        this conditionally.
        Nr   )r   r!   r?   r   r   r   r   r   &  s    z*BaseRestSerializer._serialize_content_typec                 C   s   dS )z
        Some protocols require a specific body to represent an empty
        payload. This allows subclasses to apply this conditionally.
        Fr   r   r   r   r   r   .  s    z'BaseRestSerializer._requires_empty_bodyc                 C   s   |dk	o|| j dkS )z5Determine if payload is streaming (a blob or string).N)Zblobstring)rd   )r   r   r   r   r   r   r   5  s    z)BaseRestSerializer._has_streaming_payloadc                 C   s   t |tr|| jS |S r
   )r1   rB   rC   rD   )r   r   r   r   r   r   <  s    
z"BaseRestSerializer._encode_payloadc                 C   s4  || }|j d}|j d|}|dkr8||d |< n|dkrt|trZ|d | ndt|tr~t| }||d |< n@|jdkr|j d| j	}	| 
||	}
|
|d |< n||d |< np|d	kr|| }|s|jd
krd S | ||}t||d |< n.|dkr$|}| ||d | n||d |< d S )Nlocationr;   r   r   r   r   	timestampr}   r   listr   r   )r=   r>   r1   rW   updateboolrB   r5   rd   QUERY_STRING_TIMESTAMP_FORMATr9   _convert_header_value_do_serialize_header_map)r   r   r   r   r   rK   r   key_nameZbool_strr(   r   r?   r'   header_prefixr   r   r   r   A  sH    


  

  z(BaseRestSerializer._partition_parametersc                 C   s&   |  D ]\}}|| }|||< qd S r
   )rQ   )r   r   r   Z
user_inputrg   valZfull_keyr   r   r   r   r  s    z+BaseRestSerializer._do_serialize_header_mapc                 C   s   t dd S )Nr   r   )r   r   r?   r   r   r   r   w  s    z)BaseRestSerializer._serialize_body_paramsc                    s   j dkr<t|}t| }jd j} ||S j dkrd fdd|D }d	|S t
r tj|ddS |S d S )	Nr   r}   r   c                    s"   g | ]}|d k	r  j|qS r
   )r   rK   )rJ   vr   r   r   rL     s   z<BaseRestSerializer._convert_header_value.<locals>.<listcomp>,)r   :)
separators)rd   r   r+   r,   utctimetupler=   r>   HEADER_TIMESTAMP_FORMATr9   rn   r   rI   r   r   )r   r?   r'   r7   r   r(   Zconverted_valuer   r   r   r   z  s     
 

z(BaseRestSerializer._convert_header_valueN)rT   rU   rV   r   r   r   ZKNOWN_LOCATIONSr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s    
; 1r   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
RestJSONSerializerc                 C   s   dS )Ns   {}r   r   r   r   r   r     s    z(RestJSONSerializer._serialize_empty_bodyc                 C   s(   |j  D ]\}}d|jkr
 dS q
dS )zq
        Serialize an empty JSON object whenever the shape has
        members not targeting a location.
        r   TF)rP   rQ   r=   )r   r?   rK   r   r   r   r   r     s    
z'RestJSONSerializer._requires_empty_bodyc                 C   sN   |j d}| ||rdS |d dk}td|d }|rJ|sJd|d d< dS )z?Set Content-Type to application/json for all structured bodies.r   Nr   r   rZ   r   zapplication/json)r=   r>   r   r   )r   r!   r?   r   r   Zhas_bodyZhas_content_typer   r   r   r     s    z*RestJSONSerializer._serialize_content_typec                 C   s(   |   }| ||| t|| jS r
   )rX   r`   r   r   rC   rD   )r   r   r?   Zserialized_bodyr   r   r   r     s    z)RestJSONSerializer._serialize_body_paramsN)rT   rU   rV   r   r   r   r   r   r   r   r   r     s   
r   c                   @   sX   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd ZdS )RestXMLSerializerr	   c                 C   s@   |j d }td}| |||| t|d }tj|| jdS )Nr;   r   r   )encoding)r=   r   Elementr`   r   tostringrD   )r   r   r?   	root_nameZpseudo_rootZ	real_rootr   r   r   r     s
    

z(RestXMLSerializer._serialize_body_paramsc                 C   s&   t | d|j | j}||||| d S r   rc   )r   r?   r   xmlnoder;   r   r   r   r   r`     s    zRestXMLSerializer._serializec                 C   s   t ||}d|jkrL|jd }d}|dr>|d|d  7 }|d |j|< | D ]`\}}	|j| }
|
jd|}|	d kr d S |
jdr|
jd }|	|j|< qT| |
|	|| qTd S )NZxmlNamespacexmlnsrb   z:%sr   r;   ZxmlAttribute)r   
SubElementr=   r>   attribrQ   rP   r`   )r   r   r   r?   r;   Zstructure_nodeZnamespace_metadataZattribute_namerg   r'   rh   member_nameZxml_attribute_namer   r   r   ri     s"    





z+RestXMLSerializer._serialize_type_structurec           	      C   sT   |j }|jdr|}|}n|jdd}t||}|D ]}| |||| q:d S )Nr   r;   rK   )rK   r=   r>   r   r   r`   )	r   r   r   r?   r;   rh   Zelement_nameZ	list_nodeitemr   r   r   ru     s    z&RestXMLSerializer._serialize_type_listc                 C   sr   t ||}| D ]X\}}t |d}| j|jdd}	| j|jdd}
| |j|||	 | |j|||
 qd S )Nentryrg   rj   r'   )r   r   rQ   rA   rg   r'   r`   )r   r   r   r?   r;   noderg   r'   Z
entry_noder   Zval_namer   r   r   rx     s    	 z%RestXMLSerializer._serialize_type_mapc                 C   s$   t ||}|rd}nd}||_d S r   )r   r   text)r   r   r   r?   r;   r   Z	str_valuer   r   r   r     s
    z)RestXMLSerializer._serialize_type_booleanc                 C   s   t ||}| ||_d S r
   )r   r   rI   r   r   r   r   r?   r;   r   r   r   r   r{     s    z&RestXMLSerializer._serialize_type_blobc                 C   s&   t ||}| ||jd|_d S r|   )r   r   r9   r=   r>   r   r   r   r   r   r     s
     
z+RestXMLSerializer._serialize_type_timestampc                 C   s   t ||}t||_d S r
   )r   r   rB   r   r   r   r   r   re     s    z$RestXMLSerializer._default_serializeN)rT   rU   rV   r4   r   r`   ri   ru   rx   r   r{   r   re   r   r   r   r   r     s   r   )Zec2queryr   z	rest-jsonzrest-xml)T)r   rE   r+   r2   r   r   Z	xml.etreer   Zbotocorer   Zbotocore.compatr   Zbotocore.utilsr   r   r   r   ZDEFAULT_TIMESTAMP_FORMATr%   r$   r   r   rY   r   r   r   r   r   r   r   r   r   r   <module>   s6   
	wmY o f