improve component decorator check

Author: rmorshea

flake8_idom_hooks/utils.py

@@ -1,8 +1,12 @@
+import re
 import ast
 from contextlib import contextmanager
 from typing import Any, Iterator, List, Tuple
 
 
+_COMPONENT_DECORATOR_NAME_PATTERN = re.compile("^(idom.(\w+\.)*)?component$")
+
+
 @contextmanager
 def set_current(obj: Any, **attrs: Any) -> Iterator[None]:
     old_attrs = {k: getattr(obj, f"_current_{k}") for k in attrs}
@@ -28,17 +32,25 @@ def is_hook_def(node: ast.FunctionDef) -> bool:
 
 
 def is_component_def(node: ast.FunctionDef) -> bool:
-    return any(
-        is_idom_component_decorator(decorator) for decorator in node.decorator_list
-    )
+    return any(map(_is_component_decorator, node.decorator_list))
 
 
-def is_idom_component_decorator(node: Any) -> bool:
-    return getattr(node, "id", None) == "component" or (
-        getattr(getattr(node, "value", None), "id", None) == "idom"
-        and getattr(node, "attr", None) == "component"
+def is_hook_function_name(name: str) -> bool:
+    return name.lstrip("_").startswith("use_")
+
+
+def _is_component_decorator(node: ast.AST) -> bool:
+    return (
+        _COMPONENT_DECORATOR_NAME_PATTERN.match(
+            ".".join(reversed(list(_get_dotted_name(node))))
+        )
+        is not None
     )
 
 
-def is_hook_function_name(name: str) -> bool:
-    return name.lstrip("_").startswith("use_")
+def _get_dotted_name(node: ast.AST) -> Iterator[str]:
+    while isinstance(node, ast.Attribute):
+        yield node.attr
+        node = node.value
+    if isinstance(node, ast.Name):
+        yield node.id