pyintervals 1.0.3__tar.gz → 1.2.0__tar.gz

{pyintervals-1.0.3 → pyintervals-1.2.0}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: pyintervals
-Version: 1.0.3
+Version: 1.2.0
 Summary: Efficient interval operations.
 License: MIT
 Keywords: interval,timespan
 Author: Serkan Kalay
 Author-email: serkanosmankalay@gmail.com
-Requires-Python: >=3.8.1,<4.0.0
+Requires-Python: >=3.9,<4.0.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
@@ -17,6 +17,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: importlib-metadata (>=1,<5) ; python_version < "3.8"
 Requires-Dist: sortedcontainers (>=2.4.0,<3.0.0)
 Requires-Dist: sortedcontainers-stubs (>=2.4.2,<3.0.0)
+Requires-Dist: twine (>=6.2.0,<7.0.0)
 Project-URL: Repository, https://github.com/serkankalay/pyintervals
 Description-Content-Type: text/x-rst
 

{pyintervals-1.0.3 → pyintervals-1.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyintervals"
-version = "1.0.3"
+version = "1.2.0"
 description = "Efficient interval operations."
 authors = ["Serkan Kalay <serkanosmankalay@gmail.com>"]
 license = "MIT"
@@ -24,10 +24,11 @@ keywords = [
 ]
 
 [tool.poetry.dependencies]
-python = "^3.8.1,<4.0"
+python = ">=3.9,<4.0.0"
 importlib-metadata = {version = ">=1,<5", python = "<3.8"}
 sortedcontainers = "^2.4.0"
 sortedcontainers-stubs = "^2.4.2"
+twine = "^6.2.0"
 
 [tool.poetry.group.test.dependencies]
 pytest = ">=7.0.1,<9.0.0"
@@ -68,4 +69,4 @@ add_imports = ['from __future__ import annotations']
 
 [build-system]
 requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+build-backend = "poetry.core.masonry.api"

{pyintervals-1.0.3 → pyintervals-1.2.0}/src/pyintervals/constants.py

@@ -2,4 +2,4 @@ from __future__ import annotations
 
 from datetime import datetime
 
-TIME_ZERO: datetime = datetime(1970, 1, 1)
+TIME_ZERO: datetime = datetime.min

{pyintervals-1.0.3 → pyintervals-1.2.0}/src/pyintervals/interval.py

@@ -73,3 +73,15 @@ def contains(interval: Interval, other: Interval) -> bool:
 
     # We have at least one non-degenerate interval.
     return interval.start <= other.start and interval.end >= other.end
+
+
+def intersection(
+    interval: Interval, other_interval: Interval
+) -> Interval | None:
+    if overlaps(interval, other_interval):
+        return Interval(
+            max([interval.start, other_interval.start]),
+            min([interval.end, other_interval.end]),
+        )
+    else:
+        return None

{pyintervals-1.0.3 → pyintervals-1.2.0}/src/pyintervals/interval_handler.py

@@ -1,13 +1,14 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
-from datetime import datetime
+from datetime import datetime, timedelta
 from typing import Collection, Iterable, Sequence
+from zoneinfo import ZoneInfo
 
 from sortedcontainers import SortedList
 
 from .constants import TIME_ZERO
-from .interval import Interval
+from .interval import Interval, intersection
 from .search import weak_predecessor
 from .time_value_node import TimeValueNode, _simplify
 
@@ -59,19 +60,35 @@ def _relevant_nodes(
     )
 
 
+def _area_during_interval(
+    handler: IntervalHandler,
+    during: Interval,
+) -> timedelta:
+    return sum(
+        (
+            interval.value * during.value * overlap.duration()
+            for interval in handler.intervals
+            if (overlap := intersection(during, interval))
+        ),
+        start=timedelta(0),
+    )
+
+
 @dataclass
 class IntervalHandler:
     __intervals: list[Interval]
     __projection_graph: SortedList[TimeValueNode]
 
-    def __init__(self, intervals: Iterable[Interval] = []):
-        self._initialize()
+    def __init__(
+        self, intervals: Iterable[Interval] = [], tz: ZoneInfo | None = None
+    ):
+        self._initialize(tz)
         self.add(intervals)
 
-    def _initialize(self) -> None:
+    def _initialize(self, tz: ZoneInfo | None) -> None:
         self.__intervals = list()
         self.__projection_graph = SortedList(
-            [TimeValueNode(time_point=TIME_ZERO)]
+            [TimeValueNode(time_point=TIME_ZERO.replace(tzinfo=tz))]
         )
 
     @property
@@ -104,3 +121,6 @@ class IntervalHandler:
 
     def value_at_time(self, when: datetime) -> float:
         return _active_node_at_time(self.__projection_graph, when).value
+
+    def get_area(self, during: Interval) -> timedelta:
+        return _area_during_interval(self, during)