kytos-ng / pathfinder

utils.py

"""Utils module of kytos/pathfinder Kytos Network Application."""
# pylint: disable=unused-argument


def lazy_filter(filter_type, filter_func):
    """
    Lazy typed filter on top of the built-in function.

    It's meant to be used when the values to be filtered for
    are only defined later on dynamically at runtime.
    """

    def filter_closure(value, items):
        if not isinstance(value, filter_type):
            raise TypeError(f"Expected type: {filter_type}")
        return filter(filter_func(value), items)

    return filter_closure


def nx_edge_data_weight(edge_u, edge_v, edge_data):
    """Return custom edge data value to be used as a callback by nx."""
    if edge_data.get("hop"):
        return edge_data["hop"]
    return 1


def nx_edge_data_delay(edge_u, edge_v, edge_data):
    """Return custom edge data value to be used as a callback by nx."""
    if edge_data.get("delay"):
        return edge_data["delay"]
    return 1


def nx_edge_data_priority(edge_u, edge_v, edge_data):
    """Return custom edge data value to be used as a callback by nx."""
    if edge_data.get("priority"):
        return edge_data["priority"]
    return 1


def filter_le(metric):
    """Lazy filter_le."""
    return lambda x: (lambda nx_edge_tup: nx_edge_tup[2].get(metric, x) <= x)

The variable x does not seem to be defined.

def filter_ge(metric):
    """Lazy filter_ge."""
    return lambda x: (lambda nx_edge_tup: nx_edge_tup[2].get(metric, x) >= x)

The variable x does not seem to be defined.

def filter_in(metric):
    """Lazy filter_in."""
    return lambda x: (lambda nx_edge_tup: x in nx_edge_tup[2].get(metric, {x}))

The variable x does not seem to be defined.