3
�gaa � @ sx d dl Z d dlmZmZmZmZmZmZmZ ddl m
Z
ddl m Z m
Z
mZmZmZmZmZmZmZmZ ddlmZmZmZmZmZm Z ddl m Z dd l m Z m!Z! G d
d � d �Z"ee#e$e%e&e'f Z(e
ee# d �d
d�Z)e
e#d �dd�Z*e
e#d �dd�Z+e
ee#ee#f d �dd�Z,e
e#d �dd�Z-e
e#d �dd�Z.e
e(d �dd�Z/e
e(d �dd �Z0e
e#d �d d �Z1e
e#e(d �d!d"�Z2dS )#� N)�Callable�Dict�IO�List�Optional�Tuple�Union� )� KeyValueArg)
�SEPARATORS_GROUP_MULTIPART�"SEPARATOR_DATA_EMBED_FILE_CONTENTS�"SEPARATOR_DATA_EMBED_RAW_JSON_FILE�SEPARATOR_DATA_RAW_JSON�SEPARATOR_DATA_STRING�SEPARATOR_FILE_UPLOAD�SEPARATOR_FILE_UPLOAD_TYPE�SEPARATOR_HEADER�SEPARATOR_HEADER_EMPTY�SEPARATOR_QUERY_PARAM)�MultipartRequestDataDict�RequestDataDict�RequestFilesDict�RequestHeadersDict�RequestJSONDataDict�RequestQueryParamsDict)�
ParseError� )�get_content_type�&load_json_preserve_order_and_dupe_keysc @ s0 e Zd Zddd�Zed ee d d�dd��ZdS )
� RequestItemsFc C s6 t � | _|rt� nt� | _t� | _t� | _t � | _
d S )N) r �headersr r �datar �filesr �paramsr �multipart_data)�self�as_form� r' �"/usr/lib/python3.6/requestitems.py�__init__ s
zRequestItems.__init__)�request_item_args�returnc C s� | |d�}t t|jftt|jftt|jftt |j
ft t |j
ftt|j
ftt|j
ftt|j
fi}xB|D ]:}||j \}}||�}|||j< |jtkrd||j|j< qdW |S )N)r& )r �process_header_argr r �process_empty_header_argr �process_query_param_argr# r �process_file_upload_argr" r �process_data_item_argr! r �$process_data_embed_file_contents_argr � process_data_raw_json_embed_argr
�$process_data_embed_raw_json_file_arg�sep�keyr r$ ) �clsr* r& �instanceZrules�argZprocessor_func� target_dict�valuer' r' r( � from_args s@
zRequestItems.from_argsN)F)F)�__name__�
__module__� __qualname__r) � classmethodr r
r; r' r' r' r( r s
r )r8 r+ c C s
| j pd S )N)r: )r8 r' r' r( r, W s r, c C s" | j s | j S td| j�d���d S )Nz
Invalid item z+ (to specify an empty header use `Header;`))r: r �orig)r8 r' r' r( r- [ s r- c C s | j S )N)r: )r8 r' r' r( r. c s r. c C s� | j jt�}|d }t|�dkr(|d nd }yttjj|�d�}W n6 tk
rx } zt | j
�d|� ���W Y d d }~X nX tjj |�||p�t |�fS )Nr r �rbz: )
r: �splitr �len�open�os�path�
expanduser�OSErrorr r@ �basenamer )r8 �parts�filenameZ mime_type�f�er' r' r( r/ g s &
r/ c C s | j S )N)r: )r8 r' r' r( r0 v s r0 c C s t | �S )N)�load_text_file)r8 r' r' r( r1 z s r1 c C s t | �}t| |�}|S )N)rN � load_json)r8 �contentsr: r' r' r( r3 ~ s
r3 c C s t | | j�}|S )N)rO r: )r8 r: r' r' r( r2 � s r2 )�itemr+ c C s� | j }y*ttjj|�d��}|j� j� S Q R X W n` tk
rf } zt| j �d|� ���W Y d d }~X n, t
k
r� t| j �d| j �d���Y nX d S )NrA z: z : cannot embed the content of z(, not a UTF-8 or ASCII-encoded text file) r: rD rE rF rG �read�decoderH r r@ �UnicodeDecodeError)rQ rF rL rM r' r' r( rN � s $rN )r8 rP r+ c C sD yt |�S tk
r> } zt| j�d|� ���W Y d d }~X nX d S )Nz: )r �
ValueErrorr r@ )r8 rP rM r' r' r( rO � s rO )3rE Ztypingr r r r r r r Zargtypesr
Z constantsr r r
r r r r r r r Zdictsr r r r r r �
exceptionsr Zutilsr r r �str�bool�int�list�dictZJSONTyper, r- r. r/ r0 r1 r3 r2 rN rO r' r' r' r( �<module> s$ $ 0 ?| Name | Type | Size | Permission | Actions |
|---|---|---|---|---|
| __init__.cpython-36.opt-1.pyc | File | 113 B | 0644 |
|
| __init__.cpython-36.pyc | File | 113 B | 0644 |
|
| argparser.cpython-36.opt-1.pyc | File | 11.82 KB | 0644 |
|
| argparser.cpython-36.pyc | File | 11.86 KB | 0644 |
|
| argtypes.cpython-36.opt-1.pyc | File | 7.43 KB | 0644 |
|
| argtypes.cpython-36.pyc | File | 7.43 KB | 0644 |
|
| constants.cpython-36.opt-1.pyc | File | 2.02 KB | 0644 |
|
| constants.cpython-36.pyc | File | 2.02 KB | 0644 |
|
| definition.cpython-36.opt-1.pyc | File | 17.74 KB | 0644 |
|
| definition.cpython-36.pyc | File | 17.74 KB | 0644 |
|
| dicts.cpython-36.opt-1.pyc | File | 2.1 KB | 0644 |
|
| dicts.cpython-36.pyc | File | 2.13 KB | 0644 |
|
| exceptions.cpython-36.opt-1.pyc | File | 276 B | 0644 |
|
| exceptions.cpython-36.pyc | File | 276 B | 0644 |
|
| requestitems.cpython-36.opt-1.pyc | File | 4.26 KB | 0644 |
|
| requestitems.cpython-36.pyc | File | 4.26 KB | 0644 |
|