Merge pull request #15 from AbdelrhmanBassiouny/main

Role, BinaryPredicate, Performance

Author: tomsch420

examples/eql/old_examples/example_with_concatenate.md

@@ -9,7 +9,7 @@ Below is a minimal, self-contained example that mirrors the behavior tested in t
 
 ```python
 from dataclasses import dataclass, field
-from typing import List
+from typing_extensions import List
 
 from krrood.entity_query_language import symbolic_mode, let, concatenate, not_, in_, entity, an, From, symbol
 

examples/eql/old_examples/example_with_flatten.md

@@ -9,7 +9,7 @@ In this example, we place a single "cabinet" view into `world.views`. The cabine
 
 ```python
 from dataclasses import dataclass, field
-from typing import List
+from typing_extensions import List
 
 from krrood.entity_query_language import a, set_of, symbolic_mode, let, flatten, symbol
 

examples/eql/old_examples/example_with_for_all.md

@@ -6,7 +6,7 @@ Below is a minimal, self-contained example that mirrors the behavior used in tes
 
 ```python
 from dataclasses import dataclass, field
-from typing import List
+from typing_extensions import List
 
 from krrood.entity_query_language import symbolic_mode, let, entity, an, From, the, for_all, symbol
 

src/krrood/class_diagrams/attribute_introspector.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, Field
+from abc import ABC, abstractmethod
+from dataclasses import fields as dc_fields
+
+from typing_extensions import Dict, Iterable, List, Type, Optional, TYPE_CHECKING
+
+
+@dataclass
+class DiscoveredAttribute:
+    """Attribute discovered on a class."""
+
+    field: Field
+    """The dataclass field object that is wrapped."""
+    public_name: Optional[str] = None
+    """The public name of the field."""
+
+    def __post_init__(self):
+        if self.public_name is None:
+            self.public_name = self.field.name
+
+    def __hash__(self) -> int:
+        return hash(self.field)
+
+
+@dataclass
+class AttributeIntrospector(ABC):
+    """Strategy that discovers class attributes for diagramming.
+
+    Implementations return the set of dataclass-backed attributes that
+    should appear on a class diagram, including their public names.
+    """
+
+    @abstractmethod
+    def discover(self, owner_cls: Type) -> List[DiscoveredAttribute]:
+        """Return discovered attributes for `owner_cls`.
+
+        The `field` of each result must be a dataclass `Field` belonging to
+        `owner_cls`, while `public_name` is how it should be addressed and displayed.
+        """
+        raise NotImplementedError
+
+
+@dataclass
+class DataclassOnlyIntrospector(AttributeIntrospector):
+    """Discover only public dataclass fields (no leading underscore)."""
+
+    def discover(self, owner_cls: Type) -> List[DiscoveredAttribute]:
+        result: List[DiscoveredAttribute] = []
+        for f in dc_fields(owner_cls):
+            if not f.name.startswith("_"):
+                result.append(DiscoveredAttribute(public_name=f.name, field=f))
+        return result