U
    HaT                      @   s   d dl m Z  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Zd dlZd dlZd dlmZ dd ZG dd dejZd	d
 ZdS )    )glob)convert_pathN)unique_everseenc                 C   s   t | t | jtjB  d S N)oschmodstatst_modeS_IWRITE)target r   ?/tmp/pip-unpacked-wheel-i11yjail/setuptools/command/build_py.pymake_writable   s    r   c                   @   s   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zedd Zd S )!build_pyaX  Enhanced 'build_py' command that includes data files with packages

    The data files are specified via a 'package_data' argument to 'setup()'.
    See 'setuptools.dist.Distribution' for more details.

    Also, this version of the 'build_py' command allows you to specify both
    'py_modules' and 'packages' in the same setup operation.
    c                 C   s@   t j|  | jj| _| jjp i | _d| jkr6| jd= g | _d S )N
data_files)origr   finalize_optionsdistributionpackage_dataexclude_package_data__dict___build_py__updated_filesselfr   r   r   r      s    

zbuild_py.finalize_optionsc                 C   sN   | j s| jsdS | j r|   | jr4|   |   | tjj| dd dS )z?Build modules, packages, and copy data files to build directoryNr   )Zinclude_bytecode)	
py_modulespackagesZbuild_modulesZbuild_packagesbuild_package_databyte_compiler   r   get_outputsr   r   r   r   run$   s    zbuild_py.runc                 C   s&   |dkr|   | _| jS tj| |S )zlazily compute data filesr   )_get_data_filesr   r   r   __getattr__)r   attrr   r   r   r!   4   s    
zbuild_py.__getattr__c                 C   s.   t j| |||\}}|r&| j| ||fS r   )r   r   build_moduler   append)r   moduleZmodule_filepackageoutfilecopiedr   r   r   r#   ;   s    zbuild_py.build_modulec                 C   s   |    tt| j| jpdS )z?Generate list of '(package,src_dir,build_dir,filenames)' tuplesr   )analyze_manifestlistmap_get_pkg_data_filesr   r   r   r   r   r    A   s    zbuild_py._get_data_filesc                    sJ   |  | tjj| jg|d  } fdd| | D }| ||fS )N.c                    s   g | ]}t j| qS r   )r   pathrelpath).0filesrc_dirr   r   
<listcomp>N   s   z0build_py._get_pkg_data_files.<locals>.<listcomp>)get_package_dirr   r.   join	build_libsplitfind_data_files)r   r&   	build_dir	filenamesr   r2   r   r,   F   s    


zbuild_py._get_pkg_data_filesc                 C   sX   |  | j||}tt|}tj|}ttj	j
|}t| j|g |}| |||S )z6Return filenames for package's data files in 'src_dir')_get_platform_patternsr   r+   r   	itertoolschainfrom_iterablefilterr   r.   isfilemanifest_filesgetexclude_data_files)r   r&   r3   patternsZglobs_expandedZglobs_matchesZ
glob_filesfilesr   r   r   r9   T   s    
zbuild_py.find_data_filesc           
      C   st   | j D ]h\}}}}|D ]V}tj||}| tj| tj||}| ||\}}	t| tj|}qqdS )z$Copy data files into build directoryN)	r   r   r.   r6   mkpathdirname	copy_filer   abspath)
r   r&   r3   r:   r;   filenamer   srcfileoutfr(   r   r   r   r   e   s    zbuild_py.build_package_datac                 C   s   i  | _ }| jjsd S i }| jp"dD ]}||t| |< q$| d | d}|jj	D ]}t
jt|\}}d }|}	|r||kr||kr|}t
j|\}}
t
j|
|}qx||krX|dr||	krqX||| g | qXd S )Nr   egg_infoz.py)rB   r   Zinclude_package_datar   assert_relativer5   run_commandget_finalized_commandfilelistrF   r   r.   r8   r6   endswith
setdefaultr$   )r   ZmfZsrc_dirsr&   Zei_cmdr.   dfprevZoldfZdfr   r   r   r)   p   s(    


zbuild_py.analyze_manifestc                 C   s   d S r   r   r   r   r   r   get_data_files   s    zbuild_py.get_data_filesc              	   C   s   z| j | W S  tk
r    Y nX tj| ||}|| j |< |rH| jjsL|S | jjD ]}||ksn||d rT qxqT|S t	|d}|
 }W 5 Q R X d|krtjd|f |S )z8Check namespace packages' __init__ for declare_namespacer-   rbs   declare_namespacezNamespace package problem: %s is a namespace package, but its
__init__.py does not call declare_namespace()! Please fix it.
(See the setuptools manual under "Namespace Packages" for details.)
")packages_checkedKeyErrorr   r   check_packager   Znamespace_packages
startswithioopenread	distutilserrorsDistutilsError)r   r&   package_dirZinit_pypkgrV   contentsr   r   r   r\      s*    
zbuild_py.check_packagec                 C   s   i | _ tj|  d S r   )rZ   r   r   initialize_optionsr   r   r   r   rg      s    zbuild_py.initialize_optionsc                 C   s0   t j| |}| jjd k	r,tj| jj|S |S r   )r   r   r5   r   Zsrc_rootr   r.   r6   )r   r&   resr   r   r   r5      s    zbuild_py.get_package_dirc                    s\   t | | j||}fdd|D }tj|}t|  fddD }t t|S )z6Filter filenames for package's data files in 'src_dir'c                 3   s   | ]}t  |V  qd S r   )fnmatchr@   r0   pattern)rF   r   r   	<genexpr>   s     z.build_py.exclude_data_files.<locals>.<genexpr>c                 3   s   | ]}| kr|V  qd S r   r   )r0   fn)badr   r   rl      s      )r*   r<   r   r=   r>   r?   setr   )r   r&   r3   rF   rE   Zmatch_groupsmatchesZkeepersr   )rn   rF   r   rD      s    zbuild_py.exclude_data_filesc                    s.   t | dg | |g } fdd|D S )z
        yield platform-specific path patterns (suitable for glob
        or fn_match) from a glob-based spec (such as
        self.package_data or self.exclude_package_data)
        matching package in src_dir.
         c                 3   s    | ]}t j t|V  qd S r   )r   r.   r6   r   rj   r2   r   r   rl      s   z2build_py._get_platform_patterns.<locals>.<genexpr>)r=   r>   rC   )specr&   r3   Zraw_patternsr   r2   r   r<      s    


zbuild_py._get_platform_patternsN)__name__
__module____qualname____doc__r   r   r!   r#   r    r,   r9   r   r)   rX   r\   rg   r5   rD   staticmethodr<   r   r   r   r   r      s"   	r   c                 C   s:   t j| s| S ddlm} td |  }||d S )Nr   )DistutilsSetupErrorz
        Error: setup script specifies an absolute path:

            %s

        setup() arguments must *always* be /-separated paths relative to the
        setup.py directory, *never* absolute paths.
        )r   r.   isabsdistutils.errorsrx   textwrapdedentlstrip)r.   rx   msgr   r   r   rO      s    
rO   )r   distutils.utilr   Zdistutils.command.build_pycommandr   r   r   ri   r{   r^   rz   ra   r=   r   Z setuptools.extern.more_itertoolsr   r   rO   r   r   r   r   <module>   s    E