Zenodo Python upload with progress bar

zenodo_upload_progress.py

import os

import requests  # pip install requests
from tqdm import tqdm  # pip install tqdm
from tqdm.utils import CallbackIOWrapper

# See: https://gist.github.com/slint/92e4d38eb49dd177f46b02e1fe9761e1
# See: https://gist.github.com/tyhoff/b757e6af83c1fd2b7b83057adf02c139

def get_bucket_url(deposit_id: str, access_token: str) -> str:
    """Gets the bucket URL for an existing deposit."""
    res = requests.get(
        f"https://zenodo.org/api/deposit/depositions/{deposit_id}", 
        json={},
        params={"access_token": access_token},
    )
    bucket_url = res.json()["links"]["bucket"]
    return bucket_url


def upload_file(
    file_path: str,
    deposit_id: str,
    access_token: str,
    progress_bar: bool = True,
) -> None:
    """Uploads a file to an existing Zenodo deposit.
    
    Args:
        file_path: Path to the file to upload.
        deposit_id: Deposit identifier, from https://zenodo.org/deposit/<deposit_id>.
        access_token: Zenodo API access token with "deposit:actions" and "deposit:write" scopes.
        progress_bar: Display a progress bar.
    """
    bucket_url = get_bucket_url(deposit_id, access_token)
    file_size = os.stat(file_path).st_size
    
    with open(file_path, "rb") as f:
        with tqdm(
            total=file_size,
            unit="B",
            unit_scale=True,
            unit_divisor=1024,
            disable=(not progress_bar),
        ) as pbar:
            wrapped_file = CallbackIOWrapper(pbar.update, f, "read")
            requests.put(
                f"{bucket_url}/{os.path.basename(file_path)}",
                data=wrapped_file,
                params={"access_token": access_token},
            )

Thanks for this!

params=params in the second last line should be params={'access_token': access_token}