api/server.py

   1 ##############################################################################
   2 # Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9 import logging
  10 from itertools import ifilter
  11 import inspect
  12
  13 from flask import Flask
  14 from flask_restful import Api
  15 from flasgger import Swagger
  16
  17 from api.database import Base
  18 from api.database import engine
  19 from api.database import db_session
  20 from api.database import models
  21 from api.urls import urlpatterns
  22 from yardstick import _init_logging
  23
  24 logger = logging.getLogger(__name__)
  25
  26 app = Flask(__name__)
  27
  28 Swagger(app)
  29
  30 api = Api(app)
  31
  32
  33 @app.teardown_request
  34 def shutdown_session(exception=None):
  35     db_session.remove()
  36
  37
  38 def init_db():
  39     def func(a):
  40         try:
  41             if issubclass(a[1], Base):
  42                 return True
  43         except TypeError:
  44             pass
  45         return False
  46
  47     subclses = ifilter(func, inspect.getmembers(models, inspect.isclass))
  48     logger.debug('Import models: %s', [a[1] for a in subclses])
  49     Base.metadata.create_all(bind=engine)
  50
  51
  52 init_db()
  53 reduce(lambda a, b: a.add_resource(b.resource, b.url,
  54                                    endpoint=b.endpoint) or a, urlpatterns, api)
  55
  56 if __name__ == '__main__':
  57     _init_logging()
  58     logger.setLevel(logging.DEBUG)
  59     logger.info('Starting server')
  60     app.run(host='0.0.0.0')