U
    Ha_4                     @   s   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mZm	Z	 ddl
mZ G dd dZdd	 ZG d
d deZejfddZdd ZdddZdS )zsdistutils.filelist

Provides the FileList class, used for poking about the filesystem
and building lists of files.
    Nconvert_path)DistutilsTemplateErrorDistutilsInternalError)logc                   @   s|   e Zd ZdZdddZdd Zejf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dZdS ) FileLista  A list of files built by on exploring the filesystem and filtered by
    applying various patterns to what we find there.

    Instance attributes:
      dir
        directory from which files will be taken -- only used if
        'allfiles' not supplied to constructor
      files
        list of filenames currently being built/filtered/manipulated
      allfiles
        complete list of files under consideration (ie. without any
        filtering applied)
    Nc                 C   s   d | _ g | _d S N)allfilesfiles)selfwarndebug_print r   B/tmp/pip-unpacked-wheel-i11yjail/setuptools/_distutils/filelist.py__init__    s    zFileList.__init__c                 C   s
   || _ d S r   )r	   )r   r	   r   r   r   set_allfiles&   s    zFileList.set_allfilesc                 C   s   t || _d S r   )findallr	   )r   dirr   r   r   r   )   s    zFileList.findallc                 C   s   ddl m} |rt| dS )z~Print 'msg' to stdout if the global DEBUG (taken from the
        DISTUTILS_DEBUG environment variable) flag is true.
        r   )DEBUGN)distutils.debugr   print)r   msgr   r   r   r   r   ,   s    zFileList.debug_printc                 C   s   | j | d S r   )r
   append)r   itemr   r   r   r   6   s    zFileList.appendc                 C   s   | j | d S r   )r
   extend)r   itemsr   r   r   r   9   s    zFileList.extendc                 C   s<   t ttjj| j}g | _|D ]}| jtjj|  qd S r   )sortedmapospathsplitr
   r   join)r   Zsortable_filesZ
sort_tupler   r   r   sort<   s    zFileList.sortc                 C   s@   t t| jd ddD ]$}| j| | j|d  kr| j|= qd S )N   r   )rangelenr
   )r   ir   r   r   remove_duplicatesE   s    zFileList.remove_duplicatesc                 C   s   |  }|d }d  } }}|dkrTt|dk r<td| dd |dd  D }n~|dkrt|d	k rttd
| t|d }dd |dd  D }n:|dkrt|dkrtd| t|d }ntd| ||||fS )Nr   )includeexcludeglobal-includeglobal-exclude   z&'%s' expects <pattern1> <pattern2> ...c                 S   s   g | ]}t |qS r   r   .0wr   r   r   
<listcomp>X   s     z1FileList._parse_template_line.<locals>.<listcomp>r#   )recursive-includerecursive-exclude   z,'%s' expects <dir> <pattern1> <pattern2> ...c                 S   s   g | ]}t |qS r   r   r.   r   r   r   r1   ^   s     )graftprunez#'%s' expects a single <dir_pattern>zunknown action '%s')r    r&   r   r   )r   linewordsactionpatternsr   dir_patternr   r   r   _parse_template_lineM   s0    zFileList._parse_template_linec                 C   sD  |  |\}}}}|dkrV| dd|  |D ]}| j|dds2td| q2n|dkr| dd|  |D ]}| j|ddsvtd	| qvn|d
kr| dd|  |D ]}| j|ddstd| qnb|dkr(| dd|  |D ]"}| j|dds td| q n|dkrz| d|d|f  |D ](}| j||dsNd}t||| qNn|dkr| d|d|f  |D ]$}| j||dstd|| qnx|dkr| d|  | jd |ds@td| nB|dkr4| d|  | jd |ds@td| ntd| d S )Nr)   zinclude  r#   )anchorz%warning: no files found matching '%s'r*   zexclude z9warning: no previously-included files found matching '%s'r+   zglobal-include r   z>warning: no files found matching '%s' anywhere in distributionr,   zglobal-exclude zRwarning: no previously-included files matching '%s' found anywhere in distributionr2   zrecursive-include %s %s)prefixz:warning: no files found matching '%s' under directory '%s'r3   zrecursive-exclude %s %szNwarning: no previously-included files matching '%s' found under directory '%s'r5   zgraft z+warning: no directories found matching '%s'r6   zprune z6no previously-included directories found matching '%s'z'this cannot happen: invalid action '%s')r<   r   r!   include_patternr   r   exclude_patternr   )r   r7   r9   r:   r   r;   patternr   r   r   r   process_template_linei   s~    





 


