a
    Ùc¿_­  ã                   @   s´   d 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 eed
dƒeeddƒeeddƒdœ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)ÚlogZmd5Zsha256Zblake2b)Z
md5_digestZsha256_digestZblake2_256_digestc                   @   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.9/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   1   s    
ÿ



zupload.finalize_optionsc                 C   s:   | j jsd}t|ƒ‚| j jD ]\}}}|  |||¡ q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   ÚrunC   s
    z
upload.runc           "      C   s
  t | jƒ\}}}}}}	|s"|s"|	r0td| j ƒ‚|dvrDtd| ƒ‚| jr|ddd|g}
| jrnd| jg|
dd…< t|
| jd	 t|d
ƒ}z| ¡ }W | 	¡  n
| 	¡  0 | j
j}dd| ¡ | ¡ tj |¡|f||d| ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ | ¡ dœ}d|d< t ¡ D ]B\}}|d u rLq6z||ƒ  ¡ ||< W n t!yt   Y n0 q6| jrÎt|d d
ƒ,}tj |¡d | ¡ f|d< W d   ƒ n1 sÄ0    Y  | j"d | j#  $d¡}dt%|ƒ &d¡ }d}d| $d¡ }|d }t' (¡ }| ¡ D ]”\}}d| }t)|t*ƒsB|g}|D ]j}t+|ƒt,u rr|d|d  7 }|d }nt-|ƒ $d¡}| .|¡ | .| $d¡¡ | .d¡ | .|¡ qFq | .|¡ | /¡ }d|| jf }|  0|t1j2¡ d | t-t3|ƒƒ|d!œ}t4| j||d"}zt5|ƒ}| 6¡ }|j7}W nj t8yX }  z| j9}| j7}W Y d } ~ nBd } ~ 0  t:y }  z |  0t-| ƒt1j;¡ ‚ W Y d } ~ n
d } ~ 0 0 |d#krä|  0d$||f t1j2¡ | j<r|  =|¡}!d% >d&|!d&f¡}|  0|t1j2¡ n"d'||f }|  0|t1j;¡ t?|ƒ‚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metadata_versionZsummaryZ	home_pageZauthorZauthor_emailÚlicenseÚdescriptionÚkeywordsÚplatformZclassifiersZdownload_urlZprovidesZrequiresZ	obsoletesr   Z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)@r   r   ÚAssertionErrorr   r   r
   r$   ÚopenÚreadÚcloser   ÚmetadataZget_nameZget_versionÚosÚpathÚbasenameZ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_obsoletesÚ_FILE_CONTENT_DIGESTSÚitemsZ	hexdigestÚ
ValueErrorr   r   Úencoder   ÚdecodeÚioÚBytesIOÚ
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urlÚparamsZqueryZ	fragmentsZgpg_argsÚfr)   Úmetar1   Zdigest_nameZdigest_consZ	user_passZauthÚboundaryZsep_boundaryZend_boundaryÚbodyÚkeyÚvalueÚtitler   r2   ZrequestÚresultZstatusÚreasonÚeÚtextr   r   r   r   K   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:   rB   ZhashlibÚbase64r   Zurllib.requestr   r   r   Zurllib.parser   Zdistutils.errorsr   r   Zdistutils.corer	   Zdistutils.spawnr
   Z	distutilsr   Úgetattrr=   r   r   r   r   r   Ú<module>   s   


ý