B
    È S]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   ú+lib/python3.7/distutils/command/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ƒ| _d| j  krœdks¢n t‚W n  ttfk
rÄ   tdƒ‚Y nX d S )NZbuild)r   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ÚAssertionErrorÚ
ValueErrorZ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   )r1   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   )r4   r$   Úisfile)r/   Úfn)Úfilesr   r   r2   ‚   s    
z,build_py.find_data_files.<locals>.<listcomp>)r   Úgetr   r4   r$   r5   r   Úextend)r   r   r;   ZglobsÚpatternZfilelistr   )rA   r   r8   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   r4   r$   r5   Ú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)r6   r   r4   r$   r5   ÚKeyErrorÚinsertrB   )r   r   r$   ÚtailZpdirr   r   r   r3      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))	r4   r$   ÚexistsZDistutilsFileErrorÚisdirr5   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)r4   r$   r?   r   rQ   )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)rS   r   r4   r$   r5   Úabspathr   Úscript_nameÚsplitextÚbasenamer9   Zdebug_print)	r   r   r   Zmodule_filesÚmodulesZsetup_scriptr   Zabs_frT   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   rK   r.   Ú__init__z.py)
r   r6   r5   rL   r3   rS   r9   r4   r$   rV   )r   r   r[   rT   r$   r   Zmodule_baser   ÚcheckedrR   rU   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   rC   r_   r   r3   r\   )r   r[   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 )rK   r   )r/   rT   r   r   r   r2   -  s    z-build_py.get_source_files.<locals>.<listcomp>)ra   )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)Úlistr4   r$   r5   )r   r<   r   rT   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   )r4   r$   r5   )r/   r   r;   r<   r=   rI   r   r   r   r2   C  s   
z(build_py.get_outputs.<locals>.<listcomp>)ra   r6   rd   r   r9   r
   Ú	importlibÚutilÚcache_from_sourcer   r   )r   r&   r[   Zoutputsr   rT   rU   rI   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   )rE   )r   Ústrr6   rc   ÚtupleÚ	TypeErrorrd   r   r4   r$   rG   rF   rH   )r   rT   rU   r   ZoutfileÚdirr   r   r   Úbuild_moduleJ  s    

zbuild_py.build_modulec             C   s.   |   ¡ }x |D ]\}}}|  |||¡ qW d S )N)r_   rm   )r   r[   r   rT   rU   r   r   r   r'   Y  s    zbuild_py.build_modulesc             C   sX   xR| j D ]H}|  |¡}|  ||¡}x,|D ]$\}}}||ks>t‚|  |||¡ q(W qW d S )N)r   r3   r\   r!   rm   )r   r   r   r[   Zpackage_rT   rU   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*   rK   )r   r   ÚprefixÚdry_run)ÚsysÚdont_write_bytecoderQ   Údistutils.utilr*   r   r4   Úsepr
   r   ro   r   )r   rA   r*   rn   r   r   r   r*   v  s    


zbuild_py.byte_compileN)r.   )Ú__name__Ú
__module__Ú__qualname__ZdescriptionZuser_optionsZboolean_optionsZnegative_optr   r%   r,   r   r8   r)   r3   rS   rV   r\   r_   ra   rb   rd   r+   rm   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   rm   rx   r9   )r   rT   rU   r   Zresr   r   r   rm   ›  s    zbuild_py_2to3.build_moduleN)rt   ru   rv   r,   rm   r   r   r   r   rw   Š  s   rw   )Ú__doc__r4   Úimportlib.utilrf   rp   r   Zdistutils.corer   Zdistutils.errorsrr   r   r   Z	distutilsr   r   rw   r   r   r   r   Ú<module>   s     }