U
    ºÛui“  ã                   @  s¢   d dl m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	m
Z
 ddlmZ dd	lmZ d dlm  mZ d d
lmZ eZddd„ZG dd„ dejƒZdS )é    )ÚannotationsN)Úchain)ÚClassVaré   )Úmetadata)ÚDistributioné   )Ú_ORIGINAL_SUBCOMMANDS)ÚlogÚ c                 c  s(   t jddD ]}| ¡ | ƒE dH  qdS )z%Find all files under revision controlzsetuptools.file_finders)ÚgroupN)r   Úentry_pointsÚload)ÚdirnameÚep© r   ú</tmp/pip-unpacked-wheel-9osl9suo/setuptools/command/sdist.pyÚwalk_revctrl   s    r   c                      sð   e Zd ZU dZdddddgZded< i Zd	ed
< ddddgZedd„ eD ƒƒZ	dd„ Z
dd„ Zdd„ Zeejdd„ ƒƒZ‡ fdd„Z‡ fdd„Zdd„ Zdd „ Zd!d"„ Zd#d$„ Z‡ fd%d&„Z‡ fd'd(„Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Z‡  ZS )1Úsdistz=Smart sdist that finds anything supported by revision control)zformats=Nz6formats for source distribution (comma-separated list))z	keep-tempÚkz@keep the distribution tree around after creating archive file(s))z	dist-dir=ÚdzFdirectory to put the source distribution archive(s) in [default: dist])zowner=Úuz@Owner name used when creating a tar file [default: current user])zgroup=ÚgzAGroup name used when creating a tar file [default: current group]r   ÚdistributionzClassVar[dict[str, str]]Únegative_optr   z.rstz.txtz.mdc                 c  s   | ]}d   |¡V  qdS )z	README{0}N)Úformat)Ú.0Úextr   r   r   Ú	<genexpr>8   s     zsdist.<genexpr>c                 C  s–   |   d¡ |  d¡}|j| _| j tj |jd¡¡ |  ¡  |  	¡ D ]}|   |¡ qD|  
¡  t| jdg ƒ}| jD ] }dd|f}||krp| |¡ qpd S )NÚegg_infozSOURCES.txtÚ
dist_filesr   r   )Úrun_commandÚget_finalized_commandÚfilelistÚappendÚosÚpathÚjoinr   Úcheck_readmeÚget_sub_commandsÚmake_distributionÚgetattrr   Zarchive_files)ÚselfZei_cmdÚcmd_namer    ÚfileÚdatar   r   r   Úrun:   s    



z	sdist.runc                 C  s   t j | ¡ d S ©N)Úorigr   Úinitialize_options©r,   r   r   r   r3   M   s    zsdist.initialize_optionsc              	   C  s$   |   ¡  tj | ¡ W 5 Q R X dS )z%
        Workaround for #516
        N)Ú_remove_os_linkr2   r   r*   r4   r   r   r   r*   P   s    
