B
    18™\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   ú^/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/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 )NÚbuild)r   r   )r   r   r   é   zoptimize must be 0, 1, or 2)Úset_undefined_optionsÚdistributionÚpackagesr   r   r   Úitemsr   Úget_data_filesÚ
data_filesÚ
isinstancer   ÚintÚAssertionErrorÚ
ValueErrorÚ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   )r4   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   )r7   r'   Úisfile)r2   Úfn)Úfilesr   r   r5   ‚   s    
z,build_py.find_data_files.<locals>.<listcomp>)r   Úgetr   r7   r'   r8   r   Úextend)r   r   r>   ÚglobsÚpatternÚfilelistr   )rD   r   r;   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    r7   r'   r8   Ú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).r0   rA   r   éÿÿÿÿN)r9   r   r7   r'   r8   ÚKeyErrorÚinsertrE   )r   r   r'   ÚtailZpdirr   r   r   r6      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 )NrA   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))	r7   r'   ÚexistsÚDistutilsFileErrorÚisdirr8   rB   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)r7   r'   rB   r   rW   )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)rY   r   r7   r'   r8   Úabspathr   Úscript_nameÚsplitextÚbasenamer<   Údebug_print)	r   r   r   Zmodule_filesÚmodulesZsetup_scriptr   Zabs_frZ   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.
        r0   r   rP   r1   Ú__init__z.py)
r   r9   r8   rQ   r6   rY   r<   r7   r'   r\   )r   r   rb   rZ   r'   r   Zmodule_baser   ÚcheckedrX   r[   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   rF   rf   r   r6   rc   )r   rb   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 )rP   r   )r2   rZ   r   r   r   r5   -  s    z-build_py.get_source_files.<locals>.<listcomp>)rh   )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)Úlistr7   r'   r8   )r   r?   r   rZ   Zoutfile_pathr   r   r   Úget_module_outfile/  s    zbuild_py.get_module_outfiler1   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 )Nr0   rA   )Úoptimizationr   c             S   s,   g | ]$\}}}}|D ]}t j ||¡‘qqS r   )r7   r'   r8   )r2   r   r>   r?   r@   rN   r   r   r   r5   C  s   
z(build_py.get_outputs.<locals>.<listcomp>)rh   r9   rk   r   r<   r
   Ú	importlibÚutilÚcache_from_sourcer   r    )r   r)   rb   Úoutputsr   rZ   r[   rN   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 )Nr0   z:'package' must be a string (dot-separated), list, or tupler   )rJ   )r!   Ústrr9   rj   ÚtupleÚ	TypeErrorrk   r   r7   r'   rL   rK   rM   )r   rZ   r[   r   ÚoutfileÚdirr   r   r   Úbuild_moduleJ  s    

zbuild_py.build_modulec             C   s.   |   ¡ }x |D ]\}}}|  |||¡ qW d S )N)rf   rv   )r   rb   r   rZ   r[   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   r6   rc   r#   rv   )r   r   r   rb   Zpackage_rZ   r[   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-   rP   )r   r   ÚprefixÚdry_run)ÚsysÚdont_write_bytecoderW   Údistutils.utilr-   r   r7   Úsepr
   r   rx   r   )r   rD   r-   rw   r   r   r   r-   v  s    


zbuild_py.byte_compileN)r1   )Ú__name__Ú
__module__Ú__qualname__ÚdescriptionÚuser_optionsÚboolean_optionsÚnegative_optr   r(   r/   r   r;   r,   r6   rY   r\   rc   rf   rh   ri   rk   r.   rv   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,   Úrun_2to3r-   r.   )r   r   r   r   r/   ‹  s    zbuild_py_2to3.runc             C   s,   t  | |||¡}|d r(| j |d ¡ |S )Nr1   r   )r   rv   r…   r<   )r   rZ   r[   r   Úresr   r   r   rv   ›  s    zbuild_py_2to3.build_moduleN)r}   r~   r   r/   rv   r   r   r   r   r„   Š  s   r„   )Ú__doc__r7   Úimportlib.utilrm   ry   r   Údistutils.corer   Údistutils.errorsr{   r   r   Ú	distutilsr   r   r„   r   r   r   r   Ú<module>   s     }