3
�ga�E � @ s d dl Z d dlZd dlZd dlZd dlZd dl mZ d dlmZ d dlm Z d dl
m Z ddl m
Z
mZmZmZmZ ddlmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z dd l!m"Z" dd
l#m$Z$ d d l%m&Z& d d
l'm(Z( d dl)m*Z*m+Z+ G dd� de�Z,G dd� de j-�Z.dS )� N)�RawDescriptionHelpFormatter)�dedent)�urlsplit)�get_netrc_auth� )�AuthCredentials�KeyValueArgType� PARSED_DEFAULT_FORMAT_OPTIONS�
parse_auth�parse_format_options)�HTTP_GET� HTTP_POST�OUTPUT_OPTIONS�OUTPUT_OPTIONS_DEFAULT� OUTPUT_OPTIONS_DEFAULT_OFFLINE�(OUTPUT_OPTIONS_DEFAULT_STDOUT_REDIRECTED�
OUT_RESP_BODY�
PRETTY_MAP�PRETTY_STDOUT_TTY_ONLY� RequestType�SEPARATOR_CREDENTIALS�SEPARATOR_GROUP_ALL_ITEMS�SEPARATOR_GROUP_DATA_ITEMS�
URL_SCHEME_RE)�
ParseError)� RequestItems� )� Environment)�plugin_manager)�ExplicitNullAuth�get_content_typec s* e Zd ZdZd� fdd� Zdd� Z� ZS )�HTTPieHelpFormatterz�A nicer help formatter.
Help for arguments can be indented and contain new lines.
It will be de-dented and arguments in the help
will be separated by a blank line for better readability.
� c s ||d< t � j||� d S )N�max_help_position)�super�__init__)�selfr# �args�kwargs)� __class__� � /usr/lib/python3.6/argparser.pyr% ) s z HTTPieHelpFormatter.__init__c C s t |�j� d }|j� S )Nz
)r �strip�
splitlines)r& �text�widthr* r* r+ � _split_lines. s z HTTPieHelpFormatter._split_lines)r" )�__name__�
__module__� __qualname__�__doc__r% r0 �
__classcell__r* r* )r) r+ r! s r! c s� e Zd ZdZed�� fdd�
Zd'eejd�� fdd�
Z d d
� Z
d d � Z d(� fd
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z� ZS ))�HTTPieArgumentParserz�Adds additional logic to `argparse.ArgumentParser`.
Handles all input (CLI args, file args, stdin), applies defaults,
and performs extra validation.
)�formatter_classc s: d|d< t � j|d|i|�� d | _d | _d| _d| _d S )NFZadd_helpr7 )r$ r% �envr' �has_stdin_data�has_input_data)r& r7 r' r( )r) r* r+ r% = s z HTTPieArgumentParser.__init__N)r8 �returnc s || _ t� j||�\| _}| jjr*d| j_| j joD| jj oD| j j | _ | j pX| jj
d k | _ | j |� | j
� | j� | j� | j� | j� | j� | j� | j� | j� | j� | jj
d k r�| j| jj
� n| j r�| j| j j� | jj�r| jj�r| jd� | jj �r| jd� | jS )NTz'cannot combine --compress and --chunkedz)cannot combine --compress and --multipart) r8 r$ Zparse_known_argsr' �debug� traceback�stdin� ignore_stdinZ stdin_isattyr9 �rawr: �_apply_no_options�_process_request_type�_process_download_options�_setup_standard_streams�_process_output_options�_process_pretty_options�_process_format_options�
_guess_method� _parse_items� _process_url�
_process_auth�_body_from_input�_body_from_file�compressZchunked�error� multipart)r& r8 r' � namespace�
no_options)r) r* r+ �
parse_argsF s:
z HTTPieArgumentParser.parse_argsc C s< | j j}|tjk| j _|tjk| j _|tjtjhk| j _d S )N) r' � request_typer ZJSONZjsonZ MULTIPARTrP ZFORM�form)r& rT r* r* r+ rB q s z*HTTPieArgumentParser._process_request_typec C s� t j| jj�s�tjj| jj�dkr(d}n | jj d }t
jd| jj�}|r�|j d�}|j d�}|d | j_|r�| j jd| 7 _| j j|7 _n|| jj | j_d S ) NZhttpszhttps://z://z^:(?!:)(\d*)(/?.*)$r r Z localhost�:) r �matchr' �url�os�path�basenamer8 Z program_nameZdefault_scheme�re�group)r& �schemeZ shorthandZport�restr* r* r+ rJ z s
z!HTTPieArgumentParser._process_urlc s^ t j| jjt j| jjd | jjij||�}t|d� rLt|t�rL|j| jj �}t
� j ||� d S )N�buffer) �sys�stdoutr8 �stderr�get�hasattr�
isinstance�str�encodeZstdout_encodingr$ �_print_message)r& �message�file)r) r* r+ ri � s
z#HTTPieArgumentParser._print_messagec C s� t | jj�| j_| jjrR| jj r8| jj r8| jj| j_| jj| j_| jj | j_nn| jjr�| jjj
d� y| jjj � W n2 t k
r� } z|j
t
jkr�n� W Y dd}~X nX | jj| j_d| j_| jjr�| jj| j_| jjo�| jj s�| jj| j_dS )zX
Modify `env.stdout` and `env.stdout_isatty` based on args, if needed.
r NF)�boolr' � output_fileZoutput_file_specified�downloadr8 �
stdout_isattyrb rc Z
stderr_isatty�seek�truncate�OSError�errnoZEINVAL�quiet�devnull)r& �er* r* r+ rD � s( z,HTTPieArgumentParser._setup_standard_streamsc
C s� d | j _tj� d }| j jd k }t| j j�}| j jd krv| rv|jd k rv|j}|j pVd}t
||t t j ||g�d�| j _| j jd k s�|�r�| j js�|j| j _tj
| j j�� }| j j r�| j jd kr�|jr�t| j j�}|r�t
|d |d t t j |�d�| j _|j�r| j jd k�r| jd� | j j|_|| j _t| j jt
�}| j jd k�sP|j �r^|j� | j _nf|�rn| j j} n t| j j�} | j� �r�|j�r�| j j�r�| jd� | j|j� |j| j | j d�| j _| j j �r�| j j�r�t � | j _d S )Nr � )�key�value�sepZorigr z--auth requiredz=Unable to prompt for passwords because --ignore-stdin is set.)�username�password) r' Z auth_pluginr Zget_auth_pluginsZ auth_typer rX Zauthr{ r| r r �joinZget_auth_pluginZ ignore_netrcZ netrc_parser Z auth_requirerO Zraw_authrf Z
auth_parseZget_authr
Z has_passwordZprompt_passwordr? Znetlocrx ry r )
r& Zdefault_auth_pluginZ
auth_type_setrX r{ r| ZpluginZnetrc_credentialsZalready_parsedZ credentialsr* r* r+ rK � s`
z"HTTPieArgumentParser._process_authc C s� g }xj|D ]b}|j d�s$|j|� q
d|dd� }x6| jD ]"}||jkr<t| j|j|j� P q<W |j|� q
W |r�| jddj |�� �� dS )z�For every `--no-OPTION` in `no_options`, set `args.OPTION` to
its default value. This allows for un-setting of options, e.g.,
specified in config.
z--no-z--� Nzunrecognized arguments: � )
�
startswith�appendZ_actionsZoption_strings�setattrr' �dest�defaultrO r} )r& rR Zinvalid�option�inverted�actionr* r* r+ rA s
z&HTTPieArgumentParser._apply_no_optionsc C s( | j | jj| jj� t|d|�| j_dS )zPRead the data from a file-like object.
Bytes are always read.
r` N)�_ensure_one_data_sourcer' �data�files�getattr)r& �fdr* r* r+ rM ! s z$HTTPieArgumentParser._body_from_filec C s( | j | j| jj| jj� |j� | j_dS )z%Read the data from the CLI.
N)r� r9 r' r� r� rh )r& r� r* r* r+ rL * s
z%HTTPieArgumentParser._body_from_inputc G s t |�r| jd� dS )z=There can only be one source of input request data.
z�Request body (from stdin, --raw or a file) and request data (key=value) cannot be mixed. Pass --ignore-stdin to let key/value take priority. See https://httpie.io/docs#scripting for details.N)�anyrO )r& Z
other_sourcesr* r* r+ r� 2 s z,HTTPieArgumentParser._ensure_one_data_sourcec C s� | j jdkr&| jr t| j _q�t| j _n�tjd| j j�s�y"| j jjdt t
� j | j j �� W n> t
jk
r� } z | j jrv� | j|j d � W Y dd}~X n:X | j j| j _ | jp�tdd� | j jD ��}|r�tnt| j _dS )zxSet `args.method` if not specified to either POST or GET
based on whether the request has data or not.
Nz ^[a-zA-Z]+$r c s s | ]}|j tkV qd S )N)rz r )�.0�itemr* r* r+ � <genexpr>^ s z5HTTPieArgumentParser._guess_method.<locals>.<genexpr>)r' �methodr: r
r r\ rW �
request_items�insertr r �__call__rX �argparseZArgumentTypeErrorr= rO r� )r& rv Zhas_datar* r* r+ rH < s$
" z"HTTPieArgumentParser._guess_methodc
C s@ yt j| jj| jjd�}W n< tk
rV } z | jjr6� | j|jd � W Y dd}~X n4X |j| j_|j | j_ |j
| j_
|j | j_ |j | j_ | jj
o�| jj �r<d}xT| jj
j
� D ]D\}}|dkr�| jddj| jj
j� �� �� |dk r�| jd� |}q�W |\}}}i | j_
| j|� d| jjk�r<t|�} | �r<| | jjd< dS ) zx
Parse `args.request_items` into `args.headers`, `args.data`,
`args.params`, and `args.files`.
)Zrequest_item_argsZas_formr Nrw z1Invalid file fields (perhaps you meant --form?): �,z&Can't read request from multiple filesz Content-Type)r Z from_argsr' r� rU r r= rO Zheadersr� r� ZparamsZmultipart_data�itemsr} �keysrM r )
r& r� rv Z request_filerx rk �fnr� ZctZ content_typer* r* r+ rI c s: "
z!HTTPieArgumentParser._parse_itemsc s� � fdd�}� j jr d� j _� j jdkrl� j jr@djt�� j _n,� j jrRt� j _n� jj sdt
� j _nt � j _� j j dkr�� j j� j _ |� j jd� |� j j d� � j j
r�t� j jkr�tt� j j�tt� �� j _dS )z�Apply defaults to output options, or validate the provided ones.
The default output options are stdout-type-sensitive.
c s0 t | �t }|r,� jd|� ddj|�� �� d S )NzUnknown output options: �=r� )�setr rO r} )ry r� �unknown)r& r* r+ �
check_options� s zCHTTPieArgumentParser._process_output_options.<locals>.check_optionsTNrw z--printz--history-print)r' �verbose�allZoutput_optionsr} r �offliner r8 ro r r Zoutput_options_historyrn r rg r� )r& r� r* )r& r+ rE � s$
z,HTTPieArgumentParser._process_output_optionsc C s^ | j jtkr&t| jjrdnd | j _n4| j jrJ| jjrJ| j jrJ| jd� nt| j j | j _d S )Nr� Znonez1Only terminal output can be colorized on Windows.) r' Zprettifyr r r8 ro Z
is_windowsrm rO )r& r* r* r+ rF � s z,HTTPieArgumentParser._process_pretty_optionsc C s^ | j jr d| j _d| j _d S | j js6| j jr6| jd� | j jrZ| j joL| j j rZ| jd� d S )NFz%--continue only works with --downloadz,--continue requires --output to be specified)r' r� rn Zdownload_resumerO rm )r& r* r* r+ rC � s
z.HTTPieArgumentParser._process_download_optionsc C s6 | j jp
g }t}x|D ]}t||d�}qW || j _d S )N)Zdefaults)r' �format_optionsr r )r& r� Zparsed_optionsZ
options_groupr* r* r+ rG � s
z,HTTPieArgumentParser._process_format_options)NN)N)r1 r2 r3 r4 r! r% r r� Z NamespacerS rB rJ ri rD rK rA rM rL r� rH rI rE rF rC rG r5 r* r* )r) r+ r6 5 s( & (G
',% r6 )/r� rs rY r\ ra r �textwrapr Z urllib.parser Zrequests.utilsr Zargtypesr r r r
r Z constantsr r
r r r r r r r r r r r r �
exceptionsr Z requestitemsr �contextr Zplugins.registryr Zutilsr r r! �ArgumentParserr6 r* r* r* r+ �<module> s" @