a
    h                  	   @   s   d Z ddlmZ ddlmZ ddlmZmZmZmZm	Z	m
Z
mZ e
dZg dZddgZd	d	d	d
ee e	eee ef  e	eee ef  e	eegef  eed	d	f dddZeedddZd	S )z6Contains utilities to handle paths in Huggingface Hub.    fnmatch)Path)Callable	GeneratorIterableListOptionalTypeVarUnionT).gitz.git/*z*/.gitz
**/.git/**z.cache/huggingfacez.cache/huggingface/*z*/.cache/huggingfacez**/.cache/huggingface/**r   z.cacheN)allow_patternsignore_patternskey)itemsr   r   r   returnc                #   s   t |tr|g}t |tr |g}|dur6dd |D }|durLdd |D }|du rhttddd}|}| D ]R}|| |durt fdd	|D sql|durt fd
d	|D rql|V  qldS )a  Filter repo objects based on an allowlist and a denylist.

    Input must be a list of paths (`str` or `Path`) or a list of arbitrary objects.
    In the later case, `key` must be provided and specifies a function of one argument
    that is used to extract a path from each element in iterable.

    Patterns are Unix shell-style wildcards which are NOT regular expressions. See
    https://docs.python.org/3/library/fnmatch.html for more details.

    Args:
        items (`Iterable`):
            List of items to filter.
        allow_patterns (`str` or `List[str]`, *optional*):
            Patterns constituting the allowlist. If provided, item paths must match at
            least one pattern from the allowlist.
        ignore_patterns (`str` or `List[str]`, *optional*):
            Patterns constituting the denylist. If provided, item paths must not match
            any patterns from the denylist.
        key (`Callable[[T], str]`, *optional*):
            Single-argument function to extract a path from each item. If not provided,
            the `items` must already be `str` or `Path`.

    Returns:
        Filtered list of objects, as a generator.

    Raises:
        :class:`ValueError`:
            If `key` is not provided and items are not `str` or `Path`.

    Example usage with paths:
    ```python
    >>> # Filter only PDFs that are not hidden.
    >>> list(filter_repo_objects(
    ...     ["aaa.PDF", "bbb.jpg", ".ccc.pdf", ".ddd.png"],
    ...     allow_patterns=["*.pdf"],
    ...     ignore_patterns=[".*"],
    ... ))
    ["aaa.pdf"]
    ```

    Example usage with objects:
    ```python
    >>> list(filter_repo_objects(
    ... [
    ...     CommitOperationAdd(path_or_fileobj="/tmp/aaa.pdf", path_in_repo="aaa.pdf")
    ...     CommitOperationAdd(path_or_fileobj="/tmp/bbb.jpg", path_in_repo="bbb.jpg")
    ...     CommitOperationAdd(path_or_fileobj="/tmp/.ccc.pdf", path_in_repo=".ccc.pdf")
    ...     CommitOperationAdd(path_or_fileobj="/tmp/.ddd.png", path_in_repo=".ddd.png")
    ... ],
    ... allow_patterns=["*.pdf"],
    ... ignore_patterns=[".*"],
    ... key=lambda x: x.repo_in_path
    ... ))
    [CommitOperationAdd(path_or_fileobj="/tmp/aaa.pdf", path_in_repo="aaa.pdf")]
    ```
    Nc                 S   s   g | ]}t |qS  _add_wildcard_to_directories.0pr   r   X/var/www/html/assistant/venv/lib/python3.9/site-packages/huggingface_hub/utils/_paths.py
<listcomp>m       z'filter_repo_objects.<locals>.<listcomp>c                 S   s   g | ]}t |qS r   r   r   r   r   r   r   o   r   )itemr   c                 S   s4   t | tr| S t | tr t| S td|  dd S )Nz9Please provide `key` argument in `filter_repo_objects`: `z` is not a string.)
isinstancestrr   
ValueError)r   r   r   r   	_identitys   s
    

z&filter_repo_objects.<locals>._identityc                 3   s   | ]}t  |V  qd S Nr   r   rpathr   r   	<genexpr>   r   z&filter_repo_objects.<locals>.<genexpr>c                 3   s   | ]}t  |V  qd S r!   r   r"   r$   r   r   r&      r   )r   r   r   any)r   r   r   r   r    r   r   r$   r   filter_repo_objects'   s$    ?

r(   )patternr   c                 C   s   | d dkr| d S | S )N/*r   )r)   r   r   r   r      s    r   )__doc__r   pathlibr   typingr   r   r   r   r	   r
   r   r   ZDEFAULT_IGNORE_PATTERNSZFORBIDDEN_FOLDERSr   r(   r   r   r   r   r   <module>   s"   $c