# Template for UIState (Currently supports only colors)
UI_STATE_BODY = {"vis": {"colors": None}}


def generate(dash_config, viz_config):

    colors = {}

    # Check for 'color' key in 'series' or 'seriesParams' in
    # either viz_config and dash_config

    # Note:- 'series' simplifies 'seriesParams' and 'aggs'
    # and avoids duplication

    try:
        series = dash_config["y-axis"]["series"]
        for _, value in series.items():
            try:
                colors[value["label"]] = value["color"]
            except KeyError:
                continue
    except KeyError:
        pass

    try:
        series = viz_config["series"]
        for _, value in series.items():
            try:
                colors[value["label"]] = value["color"]
            except KeyError:
                continue
    except KeyError:
        pass

    try:
        seriesParams = dash_config["y-axis"]["seriesParams"]
        for _, value in seriesParams.items():
            try:
                colors[value["label"]] = value["color"]
            except KeyError:
                continue
    except KeyError:
        pass

    try:
        seriesParams = viz_config["seriesParams"]
        for _, value in seriesParams.items():
            try:
                colors[value["label"]] = value["color"]
            except KeyError:
                continue
    except KeyError:
        pass

    UI_STATE_BODY["vis"]["colors"] = colors

    return UI_STATE_BODY