B
    È S]]  ã               @   s˜   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m	Z	m
Z
 ddlmZ ddlmZmZ ddlmZ ddlmZ dd	lmZ G d
d„ deƒZdS )zm
distutils.command.upload

Implements the Distutils 'upload' subcommand (upload package to a package
index).
é    N)Ústandard_b64encode)ÚurlopenÚRequestÚ	HTTPError)Úurlparse)ÚDistutilsErrorÚDistutilsOptionError)ÚPyPIRCCommand)Úspawn)Úlogc               @   sJ   e Zd ZdZejddg Zejdg Zdd„ Zdd„ Zd	d
„ Z	dd„ Z
dS )Úuploadzupload binary package to PyPI)ÚsignÚszsign files to upload using gpg)z	identity=ÚizGPG identity used to sign filesr   c             C   s,   t  | ¡ d| _d| _d| _d| _d | _d S )NÚ r   F)r	   Úinitialize_optionsÚusernameÚpasswordÚshow_responser   Úidentity)Úself© r   ú)lib/python3.7/distutils/command/upload.pyr       s    
zupload.initialize_optionsc             C   sr   t  | ¡ | jr| jstdƒ‚|  ¡ }|i krV|d | _|d | _|d | _|d | _	| jsn| j
jrn| j
j| _d S )Nz.Must use --sign for --identity to have meaningr   r   Ú
repositoryÚrealm)r	   Úfinalize_optionsr   r   r   Z_read_pypircr   r   r   r   Údistribution)r   Zconfigr   r   r   r   (   s    




zupload.finalize_optionsc             C   s>   | j jsd}t|ƒ‚x$| j jD ]\}}}|  |||¡ qW d S )NzHMust create and upload files in one command (e.g. setup.py sdist upload))r   Z
dist_filesr   Úupload_file)r   ÚmsgÚcommandÚ	pyversionÚfilenamer   r   r   Úrun:   s
    z
upload.runc       $      C   sê  t | jƒ\}}}}}}	|s"|s"|	r0td| j ƒ‚|dkrDtd| ƒ‚| jr|ddd|g}
| jrnd| jg|
dd…< t|
| jd	 t|d
ƒ}z| ¡ }W d | 	¡  X | j
j}dd| ¡ | ¡ tj |¡|f||t |¡ ¡ d| ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ |  ¡ dœ}d}|dkrZt! "¡ \}}}|rtd||f }n|dkrtdt!j!dd }||d< | jrªtj |¡d t|d d
ƒ ¡ f|d< | j#d | j$  %d¡}dt&|ƒ 'd¡ }d}d| %d¡ }|d }t( )¡ }x¤| *¡ D ]˜\}}d| }t+|t,ƒs |g}xr|D ]j}t-|ƒt.krR|d |d!  7 }|d }nt/|ƒ %d"¡}| 0|¡ | 0| %d"¡¡ | 0d#¡ | 0|¡ q&W qþW | 0|¡ | 1¡ }d$|| jf }|  2|t3j4¡ d%| t/t5|ƒƒ|d&œ}t6| j||d'}yt7|ƒ}| 8¡ } |j9}!W nf t:k
r: }" z|"j;} |"j9}!W d d }"~"X Y n8 t<k
rp }" z|  2t/|"ƒt3j=¡ ‚ W d d }"~"X Y nX | d(krÄ|  2d)| |!f t3j4¡ | j>ræ|  ?|¡}#d* @d+|#d+f¡}|  2|t3j4¡ n"d,| |!f }|  2|t3j=¡ tA|ƒ‚d S )-NzIncompatible url %s)ZhttpZhttpszunsupported schema Zgpgz--detach-signz-az--local-useré   )Údry_runÚrbZfile_uploadÚ1z1.0)z:actionZprotocol_versionÚnameÚversionÚcontentZfiletyper    Z
md5_digestZmetadata_versionZsummaryZ	home_pageZauthorZauthor_emailÚlicenseÚdescriptionÚkeywordsÚplatformZclassifiersZdownload_urlZprovidesZrequiresZ	obsoletesr   Z	bdist_rpmzbuilt for %s %sZ
bdist_dumbzbuilt for %sé   )ZterseÚcommentz.ascZgpg_signatureú:ÚasciizBasic z3--------------GHSKFJDLGDS7543FJKLFHRE75642756743254s   
--s   --
z+
Content-Disposition: form-data; name="%s"z; filename="%s"r   zutf-8s   

zSubmitting %s to %sz multipart/form-data; boundary=%s)zContent-typezContent-lengthZAuthorization)ÚdataÚheaderséÈ   zServer response (%s): %sÚ
zK---------------------------------------------------------------------------zUpload failed (%s): %s)Br   r   ÚAssertionErrorr   r   r
   r$   ÚopenÚreadÚcloser   ZmetadataZget_nameZget_versionÚosÚpathÚbasenameÚhashlibZmd5Z	hexdigestZget_descriptionZget_urlZget_contactZget_contact_emailZget_licenceZget_long_descriptionZget_keywordsZget_platformsZget_classifiersZget_download_urlZget_providesZget_requiresZget_obsoletesr-   Údistr   r   Úencoder   ÚdecodeÚioÚBytesIOÚitemsÚ
isinstanceÚlistÚtypeÚtupleÚstrÚwriteÚgetvalueZannouncer   ÚINFOÚlenr   r   Zgetcoder   r   ÚcodeÚOSErrorZERRORr   Z_read_pypi_responseÚjoinr   )$r   r   r    r!   ZschemaZnetlocZurlZparamsZqueryZ	fragmentsZgpg_argsÚfr)   Úmetar2   r/   r>   r(   ÚidZ	user_passZauthÚboundaryZsep_boundaryZend_boundaryZbodyÚkeyÚvalueÚtitler   r3   ZrequestÚresultZstatusÚreasonÚeÚtextr   r   r   r   B   sÂ    













zupload.upload_fileN)Ú__name__Ú
__module__Ú__qualname__r+   r	   Zuser_optionsZboolean_optionsr   r   r"   r   r   r   r   r   r      s   r   )Ú__doc__r:   rA   r-   r=   Úbase64r   Zurllib.requestr   r   r   Zurllib.parser   Zdistutils.errorsr   r   Zdistutils.corer	   Zdistutils.spawnr
   Z	distutilsr   r   r   r   r   r   Ú<module>   s   