zFileList.process_template_liner#   r   c                 C   sl   d}t ||||}| d|j  | jdkr4|   | jD ],}||r:| d|  | j| d}q:|S )a  Select strings (presumably filenames) from 'self.files' that
        match 'pattern', a Unix-style wildcard (glob) pattern.  Patterns
        are not quite the same as implemented by the 'fnmatch' module: '*'
        and '?'  match non-special characters, where "special" is platform-
        dependent: slash on Unix; colon, slash, and backslash on
        DOS/Windows; and colon on Mac OS.

        If 'anchor' is true (the default), then the pattern match is more
        stringent: "*.py" will match "foo.py" but not "foo/bar.py".  If
        'anchor' is false, both of these will match.

        If 'prefix' is supplied, then only filenames starting with 'prefix'
        (itself a pattern) and ending with 'pattern', with anything in between
        them, will match.  'anchor' is ignored in this case.

        If 'is_regex' is true, 'anchor' and 'prefix' are ignored, and
        'pattern' is assumed to be either a string containing a regex or a
        regex object -- no translation is done, the regex is just compiled
        and used as-is.

        Selected strings will be added to self.files.

        Return True if files are found, False otherwise.
        Fz%include_pattern: applying regex r'%s'Nz adding T)translate_patternr   rB   r	   r   searchr
   r   )r   rB   r>   r?   is_regexfiles_found
pattern_renamer   r   r   r@      s    


zFileList.include_patternc                 C   sr   d}t ||||}| d|j  tt| jd ddD ]4}|| j| r8| d| j|   | j|= d}q8|S )a  Remove strings (presumably filenames) from 'files' that match
        'pattern'.  Other parameters are the same as for
        'include_pattern()', above.
        The list 'self.files' is modified in place.
        Return True if files are found, False otherwise.
        Fz%exclude_pattern: applying regex r'%s'r#   r$   z
 removing T)rD   r   rB   r%   r&   r
   rE   )r   rB   r>   r?   rF   rG   rH   r'   r   r   r   rA      s    zFileList.exclude_pattern)NN)r#   Nr   )r#   Nr   )__name__
__module____qualname____doc__r   r   r   curdirr   r   r   r   r"   r(   r<   rC   r@   rA   r   r   r   r   r      s    

	M
+     r   c                 C   s0   t tj| dd}dd |D }ttjj|S )z%
    Find all files under 'path'
    T)followlinksc                 s   s,   | ]$\}}}|D ]}t j||V  qqd S r   )r   r   r!   )r/   basedirsr
   filer   r   r   	<genexpr>   s    z#_find_all_simple.<locals>.<genexpr>)_UniqueDirsfilterr   walkr   isfile)r   Z
all_uniqueresultsr   r   r   _find_all_simple   s
    rY   c                   @   s$   e Zd ZdZdd Zedd ZdS )rT   z
    Exclude previously-seen dirs from walk results,
    avoiding infinite recursion.
    Ref https://bugs.python.org/issue44497.
    c                 C   sF   |\}}}t |}|j|jf}|| k}|r6|dd= | | | S )z
        Given an item from an os.walk result, determine
        if the item represents a unique dir for this instance
        and if not, prevent further traversal.
        N)r   statst_devst_inoadd)r   Z	walk_itemrP   rQ   r
   rZ   	candidatefoundr   r   r   __call__	  s    



z_UniqueDirs.__call__c                 C   s   t |  |S r   )rU   )clsr   r   r   r   rU     s    z_UniqueDirs.filterN)rJ   rK   rL   rM   r`   classmethodrU   r   r   r   r   rT     s   rT   c                 C   s6   t | }| tjkr.tjtjj| d}t||}t|S )z
    Find all files under 'dir' and return the list of full filenames.
    Unless dir is '.', return full filenames with dir prepended.
    )start)	rY   r   rN   	functoolspartialr   relpathr   list)r   r
   Zmake_relr   r   r   r     s
    

r   c                 C   s8   t | }tj}tjdkrd}d| }td||}|S )zTranslate a shell-like glob pattern to a regular expression; return
    a string containing the regex.  Differs from 'fnmatch.translate()' in
    that '*' does not match "special characters" (which are
    platform-specific).
    \z\\\\z\1[^%s]z((?<!\\)(\\\\)*)\.)fnmatch	translater   sepresub)rB   rH   rk   escapedr   r   r   
glob_to_re)  s    

ro   r#   c           
      C   s
  |rt | trt| S | S tdd\}}}| rVt| }||rP||sZtnd}|dk	rt|}||r~||st|t	|t	|t	|  }t
j}	t
jdkrd}	|t	|t	|t	|  }d|||	||f }n|r d||t	|d f }t|S )a  Translate a shell-like wildcard pattern to a compiled regular
    expression.  Return the compiled regex.  If 'is_regex' true,
    then 'pattern' is directly compiled to a regex (if it's a string)
    or just returned as-is (assumes it's a regex object).
    _ Nrh   z\\z%s\A%s%s.*%s%sz%s\A%s)
isinstancestrrl   compilero   	partition
startswithendswithAssertionErrorr&   r   rk   )
rB   r>   r?   rF   rc   rp   endrH   Z	prefix_rerk   r   r   r   rD   @  s6    


    rD   )r#   Nr   )rM   r   rl   ri   rd   distutils.utilr   distutils.errorsr   r   	distutilsr   r   rY   setrT   rN   r   ro   rD   r   r   r   r   <module>   s    f