Source code for centreonapi.webservice.configuration.common

# -*- coding: utf-8 -*-

from centreonapi.webservice import Webservice


def build_param(param=None, objecttype=None, attr='name'):
    if param is None:
        return param

    param_list = []
    return_list = []
    if not isinstance(param, list):
        param_list.append(param)
    else:
        param_list = list(param)
    for k in param_list:
        if isinstance(k, str):
            return_list.append(k)
        elif isinstance(k, objecttype):
            return_list.append(k.__dict__[attr])
    return return_list


def build_param2(param=None, classname=None, attr='name'):
    if param is None:
        return param

    param_list = []
    return_list = []
    if not isinstance(param, list):
        param_list.append(param)
    else:
        param_list = list(param)
    for k in param_list:
        if isinstance(k, str):
            return_list.append(k)
        elif k.__class__.__name__ == classname:
            return_list.append(k.__dict__[attr])
    return return_list


class CentreonDecorator(object):

    @staticmethod
    def post_refresh(func):
        """
        Decorator that call __refresh_list() after func unless
        function's call contains 'post_refresh=False'

        eg:
        @post_refresh
        def hello(w, post_refresh=False):
            ...
        """
        def wrapper(*args, **kwargs):
            result = func(*args, **kwargs)
            # args[0] is always self
            # https://coderwall.com/p/jo39na/python-decorators-using-self
            if kwargs.get("post_refresh", True):
                args[0]._refresh_list()
            return result
        return wrapper

    @staticmethod
    def pre_refresh(func):
        def wrapper(*args, **kwargs):
            args[0]._refresh_list()
            return func(*args, **kwargs)
        return wrapper

    def _refresh_list(self):
        pass


class CentreonClass(object):

    def __init__(self):
        self.webservice = Webservice.getInstance()
        self.__clapi_action = ""

    def get(self, name):
        return self[name] or None

    def exists(self, name):
        return name in self

    def list(self):
        pass


class CentreonObject(object):

    def __repr__(self):
        return self.name

    def __str__(self):
        return self.name