B
    v9aC  ã               @   s~   d 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T ddl	m
Z
mZ ddlmZ G dd	„ d	eƒZG d
d„ deeƒZdS )zHdistutils.command.build_py

Implements the Distutils 'build_py' command.é    N)Úglob)ÚCommand)Ú*)Úconvert_pathÚ	Mixin2to3)Úlogc               @   sÈ   e Zd ZdZdddddgZddgZd	di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d2d'd(„Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Zd1S )3Úbuild_pyz5"build" pure Python modules (copy to build directory))z
build-lib=Údzdirectory to "build" (copy) to)ÚcompileÚczcompile .py to .pyc)z
no-compileNz!don't compile .py files [default])z	optimize=ÚOzlalso compile with optimization: -O1 for "python -O", -O2 for "python -OO", and -O0 to disable [default: -O0])ÚforceÚfz2forcibly build everything (ignore file timestamps)r
   r   z
no-compilec             C   s4   d | _ d | _d | _d | _d | _d| _d| _d | _d S )Nr   )Ú	build_libÚ
py_modulesÚpackageÚpackage_dataÚpackage_dirr
   Úoptimizer   )Úself© r   ú/usr/lib/python3.7/build_py.pyÚinitialize_options    s    zbuild_py.initialize_optionsc          	   C   s®   |   ddd¡ | jj| _| jj| _| jj| _i | _| jjrbx&| jj ¡ D ]\}}t|ƒ| j|< qHW |  ¡ | _	t
| jtƒsªyt| jƒ| _W n  ttfk
r¨   tdƒ‚Y nX d S )NZbuild)r   r   )r   r   zoptimize must be 0, 1, or 2)Zset_undefined_optionsÚdistributionÚpackagesr   r   r   Úitemsr   Úget_data_filesÚ
data_filesÚ
isinstancer   ÚintÚ
ValueErrorÚAssertionErrorZDistutilsOptionError)r   ÚnameÚpathr   r   r   Úfinalize_options*   s"    



