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 o2ims.adapter.clients.alarm_dict_client import load_alarm_definition
  24 from o2common.authmw import authmiddleware
  25 from o2common.authmw import authprov
  26 from o2common.config.config import get_review_url
  27 from o2common.helper import o2logging
  28
  29 # apibase = config.get_o2ims_api_base()
  30 auth = True
  31 app = Flask(__name__)
  32 logger = o2logging.get_logger(__name__)
  33
  34
  35 def _get_k8s_url():
  36     try:
  37         token_review_url = get_review_url()
  38         return token_review_url
  39     except Exception:
  40         raise Exception('Get k8s token review url failed')
  41
  42
  43 FLASK_API_VERSION = '1.0.0'
  44
  45 if auth:
  46     # perform service account identity&privilege check.
  47     _get_k8s_url()
  48     ad = authprov.auth_definer('ad')
  49     ad.sanity_check()
  50     app.wsgi_app = authmiddleware.authmiddleware(app.wsgi_app)
  51
  52 app.config.SWAGGER_UI_DOC_EXPANSION = 'list'
  53 # app.config['RESTX_MASK_HEADER'] = 'fields'
  54 app.config['RESTX_MASK_SWAGGER'] = False
  55 app.config['ERROR_INCLUDE_MESSAGE'] = False
  56 api = Api(app, version=FLASK_API_VERSION,
  57           title='INF O2 Services API',
  58           description='Swagger OpenAPI document for the INF O2 Services',
  59           )
  60 bus = bootstrap.bootstrap()
  61
  62 configure_exception(api)
  63 ims_route_configure_namespace(api)
  64
  65 load_alarm_definition(bus.uow)