U
    ”Ê¹jæ#  ã                   @   s´   d Z ddlZddlZddlZddlmZ ddlmZ ddlm	Z	m
Z
 ddlmZ ddlmZ ejd	ejd
�Zeedœdd„Zeddœdd„Zeee
ee	f e
ee	f dœdd„ZdS )zDContains utilities to validate argument values in `huggingface_hub`.é    N)Úwraps)Úchain)ÚAnyÚDict)ÚHFValidationErroré   )Ú	CallableTzñ
    ^
    (\b[\w\-.]+\b/)? # optional namespace (username or organization)
    \b               # starts with a word boundary
    [\w\-.]{1,96}    # repo_name: alphanumeric + . _ -
    \b               # ends with a word boundary
    $
    )Úflags)ÚfnÚreturnc                    s:   t  ˆ¡‰dˆjkodˆjk‰ tˆƒ‡ ‡‡fdd„ƒ}|S )a­  Validate values received as argument for any public method of `huggingface_hub`.

    The goal of this decorator is to harmonize validation of arguments reused
    everywhere. By default, all defined validators are tested.

    Validators:
        - [`~utils.validate_repo_id`]: `repo_id` must be `"repo_name"`
          or `"namespace/repo_name"`. Namespace is a username or an organization.
        - [`~utils.smoothly_deprecate_use_auth_token`]: Use `token` instead of
          `use_auth_token` (only if `use_auth_token` is not expected by the decorated
          function - in practice, always the case in `huggingface_hub`).

    Example:
    ```py
    >>> from huggingface_hub.utils import validate_hf_hub_args

    >>> @validate_hf_hub_args
    ... def my_cool_method(repo_id: str):
    ...     print(repo_id)

    >>> my_cool_method(repo_id="valid_repo_id")
    valid_repo_id

    >>> my_cool_method("other..repo..id")
    huggingface_hub.utils._validators.HFValidationError: Cannot have -- or .. in repo_id: 'other..repo..id'.

    >>> my_cool_method(repo_id="other..repo..id")
    huggingface_hub.utils._validators.HFValidationError: Cannot have -- or .. in repo_id: 'other..repo..id'.

    >>> @validate_hf_hub_args
    ... def my_cool_auth_method(token: str):
    ...     print(token)

    >>> my_cool_auth_method(token="a token")
    "a token"

    >>> my_cool_auth_method(use_auth_token="a use_auth_token")
    "a use_auth_token"

    >>> my_cool_auth_method(token="a token", use_auth_token="a use_auth_token")
    UserWarning: Both `token` and `use_auth_token` are passed (...)
    "a token"
    ```

    Raises:
        [`~utils.HFValidationError`]:
            If an input is not valid.
    Úuse_auth_tokenÚtokenc                     sh   d}t tˆj| ƒ| ¡ ƒD ].\}}|dkr4t|ƒ q|dkr|d k	rd}qˆ r^tˆj||d�}ˆ| |ŽS )NF)Úrepo_idZfrom_idZto_idr   T)Úfn_nameÚ	has_tokenÚkwargs)r   ÚzipÚ
