o
    Zh.                     @   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ddd	ee d
e	eee ef  de	eee ef  de	eegef  deeddf f
ddZdS )z6Contains utilities to handle paths in Huggingface Hub.    fnmatch)Path)Callable	GeneratorIterableListOptionalTypeVarUnionT)z.gitz.git/*z*/.gitz
**/.git/**N)allow_patternsignore_patternskeyitemsr   r   r   returnc                #   s    t |tr	|g}t |tr|g}|du r dtdtfdd}|}| D ])}|| |dur8t fdd|D s8q"|durHt fdd|D rHq"|V  q"d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")]
    ```
    Nitemr   c                 S   s0   t | tr| S t | trt| S td|  d)Nz9Please provide `key` argument in `filter_repo_objects`: `z` is not a string.)
isinstancestrr   
ValueError)r    r   S/var/www/html/lang_env/lib/python3.10/site-packages/huggingface_hub/utils/_paths.py	_identityb   s
   

z&filter_repo_objects.<locals>._identityc                 3       | ]}t  |V  qd S Nr   .0rpathr   r   	<genexpr>o       z&filter_repo_objects.<locals>.<genexpr>c                 3   r   r   r   r   r   r   r   r    s   r!   )r   r   r   any)r   r   r   r   r   r   r   r   r   filter_repo_objects   s    
?
r#   )__doc__r   pathlibr   typingr   r   r   r   r	   r
   r   r   ZIGNORE_GIT_FOLDER_PATTERNSr   r#   r   r   r   r   <module>   s(   $