Merge pull request #130 from octoenergy/half-finite-range-set

Add HalfFiniteRangeSet (copy of FiniteRangeSet)

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## v4.9.2 - 2023-12-07
+
+- Add HalfFiniteRangeSet class to better support working with sets of such ranges
+
 ## v4.9.1 - 2023-11-20
 
 - Include py.typed files in package data to restore correct type checking [#127](https://github.com/octoenergy/xocto/pull/127)

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "xocto"
-version = "4.9.1"
+version = "4.9.2"
 requires-python = ">=3.9"
 description = "Kraken Technologies Python service utilities"
 readme = "README.md"

xocto/ranges.py

@@ -726,6 +726,27 @@ def __sub__(self, other: RangeSet[T]) -> RangeSet[T]:
         return self.difference(other)
 
 
+class HalfFiniteRangeSet(RangeSet[T], Generic[T]):
+    """
+    This subclass is useful when dealing with half-finite intervals as we can offer stronger guarantees
+    than we can get with normal RangeSets - mostly around intersections being half-finite etc.
+    """
+
+    _ranges: Sequence[HalfFiniteRange[T]]
+
+    def __iter__(self) -> Iterator[HalfFiniteRange[T]]:
+        yield from self._ranges
+
+    def intersection(self, other: Range[T] | RangeSet[T]) -> "HalfFiniteRangeSet[T]":
+        return cast("HalfFiniteRangeSet[T]", super().intersection(other))
+
+    def pop(self) -> HalfFiniteRange[T]:
+        return cast(HalfFiniteRange[T], super().pop())
+
+    def __and__(self, other: RangeSet[T]) -> "HalfFiniteRangeSet[T]":
+        return self.intersection(other)
+
+
 class FiniteRangeSet(RangeSet[T]):
     """
     This subclass is useful when dealing with finite intervals as we can offer stronger guarantees