zbuild_py.finalize_optionsc             C   s:   | j r|  ¡  | jr$|  ¡  |  ¡  |  | jdd¡ d S )Nr   )Úinclude_bytecode)r   Úbuild_modulesr   Úbuild_packagesÚbuild_package_dataÚbyte_compileÚget_outputs)r   r   r   r   ÚrunC   s    zbuild_py.runc                s†   g }| j s|S xr| j D ]h}|  |¡}tjj| jg| d¡ Ž }d‰ |rRt|ƒd ‰ ‡ fdd„|  ||¡D ƒ}| 	||||f¡ qW |S )z?Generate list of '(package,src_dir,build_dir,filenames)' tuplesÚ.r   é   c                s   g | ]}|ˆ d … ‘qS )Nr   )Ú.0Úfile)Úplenr   r   ú
<listcomp>t   s    z+build_py.get_data_files.<locals>.<listcomp>)
r   Úget_package_dirÚosr#   Újoinr   ÚsplitÚlenÚfind_data_filesÚappend)r   Údatar   Úsrc_dirÚ	build_dirÚ	filenamesr   )r0   r   r   a   s    
zbuild_py.get_data_filesc                s`   | j  dg ¡| j  |g ¡ }g ‰ x:|D ]2}ttj |t|ƒ¡ƒ}ˆ  ‡ fdd„|D ƒ¡ q&W ˆ S )z6Return filenames for package's data files in 'src_dir'Ú c                s$   g | ]}|ˆ krt j |¡r|‘qS r   )r3   r#   Úisfile)r.   Úfn)Úfilesr   r   r1   ‚   s    
z,build_py.find_data_files.<locals>.<listcomp>)r   Úgetr   r3   r#   r4   r   Úextend)r   r   r:   ZglobsÚpatternZfilelistr   )r@   r   r7   y   s    
zbuild_py.find_data_filesc             C   sh   d}x^| j D ]T\}}}}xF|D ]>}tj ||¡}|  tj |¡¡ | jtj ||¡|dd qW qW dS )z$Copy data files into build directoryNF)Úpreserve_mode)r   r3   r#   r4   ÚmkpathÚdirnameÚ	copy_file)r   Zlastdirr   r:   r;   r<   ÚfilenameÚtargetr   r   r   r(   †   s    
zbuild_py.build_package_datac             C   sÄ   |  d¡}| js&|r tjj|Ž S dS nšg }x”|rŠy| jd |¡ }W n* tk
rn   | d|d ¡ |d= Y q,X | d|¡ tjj|Ž S q,W | j d¡}|dk	r¬| d|¡ |r¼tjj|Ž S dS dS )z½Return the directory, relative to the top of the source
           distribution, where package 'package' should be found
           (at least according to the 'package_dir' option, if any).r,   r=   r   éÿÿÿÿN)r5   r   r3   r#   r4   ÚKeyErrorÚinsertrA   )r   r   r#   ÚtailZpdirr   r   r   r2      s(    
	zbuild_py.get_package_dirc             C   sj   |dkr8t j |¡s td| ƒ‚t j |¡s8td| ƒ‚|rft j |d¡}t j |¡rZ|S t d|¡ d S )Nr=   z%package directory '%s' does not existz>supposed package directory '%s' exists, but is not a directoryz__init__.pyz8package init file '%s' not found (or not a regular file))	r3   r#   ÚexistsZDistutilsFileErrorÚisdirr4   r>   r   Úwarn)r   r   r   Úinit_pyr   r   r   Úcheck_package·   s    
zbuild_py.check_packagec             C   s&   t j |¡st d||¡ dS dS d S )Nz!file %s (for module %s) not foundFT)r3   r#   r>   r   rP   )r   ÚmoduleÚmodule_filer   r   r   Úcheck_moduleÒ   s    zbuild_py.check_modulec       	      C   s   |   ||¡ ttj |d¡ƒ}g }tj | jj¡}xX|D ]P}tj |¡}||krztj tj 	|¡¡d }| 
|||f¡ q8|  d| ¡ q8W |S )Nz*.pyr   zexcluding %s)rR   r   r3   r#   r4   Úabspathr   Zscript_nameÚsplitextÚbasenamer8   Zdebug_print)	r   r   r   Zmodule_filesÚmodulesZsetup_scriptr   Zabs_frS   r   r   r   Úfind_package_modulesÙ   s    
zbuild_py.find_package_modulesc          	   C   sÔ   i }g }xÆ| j D ]¼}| d¡}d |dd… ¡}|d }y|| \}}W n" tk
rj   |  |¡}d}Y nX |sœ|  ||¡}	|df||< |	rœ| |d|	f¡ tj ||d ¡}
|  	||
¡s¼q| |||
f¡ qW |S )aÈ  Finds individually-specified Python modules, ie. those listed by
        module name in 'self.py_modules'.  Returns a list of tuples (package,
        module_base, filename): 'package' is a tuple of the path through
        package-space to the module; 'module_base' is the bare (no
        packages, no dots) module name, and 'filename' is the path to the
        ".py" file (relative to the distribution root) that implements the
        module.
        r,   r   rJ   r-   Ú__init__z.py)
r   r5   r4   rK   r2   rR   r8   r3   r#   rU   )r   r   rY   rS   r#   r   Zmodule_baser   ÚcheckedrQ   rT   r   r   r   Úfind_modulesè   s*    


zbuild_py.find_modulesc             C   sR   g }| j r| |  ¡ ¡ | jrNx.| jD ]$}|  |¡}|  ||¡}| |¡ q&W |S )a4  Compute the list of all modules that will be built, whether
        they are specified one-module-at-a-time ('self.py_modules') or
        by whole packages ('self.packages').  Return a list of tuples
        (package, module, module_file), just like 'find_modules()' and
        'find_package_modules()' do.)r   rB   r]   r   r2   rZ   )r   rY   r   r   Úmr   r   r   Úfind_all_modules  s    
zbuild_py.find_all_modulesc             C   s   dd„ |   ¡ D ƒS )Nc             S   s   g | ]}|d  ‘qS )rJ   r   )r.   rS   r   r   r   r1   -  s    z-build_py.get_source_files.<locals>.<listcomp>)r_   )r   r   r   r   Úget_source_files,  s    zbuild_py.get_source_filesc             C   s$   |gt |ƒ |d g }tjj|Ž S )Nz.py)Úlistr3   r#   r4   )r   r;   r   rS   Zoutfile_pathr   r   r   Úget_module_outfile/  s    zbuild_py.get_module_outfiler-   c             C   sž   |   ¡ }g }xx|D ]p\}}}| d¡}|  | j||¡}| |¡ |r| jr`| tjj|dd¡ | j	dkr| tjj|| j	d¡ qW |dd„ | j
D ƒ7 }|S )Nr,   r=   )Úoptimizationr   c             S   s,   g | ]$\}}}}|D ]}t j ||¡‘qqS r   )r3   r#   r4   )r.   r   r:   r;   r<   rH   r   r   r   r1   C  s   
z(build_py.get_outputs.<locals>.<listcomp>)r_   r5   rb   r   r8   r
   Ú	importlibÚutilÚcache_from_sourcer   r   )r   r%   rY   Zoutputsr   rS   rT   rH   r   r   r   r*   3  s"    




