o
    Ëý°j2  ã                   @   s¤   d Z ddlmZ ddlmZ ddlmZ ddlZddlZddl	Z	ddl
Z
ddl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
„ ZG dd„ deƒZdS )z|upload_docs

Implements a Distutils 'upload_docs' subcommand (upload documentation to
sites other than PyPi such as devpi).
é    )Ústandard_b64encode)Úlog)ÚDistutilsOptionErrorN)Úiter_entry_pointsé   )Úuploadc                 C   s   |   dd¡S )Nzutf-8Úsurrogateescape)Úencode)Ús© r   ú‘/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/setuptools/command/upload_docs.pyÚ_encode   s   r   c                   @   s‚   e Zd ZdZdZdddej fddgZejZdd	„ Zd
efgZ	dd„ Z
dd„ Zdd„ Zdd„ Zedd„ ƒZedd„ ƒZdd„ ZdS )Úupload_docszhttps://pypi.python.org/pypi/z;Upload documentation to sites other than PyPi such as devpizrepository=Úrzurl of repository [default: %s])zshow-responseNz&display full response text from server)zupload-dir=Nzdirectory to uploadc                 C   s&   | j d u rtddƒD ]} dS d S d S )Nzdistutils.commandsÚbuild_sphinxT)Ú
upload_dirr   )ÚselfÚepr   r   r   Ú
has_sphinx-   s
   
þzupload_docs.has_sphinxr   c                 C   s   t  | ¡ d | _d | _d S )N)r   Úinitialize_optionsr   Ú
target_dir)r   r   r   r   r   4   s   

zupload_docs.initialize_optionsc                 C   s�   t  | ¡ | jd u r+|  ¡ r|  d¡}t|jƒd | _n|  d¡}tj	 
|jd¡| _n	|  d¡ | j| _d| jv r>t d¡ |  d| j ¡ d S )	Nr   ÚhtmlÚbuildÚdocsr   zpypi.python.orgz<Upload_docs command is deprecated for PyPi. Use RTD instead.zUsing upload directory %s)r   Úfinalize_optionsr   r   Úget_finalized_commandÚdictÚbuilder_target_dirsr   ÚosÚpathÚjoinÚ
build_baseÚensure_dirnameÚ
repositoryr   ÚwarnÚannounce)r   r   r   r   r   r   r   9   s   






zupload_docs.finalize_optionsc                 C   s¸   t  |d¡}zP|  | j¡ t | j¡D ]<\}}}|| jkr(|s(d}t|| j ƒ‚|D ]$}tj ||¡}|t	| jƒd …  
tjj¡}	tj |	|¡}
| ||
¡ q*qW | ¡  d S | ¡  w )NÚwz'no files found in upload directory '%s')ÚzipfileÚZipFileÚmkpathr   r   Úwalkr   r   r    ÚlenÚlstripÚsepÚwriteÚclose)r   ÚfilenameÚzip_fileÚrootÚdirsÚfilesÚtmplÚnameÚfullÚrelativeÚdestr   r   r   Úcreate_zipfileI   s   üü
zupload_docs.create_zipfilec              	   C   sp   |   ¡ D ]}|  |¡ qt ¡ }| jj ¡ }tj 	|d| ¡}z|  
|¡ |  |¡ W t |¡ d S t |¡ w )Nz%s.zip)Úget_sub_commandsÚrun_commandÚtempfileÚmkdtempÚdistributionÚmetadataÚget_namer   r   r    r:   Úupload_fileÚshutilÚrmtree)r   Úcmd_nameÚtmp_dirr6   r1   r   r   r   ÚrunY   s   
zupload_docs.runc                 c   s’   � | \}}d| }t |tƒs|g}|D ]3}t |tƒr'|d|d  7 }|d }nt|ƒ}|V  t|ƒV  dV  |V  |rF|dd … dkrFdV  qd S )	Nz*
Content-Disposition: form-data; name="%s"z; filename="%s"r   r   s   

