############################################################################## # Copyright (c) 2016 Huawei Technologies Co.,Ltd and others. # # All rights reserved. This program and the accompanying materials # are made available under the terms of the Apache License, Version 2.0 # which accompanies this distribution, and is available at # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## from __future__ import absolute_import import inspect import logging import socket from six.moves import filter from flasgger import Swagger from flask import Flask from flask_restful import Api from api.database import Base from api.database import db_session from api.database import engine from api.database.v1 import models from api.urls import urlpatterns from api import ApiResource from yardstick import _init_logging from yardstick.common import utils from yardstick.common import constants as consts try: from urlparse import urljoin except ImportError: from urllib.parse import urljoin logger = logging.getLogger(__name__) app = Flask(__name__) Swagger(app) api = Api(app) @app.teardown_request def shutdown_session(exception=None): db_session.remove() def get_resource(resource_name): name = ''.join(resource_name.split('_')) return next((r for r in utils.itersubclasses(ApiResource) if r.__name__.lower() == name)) def init_db(): def func(a): try: if issubclass(a[1], Base): return True except TypeError: pass return False subclses = filter(func, inspect.getmembers(models, inspect.isclass)) logger.debug('Import models: %s', [a[1] for a in subclses]) Base.metadata.create_all(bind=engine) def app_wrapper(*args, **kwargs): init_db() return app(*args, **kwargs) def get_endpoint(url): ip = socket.gethostbyname(socket.gethostname()) return urljoin('http://{}:{}'.format(ip, consts.API_PORT), url) for u in urlpatterns: api.add_resource(get_resource(u.target), u.url, endpoint=get_endpoint(u.url)) if __name__ == '__main__': _init_logging() logger.setLevel(logging.DEBUG) logger.info('Starting server') init_db() app.run(host='0.0.0.0')