3
\�- � @ s` d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl T d dl
m Z G dd� de�Z dS )� N)�warn)�
PyPIRCCommand)�*)�logc @ s� e Zd ZdZejd d g Zejdddg Zdd d
� fgZd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd!d d �ZdS )"�registerz7register the distribution with the Python package index�list-classifiersN� list the valid Trove classifiers�strict�BWill stop the registering if the meta-data are not fully compliant�verify�checkc C s dS )NT� )�selfr
r
�2/usr/lib64/python3.6/distutils/command/register.py� s zregister.c C s t j| � d| _d| _d S )Nr )r �initialize_options�list_classifiersr )r r
r
r r s
zregister.initialize_optionsc C s* t j| � d| jfdd�}|| jjd< d S )Nr � )r �restructuredtextr )r r )r �finalize_optionsr � distributionZcommand_options)r Z
check_optionsr
r
r r $ s
zregister.finalize_optionsc C sX | j � | j� x| j� D ]}| j|� qW | jr<| j� n| jrL| j� n| j� d S )N) r � _set_configZget_sub_commandsZ run_commandZdry_run�verify_metadatar � classifiers�
send_metadata)r Zcmd_namer
r
r �run+ s
z register.runc C s8 t dt� | jjd�}|j� | j|_d|_|j� d S )Nzddistutils.command.register.check_metadata is deprecated, use the check command insteadr r )r �PendingDeprecationWarningr Zget_command_objZensure_finalizedr r r )r r r
r
r �check_metadata: s zregister.check_metadatac C sz | j � }|i kr@|d | _|d | _|d | _|d | _d| _n6| jd| jfkr^td| j ��| jdkrp| j| _d| _d S ) N�username�password�
repository�realmTZpypiz%s not found in .pypircF)Z _read_pypircr r r r! �
has_configZDEFAULT_REPOSITORY�
ValueError)r �configr
r
r r D s
zregister._set_configc C s* | j d }tjj|�}tj| j|�� d S )Nz?:action=list_classifiers)r �urllib�requestZurlopenr �info�_read_pypi_response)r ZurlZresponser
r
r r U s
zregister.classifiersc C s& | j | jd��\}}tjd||� d S )Nr zServer response (%s): %s)�post_to_server�build_post_datar r' )r �code�resultr
r
r r \ s zregister.verify_metadatac
C s | j rd}| j}| j}n d}d }}dj� }x:||krf| jdtj� t� }|sTd}q.||kr.td� q.W |dk�r|x|s�td�}qtW x|s�t j d�}q�W t
j j � }t
j
j| j�d }|j| j|||� | j| jd
�|�\}}| jd ||f tj� |d k�r | j �r|| j_nj| jd
tj� | jd| j� tj� d}x&|j� dk�r\td�}|�s8d}�q8W |j� dk�r | j||� �n�|dk�r�ddi} d | d< | d< | d< d | d< x | d �s�td�| d< �q�W x�| d | d k�rNx | d �s�t j d�| d< �q�W x | d �s t j d�| d< �qW | d | d k�r�d| d< d | d< td � �q�W x | d �sltd �| d< �qRW | j| �\}}|d k�r�tjd ||� ntjd � tjd � nT|d k�r dd!i} d| d< x | d �s�td"�| d< �q�W | j| �\}}tjd ||� d S )#N�1�x� z1 2 3 4z�We need to know who you are, so please choose either:
1. use your existing login,
2. register as a new user,
3. have the server generate a new password for you (and email it to you), or
4. quit
Your selection [default 1]: z&Please choose one of the four options!z
Username: z
Password: r ZsubmitzServer response (%s): %s�� zAI can store your PyPI login so future submissions will be faster.z (the login will be stored in %s)�XZynzSave your login (y/N)?�n�y�2z:action�user�namer ZemailZconfirmz
Confirm: z!Password and confirm don't match!z
EMail: z"You will receive an email shortly.z7Follow the instructions in it to complete registration.�3Zpassword_resetzYour email address: )r" r r �split�announcer �INFO�input�print�getpassr% r& ZHTTPPasswordMgr�parseZurlparser Z add_passwordr! r) r* r Z _get_rc_file�lowerZ
_store_pypircr' )
r Zchoicer r �choices�auth�hostr+ r, �datar
r
r r c s�
zregister.send_metadatac C s� | j j}|d|j� |j� |j� |j� |j� |j� |j� |j � |j
� |j � |j � |j
� |j� |j� |j� d�}|d s�|d s�|d r�d|d< |S )Nz1.0)z:action�metadata_versionr6 �versionZsummaryZ home_pageZauthorZ author_email�license� description�keywords�platformr Z download_url�provides�requires� obsoletesrJ rK rL z1.1rD )r ZmetadataZget_nameZ get_versionZget_descriptionZget_urlZ get_contactZget_contact_emailZ get_licenceZget_long_descriptionZ get_keywordsZ
get_platformsZget_classifiersZget_download_urlZ get_providesZ get_requiresZ
get_obsoletes)r �action�metarC r
r
r r* � s* zregister.build_post_datac C s d|kr$| j d|d | jf tj� d}d| }|d }tj� }x�|j� D ]�\}}t|�tg �tf �fkrp|g}xZ|D ]R}t|�}|j |� |j d| � |j d� |j |� |rv|d d krv|j d
� qvW qJW |j |� |j d
� |j
� j d �}d | tt |��d
�} t
jj| j|| �}
t
jjt
jj|d��} d}y| j|
�} W nx t
jjk
�r� }
z$| j�rl|
jj� }|
j|
jf} W Y d d }
~
X nJ t
jjk
�r� }
zdt|
�f} W Y d d }
~
X nX | j�r�| j| �}d} | j�r d
j dd |dd f�}| j |tj� | S )Nr6 zRegistering %s to %sz3--------------GHSKFJDLGDS7543FJKLFHRE75642756743254z
--z--z*
Content-Disposition: form-data; name="%s"z
r �
�
zutf-8z/multipart/form-data; boundary=%s; charset=utf-8)z Content-typezContent-length)Z password_mgrr/ i� r0 �OK�-�K �����)r0 rQ ) r9 r r r: �io�StringIO�items�type�str�write�getvalue�encode�lenr% r& ZRequestZ build_openerZHTTPBasicAuthHandler�open�errorZ HTTPErrorZ
show_response�fp�readr+ �msgZURLErrorr( �join)r rC rA �boundaryZ sep_boundaryZ end_boundaryZbody�key�valueZheadersZreq�openerr, �erb r
r
r r) � sV
zregister.post_to_server)r Nr )r Nr
)N)�__name__�
__module__� __qualname__rG r Z user_optionsZboolean_optionsZ sub_commandsr r r r r r r r r* r) r
r
r
r r s&
z r )
r= rU Z urllib.parser% Zurllib.request�warningsr Zdistutils.corer Zdistutils.errorsZ distutilsr r r
r
r
r � s