mayutils 1.2.5__tar.gz → 1.2.6__tar.gz

{mayutils-1.2.5 → mayutils-1.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mayutils
-Version: 1.2.5
+Version: 1.2.6
 Summary: Utilities for Python from Mayuran Visakan
 Author: Mayuran Visakan
 Author-email: Mayuran Visakan <mayuran.k.v@gmail.com>

{mayutils-1.2.5 → mayutils-1.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mayutils"
-version = "1.2.5"
+version = "1.2.6"
 description = "Utilities for Python from Mayuran Visakan"
 authors = [
     { name = "Mayuran Visakan", email = "mayuran.k.v@gmail.com" }

{mayutils-1.2.5 → mayutils-1.2.6}/src/mayutils/objects/datetime.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 from contextlib import _GeneratorContextManager
 from sqlite3 import register_adapter
-from typing import Any, Optional, Self, Literal
+from typing import Any, Iterator, Optional, Self, Literal, overload
 import datetime as _datetime
 import numpy as np
 from pendulum import (
@@ -515,6 +515,46 @@ class Intervals(object):
 
         return self
 
+    def __repr__(
+        self,
+    ) -> str:
+        return f"Intervals(\n\t{'\n\t'.join([repr(interval) for interval in self.intervals])}\n)"
+
+    def __iter__(
+        self,
+    ) -> Iterator[Interval]:
+        return iter(self.intervals)
+
+    def __len__(
+        self,
+    ) -> int:
+        return len(self.intervals)
+
+    @overload
+    def __getitem__(
+        self,
+        key: int,
+    ) -> Interval: ...
+
+    @overload
+    def __getitem__(
+        self,
+        key: slice,
+    ) -> Intervals: ...
+
+    def __getitem__(
+        self,
+        key: int | slice,
+    ) -> Intervals | Interval:
+        if isinstance(key, slice):
+            return Intervals(*self.intervals[key])
+
+        elif isinstance(key, int):
+            return self.intervals[key]
+
+        else:
+            raise TypeError("Invalid key type")
+
 
 class Traveller(BaseTraveller):
     def __init__(