# -*- coding:utf-8 -*-
#############################################################################
# Copyright (c) 2020 Huawei Technologies Co.,Ltd.
#
# openGauss is licensed under Mulan PSL v2.
# You can use this software according to the terms
# and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#
#          http://license.coscl.org.cn/MulanPSL2
#
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS,
# WITHOUT WARRANTIES OF ANY KIND,
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
# See the Mulan PSL v2 for more details.
#############################################################################
try:
    import sys

    sys.path.append(sys.path[0] + "/../../../")
    from gspylib.component.BaseComponent import BaseComponent
except ImportError as e:
    sys.exit("[GAUSS-52200] : Unable to import module: %s." % str(e))


class CM(BaseComponent):
    '''
    The class is used to define cluster manager component.
    '''

    def __init__(self):
        '''
        Constructor
        '''
        super(CM, self).__init__()
        self.timeout = 300
        self.level = 2


class CmResAttr():

    def __init__(self, script, res_type='DN'):
        self.resources_type = res_type
        self.script = script
        self.check_interval = 1
        self.time_out = 120
        self.restart_times = 5
        self.restart_delay = 1
        self.restart_period = 1

    def __str__(self):
        return str(vars(self)).replace(":", '=').replace('\'', '').replace(
            ' ', '').replace('{', '"').replace('}', '"')


class DssInstAttr():

    def __init__(self, node_id, dss_id, dss_home):
        self.node_id = node_id
        self.res_instance_id = dss_id
        self.res_args = dss_home

    def __str__(self):
        return str(vars(self)).replace(":", '=').replace('\'', '').replace(
            ' ', '').replace('{', '"').replace('}', '"').replace(';', ' ')


class CmResCtrlCmd():

    def __init__(self, action='add', name='', attr=''):
        self.action = action
        self.attr_name = name
        self.attr = attr

    def __str__(self):
        cmd = ''
        if self.action == 'add':
            cmd = 'cm_ctl res --add --res_name {} --res_attr={}'.format(
                self.attr_name, self.attr)
        elif self.action == 'edit':
            cmd = 'cm_ctl res --edit --res_name {} --add_inst={}'.format(
                self.attr_name, self.attr)
        return cmd