U
    ªÃT_�  ã                   @   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mZ ddlmZmZ ddlmZ d	d
lmZ dd„ ZG dd„ deƒZdS )zpupload_docs

Implements a Distutils 'upload_docs' subcommand (upload documentation to
PyPI's pythonhosted.org).
é    )Ústandard_b64encode)Úlog)ÚDistutilsOptionErrorN)Úsix)Úhttp_clientÚurllib)Úiter_entry_pointsé   )Úuploadc                 C   s   t jr
dnd}|  d|¡S )NÚsurrogateescapeÚstrictzutf-8)r   ÚPY3Úencode)ÚsÚerrors© r   úB/tmp/pip-unpacked-wheel-sr6cxcdi/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 PyPIz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"   | j d krtddƒD ]} dS d S )Nzdistutils.commandsÚbuild_sphinxT)Ú
upload_dirr   )ÚselfÚepr   r   r   Ú
has_sphinx/   s    
zupload_docs.has_sphinxr   c                 C   s   t  | ¡ d | _d | _d S )N)r
   Úinitialize_optionsr   Ú
target_dir)r   r   r   r   r   6   s    
zupload_docs.initialize_optionsc                 C   sˆ   t  | ¡ | jd krN|  ¡ r0|  d¡}|j| _q`|  d¡}tj 	|j
d¡| _n|  d¡ | j| _d| jkrtt d¡ |  d| j ¡ d S )Nr   ÚbuildZdocsr   zpypi.python.orgz3Upload_docs command is deprecated. Use RTD instead.zUsing upload directory %s)r
   Úfinalize_optionsr   r   Úget_finalized_commandZbuilder_target_dirr   ÚosÚpathÚjoinÚ
build_baseÚensure_dirnameÚ
repositoryr   ÚwarnÚ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ÚtmplÚnameÚfullÚrelativeÚdestr   r   r   Úcreate_zipfileK   s    zupload_docs.create_zipfilec              	   C   sh   |   ¡ D ]}|  |¡ qt ¡ }| jj ¡ }tj 	|d| ¡}z|  |¡ |  |¡ W 5 t
 |¡ X d S )Nz%s.zip)Úget_sub_commandsÚrun_commandÚtempfileÚmkdtempÚdistributionÚmetadataÚget_namer    r!   r"   ÚshutilÚrmtreer<   Úupload_file)r   Úcmd_nameZtmp_dirr8   r3   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_parti   s    



zupload_docs._build_partc                 C   sn   d}d| }|d }|df}t j| j|d�}t|| ¡ ƒ}tj |¡}t ||¡}	d| d¡ }
d 	|	¡|
fS )	z=
        Build up the MIME payload for the POST data
        s3   --------------GHSKFJDLGDS7543FJKLFHRE75642756743254s   
--s   --rK   )rP   z multipart/form-data; boundary=%sÚasciió    )
Ú	functoolsÚpartialrU   ÚmapÚitemsÚ	itertoolsÚchainÚfrom_iterableÚdecoder"   )Ú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   sJ  t |dƒ�}| ¡ }W 5 Q R X | jj}d| ¡ tj |¡|fdœ}t| j	d | j
 ƒ}t|ƒ}tjrn| d¡}d| }|  |¡\}}	d| j }
|  |
tj¡ tj | j¡\}}}}}}|sÂ|sÂ|rÆt‚|dkrÚt |¡}n |d	krît |¡}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�rÌd|j&|j'f }
|  |
tj¡ nb|j&dk�r| (d¡}|d k�rød| ¡  }d| }
|  |
tj¡ nd|j&|j'f }
|  |
tj$¡ | j)�rFt*d| ¡ dƒ d S )NÚrbZ
doc_upload)z:actionr8   Ú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ÚreadrA   rB   rC   r    r!   Úbasenamer   ÚusernameÚpasswordr   r   r   r_   re   r%   r'   r   ÚINFOr   ÚparseÚurlparseÚAssertionErrorr   ÚHTTPConnectionÚHTTPSConnectionÚconnectÚ
putrequestÚ	putheaderÚstrr.   Ú
endheadersÚsendÚsocketÚerrorÚERRORÚgetresponseÚstatusÚreasonÚ	getheaderÚshow_responseÚprint)r   r2   Úfrg   Úmetara   ÚcredentialsÚauthÚbodyÚctÚmsgZschemaÚnetlocÚurlÚparamsÚqueryZ	fragmentsÚconnrd   Úer   Úlocationr   r   r   rF   �   sd    ý

ÿ

zupload_docs.upload_fileN)Ú__name__Ú
__module__Ú__qualname__ÚDEFAULT_REPOSITORYÚdescriptionr
   Úuser_optionsÚboolean_optionsr   Úsub_commandsr   r   r<   rH   ÚstaticmethodrU   Úclassmethodre   rF   r   r   r   r   r      s(   ÿû


r   )Ú__doc__Úbase64r   Ú	distutilsr   Údistutils.errorsr   r    r€   r)   r?   rD   r\   rX   Zsetuptools.externr   Zsetuptools.extern.six.movesr   r   Úpkg_resourcesr   r
   r   r   r   r   r   r   Ú<module>   s    