o
     JAf&C                     @   sz   d Z ddlZddlZddlZddl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)Command)*)convert_path	Mixin2to3)logc                   @   s   e Zd ZdZg d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d.d#d$Zd%d& Zd'd( Zd)d* Zd+d, Zd-S )/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)
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   r   c                 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   1/usr/lib/python3.10/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r/| jj D ]\}}t|| j|< q#|  | _	t
| jts`zt| j| _d| j  krMdksPJ  J W d S  ttfy_   tdw 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
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_outputsr   r   r   r   runC   s   zbuild_py.runc                    s   g }| j s|S | j D ]4}| |}tjj| jg|d  }d |r(t|d   fdd| ||D }|	||||f q
|S )z?Generate list of '(package,src_dir,build_dir,filenames)' tuples.r      c                    s   g | ]}| d  qS Nr   ).0fileZplenr   r   
<listcomp>s   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                    sd   | j dg | j |g  }g  |D ]}ttjt|t|}  fdd|D  q S )z6Return filenames for package's data files in 'src_dir' c                    s$   g | ]}| v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   getglobr7   r%   r8   escaper   extend)r   r   r>   ZglobspatternZfilelistr   rD   r   r;   y   s   zbuild_py.find_data_filesc                 C   s`   d}| j D ](\}}}}|D ]}tj||}| tj| | jtj|||dd qqdS )z$Copy data files into build directoryNFZ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 g }|rCz
| jd| }W n ty4   |d|d  |d= Y nw |d| tjj| S |s| jd}|durS|d| |r[tjj| S dS )zReturn 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/   rA   r   N)r9   r   r7   r%   r8   KeyErrorinsertrF   )r   r   r%   tailZpdirr   r   r   r6      s,   

zbuild_py.get_package_dirc                 C   sj   |dkrt j|std| t j|std| |r3t j|d}t j|r-|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%   existsZ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 )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t|d}g }tj| jj}|D ](}tj|}||kr@tj	tj
|d }||||f q| d|  q|S )Nz*.pyr   zexcluding %s)rY   rG   r7   r%   r8   rH   abspathr   Zscript_namesplitextbasenamer<   Z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 }| j D ]]}|d}d|dd }|d }z|| \}}W n ty3   | |}d}Y nw |sL| ||}	|df||< |	rL||d|	f tj||d }
| 	||
s\q||||
f q|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   rQ   r0   __init__.py)
r   r9   r8   rR   r6   rY   r<   r7   r%   r\   )r   r   r`   rZ   r%   r   Zmodule_baser   checkedrX   r[   r   r   r   find_modules   s,   


zbuild_py.find_modulesc                 C   sN   g }| j r||   | jr%| jD ]}| |}| ||}|| q|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   rI   re   r   r6   ra   )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 )rQ   r   )r2   rZ   r   r   r   r5   -  s    z-build_py.get_source_files.<locals>.<listcomp>)rg   r   r   r   r   get_source_files,  s   zbuild_py.get_source_filesc                 C   s$   |gt | |d g }tjj| S )Nrc   )listr7   r%   r8   )r   r?   r   rZ   Zoutfile_pathr   r   r   get_module_outfile/  s   zbuild_py.get_module_outfiler0   c                 C   s   |   }g }|D ]8\}}}|d}| | j||}|| |r@| jr/|tjj|dd | j	dkr@|tjj|| j	d q|dd | j
D 7 }|S )Nr/   rA   )optimizationr   c                 S   s,   g | ]\}}}}|D ]	}t j||q
qS r   )r7   r%   r8   )r2   r   r>   r?   r@   rO   r   r   r   r5   B  s    
z(build_py.get_outputs.<locals>.<listcomp>)rg   r9   rj   r   r<   r	   	importlibutilcache_from_sourcer   r   )r   r(   r`   Zoutputsr   rZ   r[   rO   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   rK   )r    strr9   ri   tuple	TypeErrorrj   r   r7   r%   rM   rL   rN   )r   rZ   r[   r   Zoutfiledirr   r   r   build_moduleJ  s   

zbuild_py.build_modulec                 C   s*   |   }|D ]\}}}| ||| qd S r1   )re   rs   )r   r`   r   rZ   r[   r   r   r   r)   Y  s   zbuild_py.build_modulesc                 C   sP   | j D ]"}| |}| ||}|D ]\}}}||ksJ | ||| qqd S r1   )r   r6   ra   rs   )r   r   r   r`   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r-||d| j	|| j
d | jdkr@||| j| j	|| j
d d S d S )Nz%byte-compiling is disabled, skipping.r   )r,   rQ   )r   r   prefixdry_run)sysdont_write_bytecoderW   distutils.utilr,   r   r7   sepr	   r   ru   r   )r   rE   r,   rt   r   r   r   r,   v  s    





zbuild_py.byte_compileN)r0   )__name__
__module____qualname__descriptionZuser_optionsZboolean_optionsZnegative_optr   r&   r.   r   r;   r+   r6   rY   r\   ra   re   rg   rh   rj   r-   rs   r)   r*   r,   r   r   r   r   r      s0    


'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 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 )Nr0   r   )r   rs   r   r<   )r   rZ   r[   r   resr   r   r   rs     s   zbuild_py_2to3.build_moduleN)rz   r{   r|   r.   rs   r   r   r   r   r~     s    r~   )__doc__r7   importlib.utilrl   rv   rG   Zdistutils.corer   Zdistutils.errorsrx   r   r   Z	distutilsr   r   r~   r   r   r   r   <module>   s      }