B
� f�- � @ 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 ZdZejddg Zejdddg Zddd � fgZd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d d �Zd S ) �registerz7register the distribution with the Python package index)zlist-classifiersNz list the valid Trove classifiers)�strictNzBWill stop the registering if the meta-data are not fully compliant�verifyzlist-classifiersr �checkc C s dS )NT� )�selfr
r
�?/opt/alt/python37/lib64/python3.7/distutils/command/register.py� � zregister.c C s t �| � d| _d| _d S )Nr )r �initialize_options�list_classifiersr )r r
r
r r s
zregister.initialize_optionsc C s* t �| � d| jfdd�}|| jjd< d S )Nr )r � )r �restructuredtextr )r �finalize_optionsr � distributionZcommand_options)r Z
check_optionsr
r
r r $ s
zregister.finalize_optionsc C sX | � � | �� x| �� D ]}| �|� qW | jr<| �� n| jrL| �� n| �� 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� | j�d�}|�� | j|_d|_|�� 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 | � � }|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 Zconfigr
r
r r D s
zregister._set_configc C s* | j d }tj�|�}t�| �|�� 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& | � | �d��\}}t�d||� 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 }}d�� }x:||krf| �dtj� t� }|sTd}q.||kr.td� q.W |dk�r|x|s�td�}qtW x|s�t � d�}q�W t
j � � }t
j
�| j�d }|�| j|||� | �| �d
�|�\}}| �d ||f tj� |d k�r | j �r|| j_nj| �d
tj� | �d| �� tj� d}x&|�� dk�r\td�}|�s8d}�q8W |�� dk�r | �||� �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 � d�| d< �q�W x | d �s t � d �| d< �qW | d | d k�r�d| d< d| d< td � �q�W x | d �sltd �| d< �qRW | �| �\}}|d k�r�t�d ||� nt�d � t�d � nT|d!k�r dd"i} d| d< x | d �s�td#�| d< �q�W | �| �\}}t�d ||� 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�authZhostr( r) �datar
r
r r c s�
zregister.send_metadatac C s� | j j}|d|�� |�� |�� |�� |�� |�� |�� |� � |�
� |� � |� � |�
� |�� |�� |�� d�}|d s�|d s�|d r�d|d< |S )Nz1.0)z:action�metadata_versionr3 �versionZsummaryZ home_pageZauthorZ author_email�license� description�keywords�platformr Z download_url�provides�requires� obsoletesrF rG rH z1.1r@ )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�metar? r
r
r r' � s* zregister.build_post_dataNc
C s� d|kr$| � d|d | jf tj� d}d| }|d }t�� }x�|�� D ]�\}}t|�tg �td�fkrp|g}xZ|D ]R}t|�}|� |� |� d| � |� d� |� |� |rv|d d
krv|� d � qvW qJW |� |� |� d � |�
� � d �}d
| tt |��d�} t
j�| j|| �}
t
j�t
jj|d��} d}y| �|
�} W nx t
jjk
�r� }
z"| j�rl|
j�� }|
j|
jf} W dd}
~
X Y nJ t
jjk
�r� }
zdt|
�f} W dd}
~
X Y nX | j�r�| �| �}d} | j�r�d � d|df�}| � |tj� | S )zC Post a query to the server, and return a string response.
r3 zRegistering %s to %sz3--------------GHSKFJDLGDS7543FJKLFHRE75642756743254z
--z--r
z*
Content-Disposition: form-data; name="%s"z
������
�
zutf-8z/multipart/form-data; boundary=%s; charset=utf-8)z Content-typezContent-length)Z password_mgrr, Ni� )r- ZOKzK---------------------------------------------------------------------------) r6 r r r7 �io�StringIO�items�type�str�write�getvalue�encode�lenr" r# ZRequestZ build_openerZHTTPBasicAuthHandler�open�errorZ HTTPErrorZ
show_response�fp�readr( �msgZURLErrorr% �join)r r? r> �boundaryZ sep_boundaryZ end_boundaryZbody�key�valueZheadersZreq�openerr) �er[ r
r
r r& � sV
zregister.post_to_server)N)�__name__�
__module__� __qualname__rC 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: rN Z urllib.parser" Zurllib.request�warningsr Zdistutils.corer Zdistutils.errorsZ distutilsr r r
r
r
r � s