o2app/entrypoints/flask_application.py

   1 # Copyright (C) 2021-2022 Wind River Systems, Inc.
   2 #
   3 #  Licensed under the Apache License, Version 2.0 (the "License");
   4 #  you may not use this file except in compliance with the License.
   5 #  You may obtain a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 #  Unless required by applicable law or agreed to in writing, software
  10 #  distributed under the License is distributed on an "AS IS" BASIS,
  11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 #  See the License for the specific language governing permissions and
  13 #  limitations under the License.
  14
  15 from flask import Flask
  16 from flask_restx import Api
  17
  18
  19 from o2app import bootstrap
  20 from o2ims.views import configure_namespace as ims_route_configure_namespace
  21 from o2common.views.route_exception import configure_exception
  22
  23 from o2common.authmw import authmiddleware
  24 from o2common.authmw import authprov
  25 from o2common.config.config import get_review_url
  26 from o2common.helper import o2logging
  27
  28 # apibase = config.get_o2ims_api_base()
  29 auth = True
  30 app = Flask(__name__)
  31 logger = o2logging.get_logger(__name__)
  32
  33
  34 def _get_k8s_url():
  35     try:
  36         token_review_url = get_review_url()
  37         return token_review_url
  38     except Exception:
  39         raise Exception('Get k8s token review url failed')
  40
  41
  42 FLASK_API_VERSION = '1.0.0'
  43
  44 if auth:
  45     # perform service account identity&privilege check.
  46     _get_k8s_url()
  47     ad = authprov.auth_definer('ad')
  48     ad.sanity_check()
  49     app.wsgi_app = authmiddleware.authmiddleware(app.wsgi_app)
  50
  51 app.config.SWAGGER_UI_DOC_EXPANSION = 'list'
  52 # app.config['RESTX_MASK_HEADER'] = 'fields'
  53 app.config['RESTX_MASK_SWAGGER'] = False
  54 app.config['ERROR_INCLUDE_MESSAGE'] = False
  55 api = Api(
  56     app, version=FLASK_API_VERSION,
  57     catch_all_404s=True,
  58     title='INF O2 Services API',
  59     description='Swagger OpenAPI document for the INF O2 Services',
  60           )
  61 bus = bootstrap.bootstrap()
  62
  63 configure_exception(api)
  64 ims_route_configure_namespace(api)