o
    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
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edef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rdd |D }|dur'dd |D }|du r6dtdtfdd}|}| D ])}|| |durNt fd	d
|D sNq8|dur^t fdd
|D r^q8|V  q8d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      g | ]}t |qS  _add_wildcard_to_directories.0pr   r   p/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/huggingface_hub/utils/_paths.py
<listcomp>m       z'filter_repo_objects.<locals>.<listcomp>c                 S   r   r   r   r   r   r   r   r   o   r   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   r   r   	_identitys   s
   

z&filter_repo_objects.<locals>._identityc                 3       | ]}t  |V  qd S Nr   r   rpathr   r   	<genexpr>       z&filter_repo_objects.<locals>.<genexpr>c                 3   r"   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+   patternc                 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   DEFAULT_IGNORE_PATTERNSFORBIDDEN_FOLDERSr   r+   r   r   r   r   r   <module>   s,   $
c