build.shared.colors

Complexity

Total Complexity

6

Size/Duplication

Total Lines	56
Duplicated Lines	0 %

Test Coverage

Coverage

87.5%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	Colors.__init__()	0	6	1
A	Colors._get_colors()	0	12	3
A	Colors.get_switch_color()	0	18	2

"""
    Module to interact with Coloring Napp.
"""


import json
import requests
from kytos.core import log
from napps.amlight.sdntrace import settings


class Colors(object):
    """ Class to handle the gathering of colors from
    amlight/coloring Napp

    """

    def __init__(self):
        """ Instantiate Colors and get list of colors
        """
        self._url = settings.COLORS_URL
        self._colors = dict()
        self._get_colors()

    def _get_colors(self):
        """ Get list of colors
        """
        try:
            result = requests.get(url=self._url)
            if result.status_code == 200:
                result = json.loads(result.content)
                self._colors = result['colors']
            else:
                raise Exception
        except Exception as err:  # pylint: disable=broad-except
            log.error('Error: Can not connect to Kytos/Coloring: %s' % err)

    def get_switch_color(self, dpid):
        """ Get the color_field and color_value of a specific
        switch. At every call, queries Coloring Napp to make sure
        colors reflects a possible topology.

        Args:
            dpid: switch.dpid

        Return:
            dict: {'color_field': str, 'color_value': str}
              or
            dict: {} if not found
        """
        self._get_colors()
        try:
            return self._colors[dpid]
        except KeyError:
            return {}