zsdist.make_distributionc                  c  sX   G dd„ dƒ} t td| ƒ}zt`W n tk
r6   Y nX z
dV  W 5 || k	rR|t_X dS )zG
        In a context, remove and restore os.link if it exists
        c                   @  s   e Zd ZdS )z&sdist._remove_os_link.<locals>.NoValueN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   ÚNoValue^   s   r9   ÚlinkN)r+   r%   r:   Ú	Exception)r9   Zorig_valr   r   r   r5   W   s    
zsdist._remove_os_linkc                   s   t ƒ  ¡  |  ¡  d S r1   )ÚsuperÚadd_defaultsÚ _add_defaults_build_sub_commandsr4   ©Ú	__class__r   r   r=   l   s    
zsdist.add_defaultsc                   s&   t ƒ  ¡  tj d¡r"| j d¡ d S )Nzpyproject.toml)r<   Ú_add_defaults_optionalr%   r&   Úisfiler#   r$   r4   r?   r   r   rA   p   s    
zsdist._add_defaults_optionalc                 C  s8   | j  ¡ r4|  d¡}| j | ¡ ¡ |  |  |¡¡ dS )zgetting python filesÚbuild_pyN)r   Úhas_pure_modulesr"   r#   ÚextendÚget_source_filesÚ_add_data_filesÚ_safe_data_files©r,   rC   r   r   r   Ú_add_defaults_pythonu   s    

zsdist._add_defaults_pythonc                   sP   ˆ   d¡}t| ¡ ƒt }‡ fdd„|D ƒ}dd„ |D ƒ}ˆ j t |¡¡ d S )NÚbuildc                 3  s   | ]}ˆ   |¡V  qd S r1   )r"   ©r   Úcr4   r   r   r   €   s     z9sdist._add_defaults_build_sub_commands.<locals>.<genexpr>c                 s  s    | ]}t |d ƒr| ¡ V  qdS )rF   N)ÚhasattrrF   rL   r   r   r   r      s     
 )r"   Úsetr)   r	   r#   rE   r   Úfrom_iterable)r,   rK   Zmissing_cmdsZcmdsÚfilesr   r4   r   r>   |   s
    
z&sdist._add_defaults_build_sub_commandsc                 C  s   |j S )að  
        Since the ``sdist`` class is also used to compute the MANIFEST
        (via :obj:`setuptools.command.egg_info.manifest_maker`),
        there might be recursion problems when trying to obtain the list of
        data_files and ``include_package_data=True`` (which in turn depends on
        the files included in the MANIFEST).

        To avoid that, ``manifest_maker`` should be able to overwrite this
        method and avoid recursive attempts to build/analyze the MANIFEST.
        )Ú
data_filesrI   r   r   r   rH   „   s    zsdist._safe_data_filesc                 C  s   | j  dd„ |D ƒ¡ dS )zA
        Add data files as found in build_py.data_files.
        c                 s  s.   | ]&\}}}}|D ]}t j ||¡V  qqd S r1   )r%   r&   r'   )r   Ú_Úsrc_dirÚ	filenamesÚnamer   r   r   r   •   s   
 þz(sdist._add_data_files.<locals>.<genexpr>N)r#   rE   )r,   rR   r   r   r   rG   ‘   s    þzsdist._add_data_filesc                   s2   zt ƒ  ¡  W n tk
r,   t d¡ Y nX d S )Nz&data_files contains unexpected objects)r<   Ú_add_defaults_data_filesÚ	TypeErrorr
   Úwarnr4   r?   r   r   rW   ›   s    zsdist._add_defaults_data_filesc                   s.   t ƒ  ¡  t tj¡}| jjd| dd d S )Nz^(\.tox|\.nox|\.venv)T)Zis_regex)r<   Úprune_file_listÚreÚescaper%   Úsepr#   Zexclude_pattern)r,   r]   r?   r   r   rZ   ¡   s    
zsdist.prune_file_listc                 C  s8   | j D ]}tj |¡r d S q|  dd | j ¡ ¡ d S )Nz,standard file not found: should have one of z, )ÚREADMESr%   r&   ÚexistsrY   r'   )r,   Úfr   r   r   r(   §   s    
ÿzsdist.check_readmec                 C  s^   t j | ||¡ tj |d¡}ttdƒrJtj |¡rJt |¡ |  	d|¡ |  
d¡ |¡ d S )Nz	setup.cfgr:   r   )r2   r   Úmake_release_treer%   r&   r'   rN   r_   ÚunlinkÚ	copy_filer"   Zsave_version_info)r,   Úbase_dirrQ   Údestr   r   r   ra   °   s    
zsdist.make_release_treec              	   C  s:   t j | j¡sdS t| jdƒ}| ¡ }W 5 Q R X |dkS )NFÚrbs+   # file GENERATED by distutils, do NOT edit
)r%   r&   rB   ÚmanifestÚopenÚreadline)r,   ÚfpÚ
first_liner   r   r   Ú_manifest_is_not_generated½   s
    z sdist._manifest_is_not_generatedc              	   C  sŠ   t  d| j¡ t| jdƒ}|D ]^}z| d¡}W n& tk
rV   t  d| ¡ Y qY nX | ¡ }| d¡s|spq| j	 
|¡ q| ¡  dS )zªRead the manifest file (named by 'self.manifest') and use it to
        fill in 'self.filelist', the list of files to include in the source
        distribution.
        zreading manifest file '%s'rf   zUTF-8z"%r not UTF-8 decodable -- skippingú#N)r
   Úinforg   rh   ÚdecodeÚUnicodeDecodeErrorrY   ÚstripÚ
startswithr#   r$   Úclose)r,   rg   Úliner   r   r   Úread_manifestÆ   s    
zsdist.read_manifest)r6   r7   r8   Ú__doc__Úuser_optionsÚ__annotations__r   ZREADME_EXTENSIONSÚtupler^   r0   r3   r*   ÚstaticmethodÚ
contextlibÚcontextmanagerr5   r=   rA   rJ   r>   rH   rG   rW   rZ   r(   ra   rl   ru   Ú__classcell__r   r   r?   r   r      s:   
ï
		r   )r   )Ú
__future__r   r{   r%   r[   Ú	itertoolsr   Útypingr   Z
_importlibr   Údistr   rK   r	   Zdistutils.command.sdistÚcommandr   r2   Ú	distutilsr
   ÚlistZ_default_revctrlr   r   r   r   r   Ú<module>   s   
