hfutils 0.2.1__tar.gz → 0.2.3__tar.gz

{hfutils-0.2.1/hfutils.egg-info → hfutils-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hfutils
-Version: 0.2.1
+Version: 0.2.3
 Summary: Useful utilities for huggingface
 Home-page: https://github.com/deepghs/hfutils
 Author: narugo1992

{hfutils-0.2.1 → hfutils-0.2.3}/hfutils/config/meta.py

@@ -7,7 +7,7 @@ Overview:
 __TITLE__ = 'hfutils'
 
 #: Version of this project.
-__VERSION__ = '0.2.1'
+__VERSION__ = '0.2.3'
 
 #: Short description of the project, will be included in ``setup.py``.
 __DESCRIPTION__ = 'Useful utilities for huggingface'

{hfutils-0.2.1 → hfutils-0.2.3}/hfutils/entry/cli.py

@@ -2,11 +2,13 @@ from .dispatch import hfutilcli
 from .download import _add_download_subcommand
 from .ls import _add_ls_subcommand
 from .upload import _add_upload_subcommand
+from .whoami import _add_whoami_subcommand
 
 _DECORATORS = [
     _add_download_subcommand,
     _add_upload_subcommand,
     _add_ls_subcommand,
+    _add_whoami_subcommand,
 ]
 
 cli = hfutilcli

hfutils-0.2.3/hfutils/entry/whoami.py

@@ -0,0 +1,50 @@
+import click
+from hbutils.string import plural_word
+from huggingface_hub.utils import LocalTokenNotFoundError
+
+from .base import CONTEXT_SETTINGS
+from ..operate.base import get_hf_client
+
+
+def _add_whoami_subcommand(cli: click.Group) -> click.Group:
+    """
+    Add the 'whoami' subcommand to the CLI.
+
+    This command displays the current identification.
+
+    :param cli: The Click CLI application.
+    :type cli: click.Group
+    :return: The modified Click CLI application.
+    :rtype: click.Group
+    """
+
+    @cli.command('whoami', help='See the current identification.\n\n'
+                                'Set environment $HF_TOKEN to use your own access token.',
+                 context_settings=CONTEXT_SETTINGS)
+    def whoami():
+        """
+        Display the current identification.
+
+        This function retrieves the current user's identification from the Hugging Face Hub API and displays it.
+
+        """
+        hf_client = get_hf_client()
+        try:
+            info = hf_client.whoami()
+            username = info['name']
+            click.echo(f'Hi, {click.style(f"@{username}", fg="green", bold=True)} '
+                       f'(full name: {click.style(info["fullname"], underline=True)}'
+                       f'{", PRO" if info["isPro"] else ""}).')
+            click.echo(f'You can access all resources with this identification.')
+            if info['orgs']:
+                click.echo(f'You have entered {plural_word(len(info["orgs"]), "organizations")}:')
+                for i, org_item in enumerate(info['orgs'], start=1):
+                    click.echo(f'{i}. {click.style("@" + org_item["name"], fg="blue", bold=True)} '
+                               f'(full name: {click.style(org_item["fullname"], underline=True)})')
+        except LocalTokenNotFoundError:
+            click.echo(f'Hi, {click.style("Guest", fg="yellow", bold=True)} (not authenticated).')
+            click.echo(click.style(f'No token for huggingface authentication found.', underline=True))
+            click.echo(f'If you need to access more, just set `HF_TOKEN` environment variable, '
+                       f'or use `huggingface-cli login`.')
+
+    return cli

{hfutils-0.2.1 → hfutils-0.2.3}/hfutils/index/fetch.py

@@ -240,6 +240,7 @@ def hf_tar_file_download(repo_id: str, archive_in_repo: str, file_in_archive: st
                                 f'in {repo_type}s/{repo_id}@{revision}/{archive_in_repo}.')
 
     info = files[_n_path(file_in_archive)]
+
     url_to_download = hf_hub_url(repo_id, archive_in_repo, repo_type=repo_type, revision=revision, endpoint=endpoint)
     headers = build_hf_headers(
         token=hf_token,
@@ -256,15 +257,16 @@ def hf_tar_file_download(repo_id: str, archive_in_repo: str, file_in_archive: st
         os.makedirs(os.path.dirname(local_file), exist_ok=True)
     try:
         with open(local_file, 'wb') as f:
-            http_get(
-                url_to_download,
-                f,
-                proxies=proxies,
-                resume_size=0,
-                headers=headers,
-                expected_size=info['size'],
-                displayed_filename=file_in_archive,
-            )
+            if info['size'] > 0:
+                http_get(
+                    url_to_download,
+                    f,
+                    proxies=proxies,
+                    resume_size=0,
+                    headers=headers,
+                    expected_size=info['size'],
+                    displayed_filename=file_in_archive,
+                )
 
         if os.path.getsize(local_file) != info['size']:
             raise ArchiveStandaloneFileIncompleteDownload(

{hfutils-0.2.1 → hfutils-0.2.3/hfutils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hfutils
-Version: 0.2.1
+Version: 0.2.3
 Summary: Useful utilities for huggingface
 Home-page: https://github.com/deepghs/hfutils
 Author: narugo1992

{hfutils-0.2.1 → hfutils-0.2.3}/hfutils.egg-info/SOURCES.txt

@@ -31,6 +31,7 @@ hfutils/entry/dispatch.py
 hfutils/entry/download.py
 hfutils/entry/ls.py
 hfutils/entry/upload.py
+hfutils/entry/whoami.py
 hfutils/index/__init__.py
 hfutils/index/fetch.py
 hfutils/index/hash.py