quick-start-connectors

1
import logging
2

3
from connexion.exceptions import Unauthorized
4
from flask import abort
5
from flask import current_app as app
6

7
from . import UpstreamProviderError, provider
8

9
logger = logging.getLogger(__name__)
10

11

12
def search(body):
13
    try:
14
        data = provider.search(body["query"])
15
    except UpstreamProviderError as error:
16
        logger.error(f"Upstream search error: {error.message}")
17
        abort(502, error.message)
18
    except AssertionError as error:
19
        logger.error(f"GDrive config error: {error}")
20
        abort(502, f"GDrive config error: {error}")
21

22
    return {"results": data}, 200, {"X-Connector-Id": app.config.get("APP_ID")}
23

24

25
def apikey_auth(token):
26
    api_key = app.config.get("CONNECTOR_API_KEY", "")
27
    if api_key != "" and token != api_key:
28
        raise Unauthorized()
29
    # successfully authenticated
30
    return {}
31