parametersÚitemsÚvalidate_repo_idÚ!smoothly_deprecate_use_auth_tokenÚ__name__)Úargsr   r   Zarg_nameÚ	arg_value©Zcheck_use_auth_tokenr
   Ú	signature© ú`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/huggingface_hub/utils/_validators.pyÚ	_inner_fnb   s    
þ
z'validate_hf_hub_args.<locals>._inner_fn)Úinspectr   r   r   )r
   r   r   r   r   Úvalidate_hf_hub_args*   s
    2
r    )r   r   c                 C   sš   t | tƒs$tdt| ƒ› d| › d�ƒ‚|  d¡dkrBtd| › d�ƒ‚t | ¡s\td| › d�ƒ‚d	| ksld
| kr|td| › d�ƒ‚|  d¡r–td| › d�ƒ‚dS )a”  Validate `repo_id` is valid.

    This is not meant to replace the proper validation made on the Hub but rather to
    avoid local inconsistencies whenever possible (example: passing `repo_type` in the
    `repo_id` is forbidden).

    Rules:
    - Between 1 and 96 characters.
    - Either "repo_name" or "namespace/repo_name"
    - [a-zA-Z0-9] or "-", "_", "."
    - "--" and ".." are forbidden

    Valid: `"foo"`, `"foo/bar"`, `"123"`, `"Foo-BAR_foo.bar123"`

    Not valid: `"datasets/foo/bar"`, `".repo_id"`, `"foo--bar"`, `"foo.git"`

    Example:
    ```py
    >>> from huggingface_hub.utils import validate_repo_id
    >>> validate_repo_id(repo_id="valid_repo_id")
    >>> validate_repo_id(repo_id="other..repo..id")
    huggingface_hub.utils._validators.HFValidationError: Cannot have -- or .. in repo_id: 'other..repo..id'.
    ```

    Discussed in https://github.com/huggingface/huggingface_hub/issues/1008.
    In moon-landing (internal repository):
    - https://github.com/huggingface/moon-landing/blob/main/server/lib/Names.ts#L27
    - https://github.com/huggingface/moon-landing/blob/main/server/views/components/NewRepoForm/NewRepoForm.svelte#L138
    zRepo id must be a string, not z: 'z'.ú/r   zCRepo id must be in the form 'repo_name' or 'namespace/repo_name': 'z&'. Use `repo_type` argument if needed.z‚Repo id must use alphanumeric chars, '-', '_' or '.'. The name cannot start or end with '-' or '.' and the maximum length is 96: 'z--z..z"Cannot have -- or .. in repo_id: 'z.gitzRepo_id cannot end by '.git': 'N)Ú
isinstanceÚstrr   ÚtypeÚcountÚREPO_ID_REGEXÚmatchÚendswith)r   r   r   r   r   w   s    

ÿ

ÿ
r   )r   r   r   r   c                 C   s@   |  ¡ }| dd¡}|dk	r<|r4t d| › d�¡ n||d< |S )ac  Smoothly deprecate `use_auth_token` in the `huggingface_hub` codebase.

    The long-term goal is to remove any mention of `use_auth_token` in the codebase in
    favor of a unique and less verbose `token` argument. This will be done a few steps:

    0. Step 0: methods that require a read-access to the Hub use the `use_auth_token`
       argument (`str`, `bool` or `None`). Methods requiring write-access have a `token`
       argument (`str`, `None`). This implicit rule exists to be able to not send the
       token when not necessary (`use_auth_token=False`) even if logged in.

    1. Step 1: we want to harmonize everything and use `token` everywhere (supporting
       `token=False` for read-only methods). In order not to break existing code, if
       `use_auth_token` is passed to a function, the `use_auth_token` value is passed
       as `token` instead, without any warning.
       a. Corner case: if both `use_auth_token` and `token` values are passed, a warning
          is thrown and the `use_auth_token` value is ignored.

    2. Step 2: Once it is release, we should push downstream libraries to switch from
       `use_auth_token` to `token` as much as possible, but without throwing a warning
       (e.g. manually create issues on the corresponding repos).

    3. Step 3: After a transitional period (6 months e.g. until April 2023?), we update
       `huggingface_hub` to throw a warning on `use_auth_token`. Hopefully, very few
       users will be impacted as it would have already been fixed.
       In addition, unit tests in `huggingface_hub` must be adapted to expect warnings
       to be thrown (but still use `use_auth_token` as before).

    4. Step 4: After a normal deprecation cycle (3 releases ?), remove this validator.
       `use_auth_token` will definitely not be supported.
       In addition, we update unit tests in `huggingface_hub` to use `token` everywhere.

    This has been discussed in:
    - https://github.com/huggingface/huggingface_hub/issues/1094.
    - https://github.com/huggingface/huggingface_hub/pull/928
    - (related) https://github.com/huggingface/huggingface_hub/pull/1064
    r   Nz1Both `token` and `use_auth_token` are passed to `z‚` with non-None values. `token` is now the preferred argument to pass a User Access Token. `use_auth_token` value will be ignored.r   )ÚcopyÚpopÚwarningsÚwarn)r   r   r   Z
new_kwargsr   r   r   r   r   ­   s    %
ÿ	r   )Ú__doc__r   Úrer+   Ú	functoolsr   Ú	itertoolsr   Útypingr   r   Zhuggingface_hub.errorsr   Z_typingr   ÚcompileÚVERBOSEr&   r    r#   r   Úboolr   r   r   r   r   Ú<module>   s   ÷M6