U
    Ȃe                     @   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mZ ddlmZ d dlZeZdddZG d	d
 d
eejZdS )    )logN)sixordered_set   )sdist_add_defaults c                 c   s,   t dD ]}| | D ]
}|V  qq
dS )z%Find all files under revision controlzsetuptools.file_findersN)pkg_resourcesiter_entry_pointsload)dirnameepitem r   </tmp/pip-unpacked-wheel-sm0xk1vo/setuptools/command/sdist.pywalk_revctrl   s    r   c                       s  e Zd ZdZ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dd Zejdk pdej  kodk n  pdej  kodk n  ZereZ f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d.d/ Z  Z S )0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]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   r-   >   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formatsr/   r   r   r   r.   C   s    
zsdist._default_to_gztarc              	   C   s$   |    tj|  W 5 Q R X dS )z%
        Workaround for #516
        N)_remove_os_linkr,   r   r#   r/   r   r   r   r#   I   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   NoValueW   s   r;   linkN)r$   r   r<   	Exceptionsetattr)r;   Zorig_valr   r   r   r7   P   s    
zsdist._remove_os_linkc                 C   sL   zt j|  W n6 tk
rF   t \}}}|jjjd 	   Y nX d S )Ntemplate)
r,   r   read_templater=   r4   exc_infotb_nexttb_framef_localsclose)r&   _tbr   r   r   Z__read_template_hacke   s    zsdist.__read_template_hack)      rH   )r0   r   )r0   r      )r0   rH   )r0   rH   r   c                    s8   t jrt|  n
t   tjdr4| j	d d S )Nzpyproject.toml)
r   PY2r   _add_defaults_optionalsuperr   r   isfiler   r   r/   	__class__r   r   rL   |   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   extendZget_source_files_add_data_files_safe_data_filesr&   rQ   r   r   r   _add_defaults_python   s    

zsdist._add_defaults_pythonc                 C   s   | j jrdS |jS )z
        Extracting data_files from build_py is known to cause
        infinite recursion errors when `include_package_data`
        is enabled, so suppress it in that case.
        r   )r%   Zinclude_package_data
data_filesrV   r   r   r   rU      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   rF   src_dir	filenamesnamer   r   r   r      s   
 z(sdist._add_data_files.<locals>.<genexpr>N)r   rS   )r&   rX   r   r   r   rT      s    zsdist._add_data_filesc                    sD   z t jrt|  n
t   W n tk
r>   td Y nX d S )Nz&data_files contains unexpected objects)r   rK   r   _add_defaults_data_filesrM   	TypeErrorr   warnr/   rO   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   )r,   r   make_release_treer   r   r    hasattrr`   unlink	copy_filer   Zsave_version_info)r&   base_dirfilesdestr   r   r   rb      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   rN   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 ]d}tjr^z|d}W n& tk
r\   t d|  Y qY nX |	 }|
ds|svq| j| q|  dS )zRead 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   r   PY3decodeUnicodeDecodeErrorr^   strip
startswithr   r   rE   )r&   rj   liner   r   r   read_manifest   s    
zsdist.read_manifestc                 C   s   t  }| jd}|dd\}}|dkr8td n
|| z|| jj	j
 W n tk
rt   td Y nX |D ]&}tj|sztd| || qz| j| dS )zvChecks if license_file' or 'license_files' is configured and adds any
        valid paths to 'self.filelist'.
        metadatalicense_file)NNNz''license_file' option was not specifiedz,warning: 'license_files' option is malformedz8warning: Failed to find the configured license file '%s')r   Z
OrderedSetr%   get_option_dictgetr   debugaddupdater{   Zlicense_filesr]   r^   r   r   r`   remover   rS   )r&   rg   optsrF   r|   ra   r   r   r   check_license   s$    
zsdist.check_license)!r8   r9   r:   __doc__user_optionsnegative_optZREADME_EXTENSIONStupler_   r*   r-   r.   r#   staticmethod
contextlibcontextmanagerr7   Z_sdist__read_template_hackr4   r5   Zhas_leaky_handler@   rL   rW   rU   rT   r\   r!   rb   rq   rz   r   __classcell__r   r   rO   r   r      sB   


	

r   )r   )	distutilsr   Zdistutils.command.sdistcommandr   r,   r   r4   rk   r   Zsetuptools.externr   r   Z
py36compatr   r   listZ_default_revctrlr   r   r   r   r   <module>   s   
