3
\�- � @ sd d Z ddlZddlZddlZddlZddlmZ ddlm Z ddl
T ddl m Z G dd� de �Z
dS ) zhdistutils.command.register
Implements the Distutils 'register' command (register with the repository).
� 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� dS )zDeprecated API.zddistutils.command.register.check_metadata is deprecated, use the check command insteadr r N)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 )
z: Reads the configuration file and set attributes.
�username�password�
repository�realmTZpypiz%s not found in .pypircFN)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|�� dS )z8 Fetch the list of classifiers from the server.
z?:action=list_classifiersN)r �urllib�requestZurlopenr �info�_read_pypi_response)r ZurlZresponser
r
r r U s
zregister.classifiersc C s& | j | jd��\}}tjd||� dS )zF Send the metadata to the package index server to be checked.
r zServer response (%s): %sN)�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 ||� dS )$a_ Send the metadata to the package index server.
Well, do the following:
1. figure who the user is, and then
2. send the data as a Basic auth'ed POST.
First we try to read the username/password from $HOME/.pypirc,
which is a ConfigParser-formatted file with a section
[distutils] containing username and password entries (both
in clear text). Eg:
[distutils]
index-servers =
pypi
[pypi]
username: fred
password: sekrit
Otherwise, to figure who the user is, we offer the user three
choices:
1. use existing login,
2. register as a new user, or
3. set the password to a random string and email the user.
�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 ZemailNZconfirmz
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 dd}
~
X nJ t
jjk
�r� }
zdt|
�f} W Y dd}
~
X nX | j�r�| j| �}d} | j�r d
j dd |dd f�}| j |tj� | S )zC Post a query to the server, and return a string response.
r6 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/ Ni� 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 )�__doc__r= rU Z urllib.parser% Zurllib.request�warningsr Zdistutils.corer Zdistutils.errorsZ distutilsr r r
r
r
r � s