Add query any_of method

Author: HexDecimal

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+
+- New query `.any_of` method. This was possible before but it is easier with this method.
+
 ## [5.2.4] - 2025-03-07
 
 ### Fixed

tcod/ecs/query.py

@@ -404,6 +404,30 @@ def none_of(
             & self._query,
         )
 
+    def any_of(
+        self,
+        components: Iterable[ComponentKey[object]] = (),
+        *,
+        tags: Iterable[object] = (),
+        relations: Iterable[_RelationQuery] = (),
+        traverse: Iterable[object] = (IsA,),
+        depth: int | None = None,
+    ) -> Self:
+        """Filter entities based on having at least one of the provided elements.
+
+        .. versionadded:: Unreleased
+        """
+        _check_suspicious_tags(tags, stacklevel=2)
+        return self.__class__(
+            self.registry,
+            _QueryLogicalAnd(
+                all_of=frozenset(
+                    [_QueryLogicalOr(any_of=frozenset(self.__as_queries(components, tags, relations, traverse, depth)))]
+                )
+            )
+            & self._query,
+        )
+
     def __iter__(self) -> Iterator[Entity]:
         """Iterate over the matching entities."""
         return iter(self.get_entities())

tests/test_ecs.py

@@ -282,3 +282,11 @@ def test_world_query_bool() -> None:
     assert not world.Q.all_of(tags=["Foo"])
     world[None].tags.add("Foo")
     assert world.Q.all_of(tags=["Foo"])
+
+
+def test_any_of() -> None:
+    world = tcod.ecs.Registry()
+    world[None].tags.add("foo")
+    assert world.Q.any_of(tags=["foo"])
+    assert world.Q.any_of(tags=["foo", "bar"])
+    assert not world.Q.any_of(tags=["bar"])