zbuild_py.get_outputsc             C   sb   t |tƒr| d¡}nt |ttfƒs,tdƒ‚|  | j||¡}tj	 
|¡}|  |¡ | j||ddS )Nr,   z:'package' must be a string (dot-separated), list, or tupler   )rD   )r   Ústrr5   ra   ÚtupleÚ	TypeErrorrb   r   r3   r#   rF   rE   rG   )r   rS   rT   r   ZoutfileÚdirr   r   r   Úbuild_moduleJ  s    

zbuild_py.build_modulec             C   s.   |   ¡ }x |D ]\}}}|  |||¡ qW d S )N)r]   rk   )r   rY   r   rS   rT   r   r   r   r&   Y  s    zbuild_py.build_modulesc             C   sL   xF| j D ]<}|  |¡}|  ||¡}x |D ]\}}}|  |||¡ q(W qW d S )N)r   r2   rZ   rk   )r   r   r   rY   Zpackage_rS   rT   r   r   r   r'   b  s
    

zbuild_py.build_packagesc             C   s€   t jr|  d¡ d S ddlm} | j}|d tjkr>|tj }| jrZ||d| j	|| j
d | jdkr|||| j| j	|| j
d d S )Nz%byte-compiling is disabled, skipping.r   )r)   rJ   )r   r   ÚprefixÚdry_run)ÚsysÚdont_write_bytecoderP   Údistutils.utilr)   r   r3   Úsepr
   r   rm   r   )r   r@   r)   rl   r   r   r   r)   v  s    


zbuild_py.byte_compileN)r-   )Ú__name__Ú
__module__Ú__qualname__ÚdescriptionZuser_optionsZboolean_optionsZnegative_optr   r$   r+   r   r7   r(   r2   rR   rU   rZ   r]   r_   r`   rb   r*   rk   r&   r'   r)   r   r   r   r   r      s6   

'4
	r   c               @   s   e Zd Zdd„ Zdd„ ZdS )Úbuild_py_2to3c             C   sL   g | _ | jr|  ¡  | jr*|  ¡  |  ¡  |  | j ¡ |  | jdd¡ d S )Nr   )r%   )	Úupdated_filesr   r&   r   r'   r(   Zrun_2to3r)   r*   )r   r   r   r   r+   ‹  s    zbuild_py_2to3.runc             C   s,   t  | |||¡}|d r(| j |d ¡ |S )Nr-   r   )r   rk   rw   r8   )r   rS   rT   r   Zresr   r   r   rk   ›  s    zbuild_py_2to3.build_moduleN)rr   rs   rt   r+   rk   r   r   r   r   rv   Š  s   rv   )Ú__doc__r3   Úimportlib.utilrd   rn   r   Zdistutils.corer   Zdistutils.errorsrp   r   r   Z	distutilsr   r   rv   r   r   r   r   Ú<module>   s     }