U
    -ºcŸ  ã                   @   s”   d dl mZ d dlm  m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mZ ddlmZ eZdd
d„ZG dd„ deejƒZdS )é    )ÚlogN)Úchainé   )Úsdist_add_defaultsé   )Úmetadata)Ú_ORIGINAL_SUBCOMMANDSÚ c                 c   s.   t jddD ]}| ¡ | ƒD ]
}|V  qqdS )z%Find all files under revision controlzsetuptools.file_finders)ÚgroupN)r   Úentry_pointsÚload)ÚdirnameÚepÚitem© r   ú</tmp/pip-unpacked-wheel-izj_87as/setuptools/command/sdist.pyÚwalk_revctrl   s    r   c                       sÚ   e Zd ZdZdddddgZi Zddd	d
gZedd„ eD ƒƒZd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d%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Z‡  ZS )-Ú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	   z.rstz.txtz.mdc                 c   s   | ]}d   |¡V  qdS )z	README{0}N)Úformat)Ú.0Úextr   r   r   Ú	<genexpr>-   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ÚgetattrÚdistributionZ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Ú_default_to_gztar©r*   r   r   r   r1   B   s    zsdist.initialize_optionsc                 C   s   t jdkrd S dg| _d S )N)é   é   r   Úbetar   Úgztar)ÚsysÚversion_infoÚformatsr3   r   r   r   r2   G   s    
zsdist._default_to_gztarc              	   C   s$   |   ¡  tj | ¡ W 5 Q R X dS )z%
        Workaround for #516
        N)Ú_remove_os_linkr0   r   r'   r3   r   r   r   r'   M   s    
zsdist.make_distributionc               
   c   s^   G dd„ dƒ} t td| ƒ}zt`W n tk
r6   Y nX z
dV  W 5 || k	rXttd|ƒ 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   r?   ÚlinkN)r(   r"   r@   Ú	ExceptionÚsetattr)r?   Zorig_valr   r   r   r;   T   s    
zsdist._remove_os_linkc                    s   t ƒ  ¡  |  ¡  d S r/   )ÚsuperÚadd_defaultsÚ _add_defaults_build_sub_commandsr3   ©Ú	__class__r   r   rD   i   s    
zsdist.add_defaultsc                    s&   t ƒ  ¡  tj d¡r"| j d¡ d S )Nzpyproject.toml)rC   Ú_add_defaults_optionalr"   r#   Úisfiler    r!   r3   rF   r   r   rH   m   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*   rJ   r   r   r   Ú_add_defaults_pythonr   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 r/   )r   ©r   Úcr3   r   r   r   }   s     z9sdist._add_defaults_build_sub_commands.<locals>.<genexpr>c                 s   s    | ]}t |d ƒr| ¡ V  qdS )rM   N)ÚhasattrrM   rS   r   r   r   r   ~   s     
 )r   Úsetr&   r   r    rL   r   Úfrom_iterable)r*   rR   Zmissing_cmdsZcmdsÚfilesr   r3   r   rE   y   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_filesrP   r   r   r   rO      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 r/   )r"   r#   r$   )r   Ú_Úsrc_dirÚ	filenamesÚnamer   r   r   r   ’   s   
 þz(sdist._add_data_files.<locals>.<genexpr>N)r    rL   )r*   rY   r   r   r   rN   Ž   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)rC   Ú_add_defaults_data_filesÚ	TypeErrorr   Úwarnr3   rF   r   r   r^   ˜   s    zsdist._add_defaults_data_filesc                 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#   Úexistsr`   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   )r0   r   Úmake_release_treer"   r#   r$   rU   rb   ÚunlinkÚ	copy_filer   Zsave_version_info)r*   Úbase_dirrX   Údestr   r   r   rd   ¨   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Úrbz+# file GENERATED by distutils, do NOT edit
)r"   r#   rI   ÚmanifestÚioÚopenÚreadlineÚencode)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'ri   zUTF-8z"%r not UTF-8 decodable -- skippingú#N)r   Úinforj   rl   ÚdecodeÚUnicodeDecodeErrorr`   ÚstripÚ
startswithr    r!   Úclose)r*   rj   Úliner   r   r   Úread_manifest¿   s    
zsdist.read_manifest)r<   r=   r>   Ú__doc__Úuser_optionsÚnegative_optZREADME_EXTENSIONSÚtuplera   r.   r1   r2   r'   ÚstaticmethodÚ
contextlibÚcontextmanagerr;   rD   rH   rQ   rE   rO   rN   r^   r%   rd   rq   rz   Ú__classcell__r   r   rF   r   r      s8   õ


r   )r	   )Ú	distutilsr   Zdistutils.command.sdistÚcommandr   r0   r"   r8   rk   r€   Ú	itertoolsr   Z
py36compatr   Z
_importlibr   rR   r   ÚlistZ_default_revctrlr   r   r   r   r   Ú<module>   s   
