klaatu-python 2.6.0__tar.gz → 2.7.0__tar.gz

{klaatu_python-2.6.0 → klaatu_python-2.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: klaatu-python
-Version: 2.6.0
+Version: 2.7.0
 Author: Robert Huselius
 Author-email: Robert Huselius <robert@huseli.us>
 License-Expression: GPL-3.0-only

{klaatu_python-2.6.0 → klaatu_python-2.7.0}/pyproject.toml

@@ -11,7 +11,7 @@ license-files = ["LICENSE"]
 name = "klaatu-python"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "2.6.0"
+version = "2.7.0"
 
 [project.urls]
 Homepage = "https://github.com/Eboreg/klaatu-python"

{klaatu_python-2.6.0 → klaatu_python-2.7.0}/src/klaatu_python/utils/__init__.py

@@ -1,6 +1,7 @@
 from .dicts import filter_values_not_null, group_by, group_dicts, zip_dict_lists, zip_dicts
 from .iterables import (
     circulate,
+    count_distinct,
     first_not_null,
     first_not_null_or_null,
     getitem0,
@@ -10,6 +11,7 @@ from .iterables import (
     most_common,
     nonulls,
     partition,
+    seq_contains_seq,
 )
 from .numbers import (
     can_coerce_to_int,
@@ -25,7 +27,7 @@ from .numbers import (
 )
 from .paths import deepest_common_path, sanitize_path
 from .strings import fancy_join
-from .time import daterange, round_up_timedelta
+from .time import date_to_timestamp_ms, daterange, round_to_whole_hour, round_up_timedelta
 from .urls import append_query_to_url, strip_url_query
 from .various import get_binary_reply, is_truthy
 
@@ -35,6 +37,8 @@ __all__ = [
     "can_coerce_to_int",
     "circulate",
     "coerce_between",
+    "count_distinct",
+    "date_to_timestamp_ms",
     "daterange",
     "deepest_common_path",
     "fancy_join",
@@ -58,9 +62,11 @@ __all__ = [
     "percent_rounded",
     "roman",
     "round_to_n",
+    "round_to_whole_hour",
     "round_up_timedelta",
     "rounded_percentage",
     "sanitize_path",
+    "seq_contains_seq",
     "strip_url_query",
     "to_int",
     "zip_dict_lists",

{klaatu_python-2.6.0 → klaatu_python-2.7.0}/src/klaatu_python/utils/iterables.py

@@ -20,6 +20,21 @@ def circulate(lst: Iterable[_T], rounds: int) -> list[_T]:
     return lst
 
 
+def count_distinct(values: Iterable[_HashableT]) -> dict[_HashableT, int]:
+    """
+    Returns a dict where the keys are distinct values from `values`, and the
+    values are the number of occurances of each value.
+    """
+    occurrences: dict[_HashableT, int] = {}
+
+    for value in values:
+        if value not in occurrences:
+            occurrences[value] = 0
+        occurrences[value] += 1
+
+    return occurrences
+
+
 def first_not_null(*values: _T | None) -> _T:
     ret = first_not_null_or_null(*values)
     if ret is None:
@@ -104,12 +119,7 @@ def most_common(values: Iterable[_HashableT]) -> tuple[_HashableT | None, int]:
 
     Given an _empty_ iterable, it will return `(None, 0)`.
     """
-    occurrences: dict[_HashableT, int] = {}
-
-    for value in values:
-        if value not in occurrences:
-            occurrences[value] = 0
-        occurrences[value] += 1
+    occurrences = count_distinct(values)
 
     if not occurrences:
         return None, 0

{klaatu_python-2.6.0 → klaatu_python-2.7.0}/src/klaatu_python/utils/time.py

@@ -1,8 +1,16 @@
+import datetime
 from collections.abc import Iterator
 from datetime import date, timedelta
 from math import ceil
 
 
+def date_to_timestamp_ms(date: datetime.date, tzinfo: datetime.tzinfo = datetime.timezone.utc) -> int:
+    epoch = datetime.datetime(1970, 1, 1, tzinfo=datetime.timezone.utc)
+    dt = datetime.datetime(date.year, date.month, date.day, tzinfo=tzinfo)
+
+    return int((dt - epoch).total_seconds() * 1000)
+
+
 def daterange(start_date: date, end_date: date) -> Iterator[date]:
     """
     Iterates the dates between `start_date` (inclusive) and `end_date`
@@ -12,6 +20,11 @@ def daterange(start_date: date, end_date: date) -> Iterator[date]:
         yield start_date + timedelta(days=n)
 
 
+def round_to_whole_hour(d: datetime.datetime | None = None) -> datetime.datetime:
+    d = d or datetime.datetime.now()
+    return datetime.datetime(year=d.year, month=d.month, day=d.day, hour=d.hour, tzinfo=d.tzinfo)
+
+
 def round_up_timedelta(td: timedelta) -> timedelta:
     """
     If td > 30 min, round up to nearest hour. Otherwise, to nearest 10