quick-start-connectors

1
import logging
2
from flask import abort, current_app as app
3
from connexion.exceptions import Unauthorized
4

5
from . import UpstreamProviderError, provider
6

7
logger = logging.getLogger(__name__)
8

9

10
def search(body):
11
    data = provider.search(body["query"])
12
    return {"results": data}, 200, {"X-Connector-Id": app.config.get("APP_ID")}
13

14

15
def search(body):
16
    logger.debug(f'Search request: {body["query"]}')
17

18
    try:
19
        data = provider.search(body["query"])
20
        logger.info(f"Found {len(data)} results")
21
    except UpstreamProviderError as error:
22
        logger.error(f"Upstream search error: {error.message}")
23
        abort(502, error.message)
24

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

27

28
def apikey_auth(token):
29
    if token != app.config.get("CONNECTOR_API_KEY"):
30
        raise Unauthorized()
31
    # successfully authenticated
32
    return {}
33