3
9��f� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlmZ ddlmZmZ ddlmZ d d
lmZ d d � ZG d
d� de�ZdS )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 jr
dnd}| jd|�S )N�surrogateescape�strictzutf-8)r �PY3�encode)�s�errors� r �!/usr/lib/python3.6/upload_docs.py�_encode s r c @ s� e Zd ZdZdZdddej fd d gZejZd d � Zd
efgZ dd� Z
dd� Z dd� Z dd� Z
edd� �Zedd� �Zdd� ZdS ) � upload_docsz https://pypi.python.org/pypi/z Upload documentation to PyPIz repository=�rz url of repository [default: %s]�
show-responseN�&display full response text from server� upload-dir=�directory to uploadc C s$ | j d kr xtdd�D ]}dS W d S )Nzdistutils.commands� build_sphinxT)�
upload_dirr )�selfZepr r r �
has_sphinx/ s
zupload_docs.has_sphinxr c C s t j| � 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| � | jd krN| j� r0| jd�}|j| _q`| jd�}tjj |j
d�| _n| j d� | j| _d| j krtt
jd� | jd| j � d S )Nr �buildZdocsr zpypi.python.orgz3Upload_docs command is deprecated. Use RTD instead.zUsing upload directory %s)r
�finalize_optionsr r Zget_finalized_commandZbuilder_target_dirr �os�path�joinZ
build_baseZensure_dirname�
repositoryr �warn�announce)r r r r r r r! ; s
z upload_docs.finalize_optionsc C s� t j|d�}z�| j| j� x�tj| j�D ]~\}}}|| jkrT| rTd}t|| j ��xP|D ]H}tjj||�}|t | j�d � j
tjj �} tjj| |�}
|j ||
� qZW q(W W d |j
� X d S )N�wz'no files found in upload directory '%s')�zipfileZZipFileZmkpathr r" �walkr r# r$ �len�lstrip�sep�write�close) r �filename�zip_file�root�dirs�filesZtmpl�nameZfullZrelative�destr r r �create_zipfileK s
zupload_docs.create_zipfilec C sl x| j � D ]}| j|� q
W tj� }| jjj� }tjj |d| �}z| j
|� | j |� W d t j
|� X d S )Nz%s.zip)Zget_sub_commandsZ run_command�tempfileZmkdtemp� distribution�metadata�get_namer" r# r$ r7 � upload_file�shutilZrmtree)r Zcmd_nameZtmp_dirr5 r1 r r r �run[ s
zupload_docs.runc c s� | \}}d| }t |t�s |g}xn|D ]f}t |t�rN|d|d 7 }|d }nt|�}|V t|�V dV |V |r&|dd � dkr&dV q&W 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
zupload_docs._build_partc C sn d}d| }|d }|df}t j| j|d�}t||j� �}tjj|�}tj||�} d|jd� }
dj | �|
fS ) z=
Build up the MIME payload for the POST data
s3 --------------GHSKFJDLGDS7543FJKLFHRE75642756743254s
--s --r@ )rF z multipart/form-data; boundary=%s�ascii� )
� functools�partialrK �map�items� itertools�chain�
from_iterable�decoder$ ) �cls�data�boundaryrF Z end_boundaryZ end_itemsZbuilderZ part_groups�partsZ
body_items� content_typer r r �_build_multipart} s z upload_docs._build_multipartc C sP t |d��}|j� }W d Q R X | jj}d|j� tjj|�|fd�}t| j d | j
�}t |�}t j
rn|jd�}d| }| j|�\}} d| j }
| j|
tj� tjj| j�\} } }
}}}| r�| r�| s�t�| dkr�tj| �}n | d kr�tj| �}n td
| ��d }yZ|j� |j d |
� | }|j d
|� |j dt t |��� |j d|� |j � |j!|� W n6 t"j#k
�r� } z| jt |�tj$� d S d }~X nX |j%� }|j&dk�r�d|j&|j'f }
| j|
tj� nb|j&dk�r|j(d�}|d k�r�d|j� }d| }
| j|
tj� n d|j&|j'f }
| j|
tj$� | j)�rLt*dd |j� dd � d S )N�rbZ
doc_upload)z:actionr5 �content�:rL zBasic z Submitting documentation to %sZhttpZhttpszunsupported schema � ZPOSTz Content-typezContent-lengthZ
Authorization�� zServer response (%s): %si- ZLocationz https://pythonhosted.org/%s/zUpload successful. Visit %szUpload failed (%s): %s�-�K )+�open�readr9 r: r; r" r# �basenamer ZusernameZpasswordr r r
rU r[ r% r' r �INFOr �parseZurlparse�AssertionErrorr ZHTTPConnectionZHTTPSConnectionZconnectZ
putrequestZ putheader�strr+ Z
endheaders�send�socket�errorZERRORZ getresponseZstatus�reasonZ getheaderZ
show_response�print)r r0 �fr] �metarW Z credentialsZauthZbodyZct�msgZschemaZnetlocZurlZparamsZqueryZ fragmentsZconnrZ �er �locationr r r r< � s`
zupload_docs.upload_file)r Nr )r Nr )�__name__�
__module__� __qualname__ZDEFAULT_REPOSITORY� descriptionr
Z user_optionsZboolean_optionsr Z sub_commandsr r! r7 r> � staticmethodrK � classmethodr[ r< r r r r r s"
r )�__doc__�base64r Z distutilsr Zdistutils.errorsr r" rk r) r8 r= rR rN Zsetuptools.externr Zsetuptools.extern.six.movesr r Z
pkg_resourcesr r
r r r r r r �<module> s