none-shall-parse 0.4.8__tar.gz → 0.6.0__tar.gz

{none_shall_parse-0.4.8 → none_shall_parse-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: none-shall-parse
-Version: 0.4.8
+Version: 0.6.0
 Summary: Trinity Shared Python utilities.
 Author: Andries Niemandt, Jan Badenhorst
 Author-email: Andries Niemandt <andries.niemandt@trintel.co.za>, Jan Badenhorst <jan@trintel.co.za>

{none_shall_parse-0.4.8 → none_shall_parse-0.6.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "none-shall-parse"
-version = "0.4.8"
+version = "0.6.0"
 description = "Trinity Shared Python utilities."
 readme = "README.md"
 authors = [

{none_shall_parse-0.4.8 → none_shall_parse-0.6.0}/src/none_shall_parse/dates.py

@@ -3,7 +3,7 @@ import logging
 import time
 from datetime import date, datetime
 from typing import Callable, Any, Tuple, Sequence, List
-from .types import DateTimeLike
+from .types import DateTimeLike, DateTimeOrDateLike
 
 import pendulum
 from pendulum import DateTime, Date
@@ -417,7 +417,7 @@ def month_span(mso: int) -> Callable[[DateTimeLike], Tuple[DateTime, DateTime]]:
     return find_dates
 
 
-def arb_span(dates: Sequence[str | DateTimeLike], naive: bool = False) -> Callable[
+def arb_span(dates: Sequence[str | DateTimeOrDateLike], naive: bool = False) -> Callable[
     [Any], Tuple[DateTime, DateTime]]:
     """
     Parses two given dates and returns a callable function that provides the date range
@@ -452,15 +452,20 @@ def arb_span(dates: Sequence[str | DateTimeLike], naive: bool = False) -> Callab
 
                 parsed_dates.append(parsed.start_of('day'))
             else:
-                # It's already a datetime
-                if date.tzinfo is None:
-                    # Input is naive, keep it naive using pendulum.naive()
-                    parsed = pendulum.naive(date.year, date.month, date.day,
-                                            date.hour, date.minute, date.second,
-                                            date.microsecond)
+                # It's already a datetime-like object
+                if isinstance(date, datetime):
+                    # datetime objects have tzinfo, hour, minute, etc.
+                    if date.tzinfo is None:
+                        # Input is naive, keep it naive using pendulum.naive()
+                        parsed = pendulum.naive(date.year, date.month, date.day,
+                                                date.hour, date.minute, date.second,
+                                                date.microsecond)
+                    else:
+                        # Input is timezone-aware, preserve it
+                        parsed = pendulum.instance(date)
                 else:
-                    # Input is timezone-aware, preserve it
-                    parsed = pendulum.instance(date)
+                    # date objects (no time component, no tzinfo) - treat as naive
+                    parsed = pendulum.naive(date.year, date.month, date.day)
 
                 parsed_dates.append(parsed.start_of('day'))
 
@@ -518,13 +523,21 @@ def unroll_span_func(
         DateUtilsError: If the date range cannot be processed due to invalid dates or formatting.
     """
     cover = pendulum.now() if cover is None else cover
-    naive = cover.tzinfo is None
+
     try:
         start, end = f(cover)
         start = pendulum.instance(start)
-        start = start.naive() if naive else start
         end = pendulum.instance(end)
-        end = end.naive() if naive else end
+
+        # Determine naiveness from the dates returned by f, not from cover
+        naive = start.tzinfo is None
+        if naive:
+            start = start.naive()
+            end = end.naive()
+
+        # Get actual current time for filtering future dates (always "now", not cover)
+        current_date_sentinel = pendulum.now().naive() if naive else pendulum.now()
+
     except Exception as e:
         raise DateUtilsError(f"Function f failed to compute date range: {str(e)}")
 
@@ -537,9 +550,11 @@ def unroll_span_func(
         ord_days = []
         iso_date_strings = []
         for dt in interval.range(unit="days"):
-            date_range.append(dt)
-            ord_days.append(dt.day_of_year)
-            iso_date_strings.append(dt.format('YYYY-MM-DD'))
+            # Filter: only include dates up to today (not future dates)
+            if dt <= current_date_sentinel and dt < end:
+                date_range.append(dt)
+                ord_days.append(dt.day_of_year)
+                iso_date_strings.append(dt.format('YYYY-MM-DD'))
 
         return date_range, ord_days, iso_date_strings, start, end