CasperSecurity
o
�_�a2 � @ s� d Z ddlmZ ddlmZ ddlmZ ddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlmZ ddlmZ d d
� ZG d d � d e�ZdS )
z|upload_docs
Implements a Distutils 'upload_docs' subcommand (upload documentation to
sites other than PyPi such as devpi).
� )�standard_b64encode)�log)�DistutilsOptionErrorN)�iter_entry_points� )�uploadc C s | � dd�S )Nzutf-8�surrogateescape)�encode)�s� r �@/usr/lib/python3/dist-packages/setuptools/command/upload_docs.py�_encode s r
c @ s� e Zd ZdZdZdddej fddgZejZdd � Zd
efgZ d d � Z
d
d� Z dd� Z dd� Z
edd� �Zedd� �Zdd� ZdS )� 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=Nzdirectory to uploadc C s& | j d u rtdd�D ]} dS d S d S )Nzdistutils.commands� build_sphinxT)�
upload_dirr )�self�epr r r �
has_sphinx- s
�zupload_docs.has_sphinxr c C s t �| � d | _d | _d S )N)r �initialize_optionsr �
target_dir)r r r r r 4 s
z upload_docs.initialize_optionsc C s� t �| � | jd u r+| �� r | �d�}t|j�d | _n| �d�}tj �
|j d�| _n | � d� | j| _d| j
v r>t�d� | �d| j � d S ) Nr �html�build�docsr zpypi.python.orgz<Upload_docs command is deprecated for PyPi. Use RTD instead.zUsing upload directory %s)r �finalize_optionsr r �get_finalized_command�dict�builder_target_dirsr �os�path�join�
build_base�ensure_dirname�
repositoryr �warn�announce)r r r r r r r 9 s
z upload_docs.finalize_optionsc C s� t �|d�}zP| �| j� t�| j�D ]<\}}}|| jkr(|s(d}t|| j ��|D ]$}tj�||�}|t | j�d � �
tjj �} tj�| |�}
|� ||
� q*qW |�
� d S |�
� w )N�wz'no files found in upload directory '%s')�zipfile�ZipFile�mkpathr r �walkr r r �len�lstrip�sep�write�close) r �filename�zip_file�root�dirs�files�tmpl�name�full�relative�destr r r �create_zipfileI s ��
zupload_docs.create_zipfilec C sp | � � D ]}| �|� qt�� }| jj�� }tj� |d| �}z| �
|� | � |� W t �
|� d S t �
|� w )Nz%s.zip)�get_sub_commands� run_command�tempfile�mkdtemp� distribution�metadata�get_namer r r r: � upload_file�shutil�rmtree)r �cmd_name�tmp_dirr6 r1 r r r �runY s
zupload_docs.runc c s� � | \}}d| }t |t�s|g}|D ]3}t |t�r'|d|d 7 }|d }nt|�}|V t|�V dV |V |rF|dd � dkrFdV qd 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_partg s$ �
��zupload_docs._build_partc C sn d}d|� d� }|d }|df}tj| j|d�}t||�� �}tj�|�}t�||�} d| }
d� | �|
fS ) z=
Build up the MIME payload for the POST data
z3--------------GHSKFJDLGDS7543FJKLFHRE75642756743254s
--�asciis --rJ )rO z multipart/form-data; boundary=%s� )
r � functools�partialrT �map�items� itertools�chain�
from_iterabler ) �cls�data�boundaryrO � end_boundary� end_items�builder� part_groups�parts�
body_items� content_typer r r �_build_multipart{ s � z upload_docs._build_multipartc
C sT t |d�� }|�� }W d � n1 sw Y | jj}d|�� tj�|�|fd�}t| j d | j
�}t |�� d�}d| }| �
|�\}} d| j }
| �|
tj� tj�| j�\} } }
}}}|sf|sf|rhJ �| dkrstj�| �}n| d kr~tj�| �}ntd
| ��d }z-|�� |�d |
� | }|� d
|� |� dt t |��� |� d|� |� � |� |� W n t!j"y� } z| �t |�tj#� W Y d }~d S d }~ww |�$� }|j%dkr�d|j%|j&f }
| �|
tj� n0|j%dk�r |�'d�}|d u r�d|�� }d| }
| �|
tj� nd|j%|j&f }
| �|
tj#� | j(�r(t)d|�� d� d S d S )N�rb�
doc_upload)z:actionr6 �content�:rU zBasic z Submitting documentation to %s�http�httpszunsupported schema � �POSTz Content-typezContent-length�
Authorization�� zServer response (%s): %si- �Locationz https://pythonhosted.org/%s/zUpload successful. Visit %szUpload failed (%s): %szK---------------------------------------------------------------------------)*�open�readr? r@ rA r r �basenamer
�username�passwordr �decoderh r# r% r �INFO�urllib�parse�urlparserm �client�HTTPConnection�HTTPSConnection�AssertionError�connect�
putrequest� putheader�strr+ �
endheaders�send�socket�error�ERROR� getresponse�status�reason� getheader�
show_response�print)r r0 �frk �metar_ � credentials�auth�body�ct�msg�schema�netloc�url�params�query� fragments�connrg �er �locationr r r rB � sh
��
� ��
�zupload_docs.upload_fileN)�__name__�
__module__� __qualname__�DEFAULT_REPOSITORY� descriptionr � user_options�boolean_optionsr � sub_commandsr r r: rG � staticmethodrT � classmethodrh rB r r r r r s* ��
r )�__doc__�base64r � distutilsr �distutils.errorsr r r� r' r= rC r[ rW � http.clientrm � urllib.parser{ �
pkg_resourcesr r r
r r r r r �<module> s"