summaryrefslogtreecommitdiffstats
path: root/python/fatcat_tools/api_auth.py
blob: b36d467c89b9452293fd5e99e6b41322195125c6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

import sys
import fatcat_client
from fatcat_client.rest import ApiException


def public_api(host_uri):
    """
    Note: unlike the authenticated variant, this helper might get called even
    if the API isn't going to be used, so it's important that it doesn't try to
    actually connect to the API host or something.
    """
    conf = fatcat_client.Configuration()
    conf.host = host_uri
    return fatcat_client.DefaultApi(fatcat_client.ApiClient(conf))

def authenticated_api(host_uri, token=None):
    """
    Note: if this helper is called, it's implied that an actual API connection
    is needed, so it does try to connect and verify credentials.
    """

    conf = fatcat_client.Configuration()
    conf.host = host_uri
    if not token:
        token = sys.env['FATCAT_API_AUTH_TOKEN']
    if not token:
        sys.stderr.write(
            'This client requires a fatcat API token (eg, in env var FATCAT_API_AUTH_TOKEN)\n')
        sys.exit(-1)

    conf.api_key["Authorization"] = token
    conf.api_key_prefix["Authorization"] = "Bearer"
    api = fatcat_client.DefaultApi(fatcat_client.ApiClient(conf))

    # verify up front that auth is working
    api.check_auth()

    return api