Source code for elasticai.creator.graph.subgraph_matching

from typing import TypeVar

from elasticai.creator.graph._types import NodeConstraintFn

from .graph import Graph
from .vf2 import find_all_matches, match

TP = TypeVar("TP")
T = TypeVar("T")



[docs]
class SubGraphMatchError(Exception):
    def __init__(self):
        super().__init__("No matching subgraph found")




[docs]
def find_subgraph(
    pattern: Graph[TP], graph: Graph[T], node_constraint: NodeConstraintFn[TP, T]
) -> dict[TP, T]:
    return match(pattern, graph, node_constraint)




[docs]
def find_all_subgraphs(
    pattern: Graph[TP], graph: Graph[T], node_constraint: NodeConstraintFn[TP, T]
) -> list[dict[TP, T]]:
    return find_all_matches(pattern, graph, node_constraint)