dycw-utilities 0.138.14__py3-none-any.whl → 0.138.16__py3-none-any.whl

{dycw_utilities-0.138.14.dist-info → dycw_utilities-0.138.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.138.14
+Version: 0.138.16
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.138.14.dist-info → dycw_utilities-0.138.16.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=cONkI_IFNfSqcsGihpBZx7cXfo2yNGxHqPOWbR64rSs,61
+utilities/__init__.py,sha256=UL4hpBrSmrPmk3IahPJQKeqtiFq6zh3H6Bk-mMUlBc0,61
 utilities/aiolimiter.py,sha256=mD0wEiqMgwpty4XTbawFpnkkmJS6R4JRsVXFUaoitSU,628
 utilities/altair.py,sha256=HeZBVUocjkrTNwwKrClppsIqgNFF-ykv05HfZSoHYno,9104
 utilities/asyncio.py,sha256=dcGeKQzjLBXxKzZkVIk5oZsFXEcynVbRB9iNB5XEDZk,38526
@@ -78,18 +78,18 @@ utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
 utilities/timer.py,sha256=oXfTii6ymu57niP0BDGZjFD55LEHi2a19kqZKiTgaFQ,2588
 utilities/traceback.py,sha256=YPD_MYNWc7k3vWF-YDWw4YUCZQyVR3ZLp6M1FAQR9Xw,8878
 utilities/typed_settings.py,sha256=C2i2VK62us_Z5jjca9NLmDFiT3_mkoSCwgU6FDtDhMI,4501
-utilities/types.py,sha256=-_pXQvmpJhTgEbI13N_zZWINdt0ODBpZ3WdJQUML9GA,17397
+utilities/types.py,sha256=WAdSOIGhnDOHIIggV8As8oOY9QrT06k3aIRC-yhgEH8,17489
 utilities/typing.py,sha256=Z-_XDaWyT_6wIo3qfNK-hvRlzxP2Jxa9PgXzm5rDYRA,13790
 utilities/tzdata.py,sha256=fgNVj66yUbCSI_-vrRVzSD3gtf-L_8IEJEPjP_Jel5Y,266
 utilities/tzlocal.py,sha256=KyCXEgCTjqGFx-389JdTuhMRUaT06U1RCMdWoED-qro,728
 utilities/uuid.py,sha256=32p7DGHGM2Btx6PcBvCZvERSWbpupMXqx6FppPoSoTU,612
 utilities/version.py,sha256=ufhJMmI6KPs1-3wBI71aj5wCukd3sP_m11usLe88DNA,5117
 utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
-utilities/whenever.py,sha256=WicmnXSFuZ64kemfULZGcV8Pt8-h6VBHJcObMHugSEo,22760
+utilities/whenever.py,sha256=4ZzVB6IOT4vTUbI0-0TgPgnqZouiZaLzLORza2jEB3Q,24210
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=oEH-nL3t4h9uawyZqWDtNtDAl6M-CLpLYGI_nI6DulM,1971
-dycw_utilities-0.138.14.dist-info/METADATA,sha256=mFTnGB2iv53YtekLXrv-Zc1Zh1HVuwHpU9ZHrd30YXE,1639
-dycw_utilities-0.138.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.138.14.dist-info/entry_points.txt,sha256=uLj5QWWVXv8tnMaRX3ZGYpt7w1xzLWU6LxbFhELEpkc,68
-dycw_utilities-0.138.14.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.138.14.dist-info/RECORD,,
+dycw_utilities-0.138.16.dist-info/METADATA,sha256=oiLhNbHe_C4E8EiVpOJkxjCsIBesSJHxhutHG7LVlQg,1639
+dycw_utilities-0.138.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.138.16.dist-info/entry_points.txt,sha256=uLj5QWWVXv8tnMaRX3ZGYpt7w1xzLWU6LxbFhELEpkc,68
+dycw_utilities-0.138.16.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.138.16.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.138.14"
+__version__ = "0.138.16"

utilities/types.py

@@ -88,6 +88,10 @@ class Dataclass(Protocol):
     __dataclass_fields__: ClassVar[dict[str, Any]]
 
 
+# datetime
+type MonthInt = Literal[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
+
+
 # enum
 type EnumLike[E: Enum] = MaybeStr[E]
 
@@ -270,6 +274,7 @@ __all__ = [
     "MaybeStr",
     "MaybeType",
     "MonthDayLike",
+    "MonthInt",
     "Number",
     "OpenMode",
     "OptExcInfo",

utilities/whenever.py

@@ -126,6 +126,21 @@ YEAR = DateDelta(years=1)
 ##
 
 
+def add_year_month(x: YearMonth, /, *, years: int = 0, months: int = 0) -> YearMonth:
+    """Add to a year-month."""
+    y = x.on_day(1) + DateDelta(years=years, months=months)
+    return y.year_month()
+
+
+def sub_year_month(x: YearMonth, /, *, years: int = 0, months: int = 0) -> YearMonth:
+    """Subtract from a year-month."""
+    y = x.on_day(1) - DateDelta(years=years, months=months)
+    return y.year_month()
+
+
+##
+
+
 def datetime_utc(
     year: int,
     month: int,
@@ -621,7 +636,39 @@ class ToNanosError(Exception):
 ##
 
 
-def to_py_time_delta(delta: DateDelta | TimeDelta | DateTimeDelta, /) -> dt.timedelta:
+@overload
+def to_py_date_or_date_time(date_or_date_time: Date, /) -> dt.date: ...
+@overload
+def to_py_date_or_date_time(date_or_date_time: ZonedDateTime, /) -> dt.datetime: ...
+@overload
+def to_py_date_or_date_time(date_or_date_time: None, /) -> None: ...
+def to_py_date_or_date_time(
+    date_or_date_time: Date | ZonedDateTime | None, /
+) -> dt.date | None:
+    """Convert a Date or ZonedDateTime into a standard library equivalent."""
+    match date_or_date_time:
+        case Date() as date:
+            return date.py_date()
+        case ZonedDateTime() as date_time:
+            return date_time.py_datetime()
+        case None:
+            return None
+        case _ as never:
+            assert_never(never)
+
+
+##
+
+
+@overload
+def to_py_time_delta(
+    delta: DateDelta | TimeDelta | DateTimeDelta, /
+) -> dt.timedelta: ...
+@overload
+def to_py_time_delta(delta: None, /) -> None: ...
+def to_py_time_delta(
+    delta: DateDelta | TimeDelta | DateTimeDelta | None, /
+) -> dt.timedelta | None:
     """Try convert a DateDelta to a standard library timedelta."""
     match delta:
         case DateDelta():
@@ -636,6 +683,8 @@ def to_py_time_delta(delta: DateDelta | TimeDelta | DateTimeDelta, /) -> dt.time
             return to_py_time_delta(delta.date_part()) + to_py_time_delta(
                 delta.time_part()
             )
+        case None:
+            return None
         case _ as never:
             assert_never(never)
 
@@ -853,6 +902,7 @@ __all__ = [
     "ToNanosError",
     "ToPyTimeDeltaError",
     "WheneverLogRecord",
+    "add_year_month",
     "datetime_utc",
     "diff_year_month",
     "format_compact",
@@ -865,12 +915,14 @@ __all__ = [
     "get_today_local",
     "mean_datetime",
     "min_max_date",
+    "sub_year_month",
     "to_date",
     "to_date_time_delta",
     "to_days",
     "to_local_plain",
     "to_months",
     "to_nanos",
+    "to_py_date_or_date_time",
     "to_py_time_delta",
     "to_zoned_date_time",
     "two_digit_year_month",