|
1
|
|
|
#!/usr/bin/env python3 |
|
2
|
|
|
|
|
3
|
|
|
import logging |
|
4
|
|
|
import os |
|
5
|
|
|
import os.path |
|
6
|
|
|
|
|
7
|
|
|
logging.basicConfig() |
|
8
|
|
|
logger = logging.getLogger("annif") |
|
9
|
|
|
logger.setLevel(level=logging.INFO) |
|
10
|
|
|
|
|
11
|
|
|
import annif.backend # noqa |
|
12
|
|
|
|
|
13
|
|
|
|
|
14
|
|
|
def create_flask_app(config_name=None): |
|
15
|
|
|
"""Create a Flask app to be used by the CLI.""" |
|
16
|
|
|
from flask import Flask |
|
17
|
|
|
|
|
18
|
|
|
app = Flask(__name__) |
|
19
|
|
|
config_name = _get_config_name(config_name) |
|
20
|
|
|
logger.debug(f"creating flask app with configuration {config_name}") |
|
21
|
|
|
app.config.from_object(config_name) |
|
22
|
|
|
app.config.from_envvar("ANNIF_SETTINGS", silent=True) |
|
23
|
|
|
return app |
|
24
|
|
|
|
|
25
|
|
|
|
|
26
|
|
|
def create_app(config_name=None): |
|
27
|
|
|
""" "Create a Connexion app to be used for the API.""" |
|
28
|
|
|
# 'cxapp' here is the Connexion application that has a normal Flask app |
|
29
|
|
|
# as a property (cxapp.app) |
|
30
|
|
|
import connexion |
|
31
|
|
|
from flask_cors import CORS |
|
32
|
|
|
|
|
33
|
|
|
from annif.openapi.validation import CustomRequestBodyValidator |
|
34
|
|
|
|
|
35
|
|
|
specdir = os.path.join(os.path.dirname(__file__), "openapi") |
|
36
|
|
|
cxapp = connexion.App(__name__, specification_dir=specdir) |
|
37
|
|
|
config_name = _get_config_name(config_name) |
|
38
|
|
|
logger.debug(f"creating connexion app with configuration {config_name}") |
|
39
|
|
|
cxapp.app.config.from_object(config_name) |
|
40
|
|
|
cxapp.app.config.from_envvar("ANNIF_SETTINGS", silent=True) |
|
41
|
|
|
|
|
42
|
|
|
validator_map = { |
|
43
|
|
|
"body": CustomRequestBodyValidator, |
|
44
|
|
|
} |
|
45
|
|
|
cxapp.add_api("annif.yaml", validator_map=validator_map) |
|
46
|
|
|
|
|
47
|
|
|
# add CORS support |
|
48
|
|
|
CORS(cxapp.app) |
|
49
|
|
|
|
|
50
|
|
|
if cxapp.app.config["INITIALIZE_PROJECTS"]: |
|
51
|
|
|
annif.registry.initialize_projects(cxapp.app) |
|
52
|
|
|
logger.info("finished initializing projects") |
|
53
|
|
|
|
|
54
|
|
|
# register the views via blueprints |
|
55
|
|
|
from annif.views import bp |
|
56
|
|
|
|
|
57
|
|
|
cxapp.app.register_blueprint(bp) |
|
58
|
|
|
|
|
59
|
|
|
# return the Flask app |
|
60
|
|
|
return cxapp.app |
|
61
|
|
|
|
|
62
|
|
|
|
|
63
|
|
|
def _get_config_name(config_name): |
|
64
|
|
|
if config_name is None: |
|
65
|
|
|
config_name = os.environ.get("ANNIF_CONFIG") |
|
66
|
|
|
if config_name is None: |
|
67
|
|
|
if os.environ.get("FLASK_RUN_FROM_CLI") == "true": |
|
68
|
|
|
config_name = "annif.default_config.Config" |
|
69
|
|
|
else: |
|
70
|
|
|
config_name = "annif.default_config.ProductionConfig" |
|
71
|
|
|
return config_name |
|
72
|
|
|
|