from __future__ import annotations
from typing import Any, TYPE_CHECKING
import re
import contextlib
from com.sun.star.awt.tree import XMutableTreeNode
from ooodev.loader import lo as mLo
if TYPE_CHECKING:
from com.sun.star.awt.tree import XTreeNode
[docs]class RuleDataRegex:
"""Rule for matching a node's data value with a regular expression pattern."""
[docs] def __init__(self, regex: re.Pattern | str = "") -> None:
"""
Constructor
Arguments:
regex (re.Pattern | str): Regular expression pattern to match.
If empty string, then the ``match_value`` of ``is_match)()`` must be a regular expression pattern.
If included then the ``match_value`` of ``is_match()`` is ignored. Defaults to ``""``.
"""
if isinstance(regex, str):
if regex:
self._regex = re.compile(regex)
else:
self._regex = None
else:
self._regex = regex
[docs] def is_match(self, node: XTreeNode, match_value: Any) -> bool:
"""
Gets the node's text value and compares it with the match value.
Args:
node (XTreeNode): Tree node to check.
match_value (Any): Value to match. Must be a regular expression pattern to match.
Returns:
bool: True if the node's data value matches the match value; Otherwise, False.
"""
if self._regex:
rx = self._regex
else:
try:
if isinstance(match_value, str):
rx = re.compile(match_value)
else:
rx = match_value
except Exception:
return False
if not isinstance(rx, re.Pattern):
return False
tree_node = mLo.Lo.qi(XMutableTreeNode, node)
if not tree_node:
return False
data_value = tree_node.DataValue
if not data_value:
return False
if not isinstance(data_value, str):
return False
with contextlib.suppress(Exception):
return rx.search(data_value) is not None
return False