nicegui/.github/workflows/update_citation.py

#!/usr/bin/env python3
import os
import sys
from pathlib import Path
from typing import Tuple

import requests
import yaml


def get_infos() -> Tuple[str,  str]:
    headers = {
        'Accept': 'application/json',
    }
    params = {
        'access_token': os.environ['ZENODO_TOKEN'],
        'q': 'nicegui',
        'sort': 'mostrecent',
        'status': 'published',
    }
    try:
        response = requests.get('https://zenodo.org/api/records', params=params, headers=headers, timeout=5)
        response.raise_for_status()
    # Hide all error details to avoid leaking the token
    except Exception:
        print('Error while getting the Zenodo infos')
        sys.exit(1)
    data = response.json()
    metadata = data['hits']['hits'][0]['metadata']
    return str(metadata['doi']), str(metadata['publication_date'])


if __name__ == '__main__':
    path = Path('CITATION.cff')
    citation = yaml.safe_load(path.read_text())
    citation['doi'], citation['date-released'] = get_infos()
    citation['version'] = sys.argv[1].removeprefix('v')
    path.write_text(yaml.dump(citation, sort_keys=False, default_flow_style=False))