éÿÿÿÿó   ó   
)Ú
isinstanceÚlistÚtupler   )ÚitemÚsep_boundaryÚkeyÚvaluesÚtitleÚvaluer   r   r   Ú_build_partg   s$   €



€õzupload_docs._build_partc                 C   sn   d}d|  d¡ }|d }|df}tj| j|d�}t|| ¡ ƒ}tj |¡}t ||¡}	d| }
d 	|	¡|
fS )	z=
        Build up the MIME payload for the POST data
        z3--------------GHSKFJDLGDS7543FJKLFHRE75642756743254s   
--Úasciis   --rJ   )rO   z multipart/form-data; boundary=%só    )
r	   Ú	functoolsÚpartialrT   ÚmapÚitemsÚ	itertoolsÚchainÚfrom_iterabler    )ÚclsÚdataÚboundaryrO   Úend_boundaryÚ	end_itemsÚbuilderÚpart_groupsÚpartsÚ
body_itemsÚcontent_typer   r   r   Ú_build_multipart{   s   þzupload_docs._build_multipartc              
   C   sT  t |dƒ�}| ¡ }W d   ƒ n1 sw   Y  | jj}d| ¡ tj |¡|fdœ}t| j	d | j
 ƒ}t|ƒ d¡}d| }|  |¡\}}	d| j }
|  |
tj¡ tj | j¡\}}}}}}|sf|sf|rhJ ‚|dkrstj |¡}n|d	kr~tj |¡}ntd
| ƒ‚d}z-| ¡  | d|¡ |	}| d|¡ | dtt|ƒƒ¡ | d|¡ | ¡  |  |¡ W n t!j"yÐ } z|  t|ƒtj#¡ W Y d }~d S d }~ww | $¡ }|j%dkrêd|j%|j&f }
|  |
tj¡ n0|j%dk�r| 'd¡}|d u rÿd| ¡  }d| }
|  |
tj¡ nd|j%|j&f }
|  |
tj#¡ | j(�r(t)d| ¡ dƒ d S d S )NÚrbÚ
doc_upload)z:actionr6   Úcontentú:rU   zBasic zSubmitting documentation to %sÚhttpÚhttpszunsupported schema Ú ÚPOSTzContent-typezContent-lengthÚAuthorizationéÈ   zServer response (%s): %si-  ÚLocationzhttps://pythonhosted.org/%s/zUpload successful. Visit %szUpload failed (%s): %szK---------------------------------------------------------------------------)*ÚopenÚreadr?   r@   rA   r   r   Úbasenamer   ÚusernameÚpasswordr   Údecoderh   r#   r%   r   ÚINFOÚurllibÚparseÚurlparserm   ÚclientÚHTTPConnectionÚHTTPSConnectionÚAssertionErrorÚconnectÚ
putrequestÚ	putheaderÚstrr+   Ú
endheadersÚsendÚsocketÚerrorÚERRORÚgetresponseÚstatusÚreasonÚ	getheaderÚshow_responseÚprint)r   r0   Úfrk   Úmetar_   ÚcredentialsÚauthÚbodyÚctÚmsgÚschemaÚnetlocÚurlÚparamsÚqueryÚ	fragmentsÚconnrg   Úer   Úlocationr   r   r   rB   Ž   sh   
ÿý
ÿ€þ

ÿzupload_docs.upload_fileN)Ú__name__Ú
__module__Ú__qualname__ÚDEFAULT_REPOSITORYÚdescriptionr   Úuser_optionsÚboolean_optionsr   Úsub_commandsr   r   r:   rG   ÚstaticmethodrT   Úclassmethodrh   rB   r   r   r   r   r      s*    ÿû


r   )Ú__doc__Úbase64r   Ú	distutilsr   Údistutils.errorsr   r   rˆ   r'   r=   rC   r[   rW   Úhttp.clientrm   Úurllib.parser{   Úpkg_resourcesr   r   r   r   r   r   r   r   Ú<module>   s"   