o
    4^c-                     @   sd   d Z ddlZddlZddlZddlZddlmZ ddlm	Z	 ddl
T ddlmZ G dd de	ZdS )	zhdistutils.command.register

Implements the Distutils 'register' command (register with the repository).
    N)warn)PyPIRCCommand)*)logc                   @   s   e Zd ZdZejddg Zejg d Zddd fgZdd	 Zd
d Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd ZdddZdS )registerz7register the distribution with the Python package index)list-classifiersNz list the valid Trove classifiers)strictNzBWill stop the registering if the meta-data are not fully compliant)verifyr   r   checkc                 C   s   dS )NT selfr   r   1/usr/lib/python3.10/distutils/command/register.py<lambda>   s    zregister.<lambda>c                 C   s   t |  d| _d| _d S )Nr   )r   initialize_optionslist_classifiersr   r   r   r   r   r      s   

zregister.initialize_optionsc                 C   s*   t |  d| jfdd}|| jjd< d S )Nr   )r      )r   restructuredtextr
   )r   finalize_optionsr   distributioncommand_options)r   check_optionsr   r   r   r   $   s
   
zregister.finalize_optionsc                 C   sX   |    |   |  D ]}| | q| jr|   d S | jr&|   d S |   d S N)	r   _set_configget_sub_commandsrun_commanddry_runverify_metadatar   classifierssend_metadata)r   cmd_namer   r   r   run+   s   zregister.runc                 C   s8   t dt | jd}|  | j|_d|_|  dS )zDeprecated API.zddistutils.command.register.check_metadata is deprecated,               use the check command insteadr
   r   N)r   PendingDeprecationWarningr   get_command_objensure_finalizedr   r   r!   )r   r
   r   r   r   check_metadata:   s   zregister.check_metadatac                 C   s|   |   }|i kr!|d | _|d | _|d | _|d | _d| _d	S | jd| jfvr0td| j | jdkr9| j| _d| _d	S )
z: Reads the configuration file and set attributes.
        usernamepassword
repositoryrealmTpypiz%s not found in .pypircFN)_read_pypircr&   r'   r(   r)   
has_configDEFAULT_REPOSITORY
ValueError)r   configr   r   r   r   D   s   






zregister._set_configc                 C   s*   | j d }tj|}t| | dS )z8 Fetch the list of classifiers from the server.
        z?:action=list_classifiersN)r(   urllibrequesturlopenr   info_read_pypi_response)r   urlresponser   r   r   r   U   s   
zregister.classifiersc                 C   s&   |  | d\}}td|| dS )zF Send the metadata to the package index server to be checked.
        r	   Server response (%s): %sN)post_to_serverbuild_post_datar   r3   )r   coderesultr   r   r   r   \   s   zregister.verify_metadatac           
      C   s  | j rd}| j}| j}nd}d }}d }||vr5| dtj t }|s)d}n||vr1td ||vs|dkr|sAtd}|r;|sJt		d}|rCt
j }t
j| jd	 }|| j||| | | d
|\}}| d||f tj |dkr| j r|| j_dS | dtj | d|   tj d}| dvrtd}|sd}| dvs| dkr| || dS dS dS |dkr\ddi}	d |	d<  |	d< |	d< d|	d< |	d std|	d< |	d r|	d |	d kr+|	d st		d|	d< |	d r|	d st		d|	d< |	d r|	d |	d kr#d|	d< d|	d< td |	d |	d ks|	d s;td|	d< |	d r0| |	\}}|dkrPtd|| dS td td  dS |d!krdd"i}	d|	d< |	d sytd#|	d< |	d rn| |	\}}td|| dS dS )$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 z1 2 3 4zWe 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   submitr7      zAI can store your PyPI login so future submissions will be faster.z (the login will be stored in %s)XynzSave your login (y/N)?ny2:actionusernamer'   emailNconfirmz
 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.3password_resetzYour email address: )r,   r&   r'   splitannouncer   INFOinputprintgetpassr0   r1   HTTPPasswordMgrparseurlparser(   add_passwordr)   r8   r9   r   _get_rc_filelower_store_pypircr3   )
r   choicer&   r'   choicesauthhostr:   r;   datar   r   r   r   c   s   





	





zregister.send_metadatac                 C   s   | j j}i d|ddd| d| d| d| d| d	| d
| d|	 d|
 d| d| d| d| d| d| }|d sc|d sc|d rgd|d< |S )NrF   metadata_versionz1.0rH   versionsummary	home_pageauthorauthor_emaillicensedescriptionkeywordsplatformr   download_urlprovidesrequires	obsoletesz1.1)r   metadataget_nameget_versionget_descriptionget_urlget_contactget_contact_emailget_licenceget_long_descriptionget_keywordsget_platformsget_classifiersget_download_urlget_providesget_requiresget_obsoletes)r   actionmetar^   r   r   r   r9      sN   	
zregister.build_post_dataNc              
   C   s  d|v r|  d|d | jf tj d}d| }|d }t }| D ]?\}}t|tg tdfvr7|g}|D ])}t|}|	| |	d|  |	d |	| |rb|d	 d
krb|	d q9q$|	| |	d |
 d}d| tt|d}	tj| j||	}
tjtjj|d}d}z||
}W n; tjjy } z| jr|j }|j|jf}W Y d}~n(d}~w tjjy } zdt|f}W Y d}~nd}~ww | jr| |}d}| jrdd|df}|  |tj |S )zC Post a query to the server, and return a string response.
        rH   zRegistering %s to %sz3--------------GHSKFJDLGDS7543FJKLFHRE75642756743254z
--z--r   z*
Content-Disposition: form-data; name="%s"z


zutf-8z/multipart/form-data; boundary=%s; charset=utf-8)zContent-typezContent-length)password_mgrr>   Ni  )r@   OKzK---------------------------------------------------------------------------)rN   r(   r   rO   ioStringIOitemstypestrwritegetvalueencodelenr0   r1   Requestbuild_openerHTTPBasicAuthHandleropenerror	HTTPErrorshow_responsefpreadr:   msgURLErrorr4   join)r   r^   r\   boundarysep_boundaryend_boundarybodykeyvalueheadersreqopenerr;   er   r   r   r   r8      sh   








zregister.post_to_serverr   )__name__
__module____qualname__rf   r   user_optionsboolean_optionssub_commandsr   r   r!   r%   r   r   r   r   r9   r8   r   r   r   r   r      s$    
zr   )__doc__rR   r   urllib.parser0   urllib.requestwarningsr   distutils.corer   distutils.errors	distutilsr   r   r   r   r   r   <module>   s    