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 o2dms.api import configure_namespace as dms_route_configure_namespace
  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 if auth:
  44     # perform service account identity&privilege check.
  45     _get_k8s_url()
  46     ad = authprov.auth_definer('ad')
  47     ad.sanity_check()
  48     app.wsgi_app = authmiddleware.authmiddleware(app.wsgi_app)
  49
  50 app.config.SWAGGER_UI_DOC_EXPANSION = 'list'
  51 # app.config['RESTX_MASK_HEADER'] = 'fields'
  52 app.config['RESTX_MASK_SWAGGER'] = False
  53 api = Api(app, version='1.0.0',
  54           title='INF O2 Services API',
  55           description='Swagger OpenAPI document for the INF O2 Services',
  56           )
  57 bus = bootstrap.bootstrap()
  58
  59 ims_route_configure_namespace(api)
  60 dms_route_configure_namespace(api)
  61
  62 load_alarm_definition(bus.uow)