figshare_client 0.0.2__tar.gz → 0.0.3__tar.gz

{figshare_client-0.0.2 → figshare_client-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: figshare_client
-Version: 0.0.2
+Version: 0.0.3
 Summary: A client to Figshare
 Keywords: snekpack,cookiecutter,figshare
 Author: Charles Tapley Hoyt

{figshare_client-0.0.2 → figshare_client-0.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "figshare_client"
-version = "0.0.2"
+version = "0.0.3"
 description = "A client to Figshare"
 readme = "README.md"
 authors = [
@@ -220,7 +220,7 @@ known-first-party = [
 docstring-code-format = true
 
 [tool.bumpversion]
-current_version = "0.0.2"
+current_version = "0.0.3"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)(?:-(?P<release>[0-9A-Za-z-]+(?:\\.[0-9A-Za-z-]+)*))?(?:\\+(?P<build>[0-9A-Za-z-]+(?:\\.[0-9A-Za-z-]+)*))?"
 serialize = [
     "{major}.{minor}.{patch}-{release}+{build}",

{figshare_client-0.0.2 → figshare_client-0.0.3}/src/figshare_client/api.py

@@ -1,6 +1,7 @@
 """Interact with the Figshare API."""
 
 from pathlib import Path
+from typing import Any
 
 import pystow
 import requests
@@ -39,7 +40,9 @@ def get_files(record_id: int) -> list[File]:
     return [File.model_validate(f) for f in res.json()]
 
 
-def ensure_files(record_id: int, *, concurrent: bool = True) -> dict[Path, File]:
+def ensure_files(
+    record_id: int, *, concurrent: bool = True, tqdm_kwargs: dict[str, Any] | None = None
+) -> dict[Path, File]:
     """Ensure all files for a record."""
     files = get_files(record_id)
     submodule = MODULE.module(str(record_id))
@@ -47,7 +50,15 @@ def ensure_files(record_id: int, *, concurrent: bool = True) -> dict[Path, File]
     def _func(file: File) -> tuple[Path, File]:
         return submodule.ensure(url=str(file.download_url), name=file.name), file
 
+    _tqdm_kwargs = {
+        "desc": f"downloading figshare:{record_id}",
+        "unit": "file",
+        "leave": False,
+    }
+    if tqdm_kwargs:
+        _tqdm_kwargs.update(tqdm_kwargs)
+
     if concurrent:
-        return dict(thread_map(_func, files))
+        return dict(thread_map(_func, files, **_tqdm_kwargs))
     else:
-        return dict(tmap(_func, files))
+        return dict(tmap(_func, files, **_tqdm_kwargs))

{figshare_client-0.0.2 → figshare_client-0.0.3}/src/figshare_client/version.py

@@ -12,7 +12,7 @@ __all__ = [
     "get_version",
 ]
 
-VERSION = "0.0.2"
+VERSION = "0.0.3"
 
 
 def get_git_hash() -> str: