#!/usr/bin/env python3
"""Little tools used by some functions of graph optimization."""
import functools
import typing
import networkx
[docs]
def node_selector(criteria: typing.Callable):
"""Decorate for selecting and appening the `node` kwargs to the decorated function.
Parameters
----------
criteria : callable
A function that take in argument
the graph (`networkx.networkx.MultiDiGraph`) and the node name (`str`).
It returns True if the node can be selected, False otherwise.
"""
assert hasattr(criteria, "__call__"), criteria.__class__.__name__
def decorator(func: typing.Callable):
@functools.wraps(func)
def decorated(graph: networkx.MultiDiGraph, *args, **kwargs):
while True:
for node in graph:
if criteria(graph, node):
func(graph, *args, node=node, **kwargs)
break
else:
break
return decorated
return decorator