Source code for app.model.lib.orcid

"""
ORCID documentation:
https://info.orcid.org/documentation/api-tutorials/api-tutorial-get-and-authenticated-orcid-id/
"""

import os
import requests




def get_login_url(orcid_client_id, app_host):
    params = {
        'client_id':     orcid_client_id,
        'response_type': 'code',
        'scope':         'openid',
        'redirect_uri':  f"https://{app_host}/login/",
    }

    query_string = '&'.join([f"{k}={v}" for k, v in params.items()])

    return f"{_get_orcid_root_url()}/oauth/authorize?{query_string}"





def authenticate_user(code, orcid_client_id, orcid_secret, app_host):
    url = f"{_get_orcid_root_url()}/oauth/token"
    data = {
        'client_id':     orcid_client_id,
        'client_secret': orcid_secret,
        'grant_type':    'authorization_code',
        'code':          code,
        'redirect_uri':  f"https://{app_host}/login/",
    }
    headers = {
        'Accept':       'application/json',
        'Content-Type': 'application/x-www-form-urlencoded',
    }

    r = requests.post(url, data=data, headers=headers)
    r.raise_for_status()

    return r.json()





def get_user_url(user):
    return f"{_get_orcid_root_url()}/{user.orcidId}"



def _get_orcid_root_url():
    app_env = os.getenv("APP_ENV", "development")

    if app_env in ('development', 'test'):
        return 'https://sandbox.orcid.org'
    elif app_env == 'production':
        return 'https://orcid.org'
    else:
        raise ValueError(f"Unknown app_env: {app_env}")