U
    -c.                     @   s   d Z ddlmZ ddlmZ ddlmZ ddlZddlZddl	Z	ddl
Z
ddl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 ZG dd deZdS )z|upload_docs

Implements a Distutils 'upload_docs' subcommand (upload documentation to
sites other than PyPi such as devpi).
    )standard_b64encode)log)DistutilsOptionErrorN   )metadata)SetuptoolsDeprecationWarning   )uploadc                 C   s   |  ddS )Nzutf-8surrogateescape)encode)s r   B/tmp/pip-unpacked-wheel-izj_87as/setuptools/command/upload_docs.py_encode   s    r   c                   @   s   e Zd ZdZdZdddej fddgZejZdd	 Zd
efgZ	dd Z
dd Zdd Zdd Zedd Zedd Zdd ZdS )upload_docszhttps://pypi.python.org/pypi/z;Upload documentation to sites other than PyPi such as devpizrepository=rzurl of repository [default: %s])zshow-responseNz&display full response text from server)zupload-dir=Nzdirectory to uploadc                 C   s   t | jd kotjdddS )Nzdistutils.commandsbuild_sphinx)groupname)bool
upload_dirr   entry_pointsselfr   r   r   
has_sphinx/   s    
zupload_docs.has_sphinxr   c                 C   s   t |  d | _d | _d S )N)r	   initialize_optionsr   
target_dirr   r   r   r   r   7   s    
zupload_docs.initialize_optionsc                 C   s   t d t|  | jd kr`|  rB| d}t|jd | _	qr| d}t
j|jd| _	n| d | j| _	| d| j	  d S )NzWUpload_docs command is deprecated. Use Read the Docs (https://readthedocs.org) instead.r   htmlbuildZdocsr   zUsing upload directory %s)r   warnr	   finalize_optionsr   r   get_finalized_commanddictZbuilder_target_dirsr   ospathjoin
build_baseensure_dirnameannounce)r   r   r   r   r   r   r    <   s    




zupload_docs.finalize_optionsc                 C   s   t |d}z| | j t| jD ]x\}}}|| jkrP|sPd}t|| j |D ]H}tj	||}|t
| jd  tjj}	tj	|	|}
|||
 qTq&W 5 |  X d S )Nwz'no files found in upload directory '%s')zipfileZipFileclosemkpathr   r#   walkr   r$   r%   lenlstripsepwrite)r   filenamezip_filerootdirsfilestmplr   fullrelativedestr   r   r   create_zipfileM   s    zupload_docs.create_zipfilec              	   C   st   t dt |  D ]}| | qt }| jj	 }t
j|d| }z| | | | W 5 t| X d S )Nziupload_docs is deprecated and will be removed in a future version. Use tools like httpie or curl instead.z%s.zip)warningsr   r   get_sub_commandsrun_commandtempfilemkdtempdistributionr   get_namer#   r$   r%   shutilrmtreer<   upload_file)r   cmd_nameZtmp_dirr   r4   r   r   r   run]   s    
zupload_docs.runc                 c   s   | \}}d| }t |ts |g}|D ]f}t |trL|d|d  7 }|d }nt|}|V  t|V  dV  |V  |r$|dd  dkr$dV  q$d S )	Nz*
Content-Disposition: form-data; name="%s"z; filename="%s"r   r   s   

      
)
isinstancelisttupler   )itemsep_boundarykeyvaluestitlevaluer   r   r   _build_partq   s    



zupload_docs._build_partc                 C   sn   d}d| d }|d }|df}tj| j|d}t|| }tj|}t||}	d| }
d	|	|
fS )	z=
        Build up the MIME payload for the POST data
        z3--------------GHSKFJDLGDS7543FJKLFHRE75642756743254s   
--asciis   --rK   )rP   z multipart/form-data; boundary=%s    )
r   	functoolspartialrU   mapitems	itertoolschainfrom_iterabler%   )clsdataboundaryrP   Zend_boundaryZ	end_itemsZbuilderZpart_groupspartsZ
body_itemscontent_typer   r   r   _build_multipart   s    zupload_docs._build_multipartc              
   C   sD  t |d}| }W 5 Q R X | jj}d| tj||fd}t| j	d | j
 }t|d}d| }| |\}}	d| j }
| |
tj tj| j\}}}}}}|s|s|rt|dkrtj|}n"|d	krtj|}ntd
| d}zZ|  |d| |	}|d| |dtt| |d| |  | | W n> t!j"k
r } z| t|tj# W Y d S d }~X Y nX |$ }|j%dkrd|j%|j&f }
| |
tj nb|j%dkr
|'d}|d krd|  }d| }
| |
tj nd|j%|j&f }
| |
tj# | j(r@t)d| d d S )NrbZ
doc_upload)z:actionr   content:rV   zBasic zSubmitting documentation to %shttphttpszunsupported schema  POSTzContent-typezContent-lengthAuthorization   zServer response (%s): %si-  ZLocationzhttps://pythonhosted.org/%s/zUpload successful. Visit %szUpload failed (%s): %szK---------------------------------------------------------------------------)*openreadrB   r   rC   r#   r$   basenamer   usernamepasswordr   decoderd   
repositoryr(   r   INFOurllibparseurlparseAssertionErrorrh   clientHTTPConnectionHTTPSConnectionconnect
putrequest	putheaderstrr/   
endheaderssendsocketerrorERRORgetresponsestatusreason	getheadershow_responseprint)r   r3   frf   metar`   credentialsauthbodyctmsgZschemanetlocurlparamsqueryZ	fragmentsconnrc   er   locationr   r   r   rF      s`    


zupload_docs.upload_fileN)__name__
__module____qualname__DEFAULT_REPOSITORYdescriptionr	   user_optionsboolean_optionsr   sub_commandsr   r    r<   rH   staticmethodrU   classmethodrd   rF   r   r   r   r   r      s(   


r   )__doc__base64r   	distutilsr   distutils.errorsr   r#   r   r*   r@   rD   r\   rX   http.clientrh   urllib.parserv   r=   Z
_importlibr   rj   r   r	   r   r   r   r   r   r   <module>   s$   