#!/usr/bin/env python
# Copyright (c) 2012 OpenStack Foundation
# All Rights Reserved.
#
#  Licensed under the Apache License, Version 2.0 (the "License"); you may
#  not use this file except in compliance with the License. You may obtain
#  a copy of the License at
#
#  http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#  License for the specific language governing permissions and limitations
#  under the License.

import orm.base_config as config
from oslo_config import cfg
from sqlalchemy import *
import sys


def main(argv=None):

    if argv is None:
        argv = sys.argv
    cfg.CONF(argv[1:], project='ranger', validate_default_values=True)

    sql_queries = []

    orm_dbs = [
        config.ranger_base + '/orm/services/audit_trail_manager/scripts/db_scripts/create_db.sql',
        config.ranger_base + '/orm/services/id_generator/scripts/db_scripts/db_create.sql',
        config.ranger_base + '/orm/services/resource_distributor/scripts/db_scripts/create_db.sql',
        config.ranger_base + '/orm/services/region_manager/scripts/db_scripts/create_db.sql',
        config.ranger_base +
        '/orm/services/customer_manager/scripts/db_scripts/ranger_cms_create_db.sql',
        config.ranger_base +
        '/orm/services/customer_manager/scripts/db_scripts/ranger_cms_update_db.sql',
        config.ranger_base +
        '/orm/services/flavor_manager/scripts/db_scripts/ranger_fms_create_db.sql',
        config.ranger_base + '/orm/services/image_manager/scripts/db_scripts/create_db.sql'
    ]

    for item in range(len(orm_dbs)):
        sql_file = open(orm_dbs[item], "r")
        query = sql_file.read()
        sql_queries.append(query)
        sql_file.close()

    engine = create_engine(config.db_url, echo=False)

    for exec_item in range(len(sql_queries)):
        conn = engine.connect()
        exec_script = conn.execute(sql_queries[exec_item])
        conn.close()

    print 'Ranger databases setup complete'