#!/usr/bin/env python

# matthew.lijun@huawei.com
# 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
import sys

from functest.core import testcase
from functest.opnfv_tests.openstack.tempest import conf_utils
from functest.utils import openstack_utils
from functest.utils.constants import CONST
import functest.utils.functest_logger as ft_logger

""" logging configuration """
logger = ft_logger.Logger("refstack_defcore").getLogger()


class TempestConf(object):
    def __init__(self):
        self.VERIFIER_ID = conf_utils.get_verifier_id()
        self.VERIFIER_REPO_DIR = conf_utils.get_verifier_repo_dir(
            self.VERIFIER_ID)
        self.DEPLOYMENT_ID = conf_utils.get_verifier_deployment_id()
        self.DEPLOYMENT_DIR = conf_utils.get_verifier_deployment_dir(
            self.VERIFIER_ID, self.DEPLOYMENT_ID)

    def generate_tempestconf(self):
        try:
            openstack_utils.source_credentials(CONST.openstack_creds)
            img_flavor_dict = conf_utils.create_tempest_resources(
                use_custom_images=True, use_custom_flavors=True)
            conf_utils.configure_tempest_defcore(
                self.DEPLOYMENT_DIR, img_flavor_dict)
        except KeyError as e:
            logger.error("defcore prepare env error with: %s", e)

    def main(self):
        try:
            self.generate_tempestconf()
            res = testcase.TestcaseBase.EX_OK
        except Exception as e:
            logger.error('Error with run: %s', e)
            res = testcase.TestcaseBase.EX_RUN_ERROR

        return res

if __name__ == '__main__':
    tempestconf = TempestConf()
    result = tempestconf.main()
    if result != testcase.TestcaseBase.EX_OK:
        sys.exit(result)