dycw-utilities 0.110.8__py3-none-any.whl → 0.111.0__py3-none-any.whl

{dycw_utilities-0.110.8.dist-info → dycw_utilities-0.111.0.dist-info}/METADATA

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

{dycw_utilities-0.110.8.dist-info → dycw_utilities-0.111.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=USEDB4wi-U2inTRQAJuZ7L5i_LnX5kcGM17wMf2dOck,60
+utilities/__init__.py,sha256=fjbCo5KvSkqKWLmPeMS8U77JZlYaYPd9js1ZVmdD1FE,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/astor.py,sha256=xuDUkjq0-b6fhtwjhbnebzbqQZAjMSHR1IIS5uOodVg,777
 utilities/asyncio.py,sha256=41oQUurWMvadFK5gFnaG21hMM0Vmfn2WS6OpC0R9mas,14757
@@ -12,13 +12,13 @@ utilities/contextvars.py,sha256=RsSGGrbQqqZ67rOydnM7WWIsM2lIE31UHJLejnHJPWY,505
 utilities/cryptography.py,sha256=HyOewI20cl3uRXsKivhIaeLVDInQdzgXZGaly7hS5dE,771
 utilities/cvxpy.py,sha256=Rv1-fD-XYerosCavRF8Pohop2DBkU3AlFaGTfD8AEAA,13776
 utilities/dataclasses.py,sha256=Q197PVnE_vUMn_SNnqJBCo4eRy4bdHtgMHWRbSJPtFk,26670
-utilities/datetime.py,sha256=GOs-MIEW_A49kzqa1yhIoeNeSqqPVgGO-h2AThtgTDk,37326
+utilities/datetime.py,sha256=XrVHvyHyYMBeU2ClC7xJKwB68l-F2ZCmDznyNXS4hT8,36862
 utilities/enum.py,sha256=HoRwVCWzsnH0vpO9ZEcAAIZLMv0Sn2vJxxA4sYMQgDs,5793
 utilities/errors.py,sha256=BtSNP0JC3ik536ddPyTerLomCRJV9f6kdMe6POz0QHM,361
 utilities/eventkit.py,sha256=6M5Xu1SzN-juk9PqBHwy5dS-ta7T0qA6SMpDsakOJ0E,13039
 utilities/fastapi.py,sha256=uwqOGbGwzIbP-lfm-ApG1ZEN3BA_TDsaiuTghhLmxb8,2413
 utilities/fpdf2.py,sha256=zM3gwOYcAfv7P4qhbyvzPmRY4PPAiAQ-ZnPC6I9SZ1M,1832
-utilities/functions.py,sha256=lAJeERNrmcpwon3drcTIlizLVRd8D-gdXojxtKFN0LM,28736
+utilities/functions.py,sha256=jgt592voaHNtX56qX0SRvFveVCRmSIxCZmqvpLZCnY8,27305
 utilities/functools.py,sha256=WrpHt7NLNWSUn9A1Q_ZIWlNaYZOEI4IFKyBG9HO3BC4,1643
 utilities/getpass.py,sha256=DfN5UgMAtFCqS3dSfFHUfqIMZX2shXvwphOz_6J6f6A,103
 utilities/git.py,sha256=wpt5dZ5Oi5931pN24_VLZYaQOvmR0OcQuVtgHzFUN1k,2359
@@ -39,14 +39,14 @@ utilities/more_itertools.py,sha256=CPUxrMAcTwRxbzbhiqPKi3Xx9hxqI0t6gkWjutaibGk,5
 utilities/numpy.py,sha256=cBgCBet8YfZP_rb4nkCJHZx9_03qPEinVENMk1dGVYQ,25683
 utilities/operator.py,sha256=0M2yZJ0PODH47ogFEnkGMBe_cfxwZR02T_92LZVZvHo,3715
 utilities/optuna.py,sha256=loyJGWTzljgdJaoLhP09PT8Jz6o_pwBOwehY33lHkhw,1923
-utilities/orjson.py,sha256=Wj5pzG_VdgoAy14a7Luhem-BgYrRtRFvvl_POiszRd0,36930
+utilities/orjson.py,sha256=DBm2zPP04kcHpY3l1etL24ksNynu-R3duFyx3U-RjqQ,36948
 utilities/os.py,sha256=D_FyyT-6TtqiN9KSS7c9g1fnUtgxmyMtzAjmYLkk46A,3587
-utilities/parse.py,sha256=fki2mnPGa1Ex-aeWiXDkUBHWb7FEk4F6AzMiHjqHXdw,19081
+utilities/parse.py,sha256=-sDMqA4nB4CFctc86uIlSoKmIUisTUd3qyBcuUnf1D0,18957
 utilities/pathlib.py,sha256=31WPMXdLIyXgYOMMl_HOI2wlo66MGSE-cgeelk-Lias,1410
-utilities/period.py,sha256=ikHXsWtDLr553cfH6p9mMaiCnIAP69B7q84ckWV3HaA,10884
+utilities/period.py,sha256=RWfcNVoNlW07RNdU47g_zuLZMKbtgfK4bE6G-9tVjY8,11024
 utilities/pickle.py,sha256=Bhvd7cZl-zQKQDFjUerqGuSKlHvnW1K2QXeU5UZibtg,657
 utilities/platform.py,sha256=NU7ycTvAXAG-fdYmDXaM1m4EOml2cGiaYwaUzfzSqyU,1767
-utilities/polars.py,sha256=woTGhyzXNLN30SQgwXz54-I1aJp1oWATJ2rpmke7gKI,58419
+utilities/polars.py,sha256=lr0l3JE2d5Rj1UbbFtagcv1z3Vw5YXjSfAwzVa9cFXw,58374
 utilities/polars_ols.py,sha256=efhXf0gjrHUpQrvS6a7g8yJQJWf_ATKtJnqqF2inCOU,5680
 utilities/pqdm.py,sha256=foRytQybmOQ05pjt5LF7ANyzrIa--4ScDE3T2wd31a4,3118
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -67,7 +67,7 @@ utilities/shelve.py,sha256=HZsMwK4tcIfg3sh0gApx4-yjQnrY4o3V3ZRimvRhoW0,738
 utilities/slack_sdk.py,sha256=SeDNMh24IPiEBWoGMdgvrflUaFa9TGlTS03H9-NKaQw,4132
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
 utilities/sqlalchemy.py,sha256=GWzp54TP3F2mGhxPTn0c56KxxDeN9VKLMagcRSELhf4,35453
-utilities/sqlalchemy_polars.py,sha256=oGyMX5gSxuLI3N8mtz_-ml3UdWKcZuj6aFRW6ifI0Kc,15617
+utilities/sqlalchemy_polars.py,sha256=wjJpoUo-yO9E2ujpG_06vV5r2OdvBiQ4yvV6wKCa2Tk,15605
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/streamlit.py,sha256=U9PJBaKP1IdSykKhPZhIzSPTZsmLsnwbEPZWzNhJPKk,2955
 utilities/sys.py,sha256=h0Xr7Vj86wNalvwJVP1wj5Y0kD_VWm1vzuXZ_jw94mE,2743
@@ -78,7 +78,7 @@ utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
 utilities/timer.py,sha256=Rkc49KSpHuC8s7vUxGO9DU55U9I6yDKnchsQqrUCVBs,4075
 utilities/traceback.py,sha256=KwHPLdEbdj0fFhXo8MBfxcvem8A-VXYDwFMNJ6f0cTM,27328
 utilities/types.py,sha256=kVY71hZkcnyYNIlYSse0mLm8yeP3OBkzhDPMME6jXxo,18126
-utilities/typing.py,sha256=WE3UWaVYO6nTkpEo8Ke3xUuxllsJtfOxsZ0-mSH69Nw,6668
+utilities/typing.py,sha256=jtc6EiGZGG0E745jo3NeLqo_HdHt7Zdaco3kCAEWIYU,11177
 utilities/tzdata.py,sha256=2ZsPmhTVM9Ptrxb4QrWKtKOB9RiH8IOO-A1u7ULdVbg,176
 utilities/tzlocal.py,sha256=42BCquGF54oIqIKe5RGziP4K8Nbm3Ey7uqcNn6m5ge8,534
 utilities/uuid.py,sha256=jJTFxz-CWgltqNuzmythB7iEQ-Q1mCwPevUfKthZT3c,611
@@ -87,7 +87,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=TjoTAJ1R27-rKXiXzdE4GzPidmYqm0W58XydDXp-QZM,17786
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-DQz5a0Ikw9jfSZtL0BEQkXOMC9yGn_xiJYNCLMiqEc,1989
-dycw_utilities-0.110.8.dist-info/METADATA,sha256=ZVWP8hzsSisjUrY3uoJqCIoh5j7pVhSy0PwDigCaJRw,13004
-dycw_utilities-0.110.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.110.8.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.110.8.dist-info/RECORD,,
+dycw_utilities-0.111.0.dist-info/METADATA,sha256=9UDVxMivd6ye3O57NEI2wEddqVpW9OE_rc_uhtMXD-c,13004
+dycw_utilities-0.111.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.111.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.111.0.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.110.8"
+__version__ = "0.111.0"

utilities/datetime.py

@@ -22,6 +22,7 @@ from utilities.math import SafeRoundError, round_, safe_round
 from utilities.platform import SYSTEM
 from utilities.sentinel import Sentinel, sentinel
 from utilities.types import MaybeStr
+from utilities.typing import is_instance_gen
 from utilities.zoneinfo import (
     UTC,
     HongKong,
@@ -140,9 +141,9 @@ def are_equal_dates_or_datetimes(
     x: DateOrDateTime, y: DateOrDateTime, /, *, strict: bool = False
 ) -> bool:
     """Check if x == y for dates/datetimes."""
-    if is_instance_date_not_datetime(x) and is_instance_date_not_datetime(y):
+    if is_instance_gen(x, dt.date) and is_instance_gen(y, dt.date):
         return x == y
-    if isinstance(x, dt.datetime) and isinstance(y, dt.datetime):
+    if is_instance_gen(x, dt.datetime) and is_instance_gen(y, dt.datetime):
         return are_equal_datetimes(x, y, strict=strict)
     raise AreEqualDatesOrDateTimesError(x=x, y=y)
 
@@ -210,7 +211,7 @@ def are_equal_months(x: DateOrMonth, y: DateOrMonth, /) -> bool:
 
 def check_date_not_datetime(date: dt.date, /) -> None:
     """Check if a date is not a datetime."""
-    if not is_instance_date_not_datetime(date):
+    if not is_instance_gen(date, dt.date):
         raise CheckDateNotDateTimeError(date=date)
 
 
@@ -604,14 +605,6 @@ YEAR = get_years(n=1)
 ##
 
 
-def is_instance_date_not_datetime(obj: Any, /) -> TypeGuard[dt.date]:
-    """Check if an object is a date, and not a datetime."""
-    return isinstance(obj, dt.date) and not isinstance(obj, dt.datetime)
-
-
-##
-
-
 def is_integral_timedelta(timedelta: dt.timedelta, /) -> bool:
     """Check if a timedelta is integral."""
     return (timedelta.seconds == 0) and (timedelta.microseconds == 0)
@@ -628,14 +621,6 @@ def is_local_datetime(obj: Any, /) -> TypeGuard[dt.datetime]:
 ##
 
 
-def is_subclass_date_not_datetime(cls: type[Any], /) -> TypeGuard[type[dt.date]]:
-    """Check if a class is a date, and not a datetime."""
-    return issubclass(cls, dt.date) and not issubclass(cls, dt.datetime)
-
-
-##
-
-
 _FRIDAY = 5
 
 
@@ -1359,10 +1344,8 @@ __all__ = [
     "get_today_local",
     "get_today_tokyo",
     "get_years",
-    "is_instance_date_not_datetime",
     "is_integral_timedelta",
     "is_local_datetime",
-    "is_subclass_date_not_datetime",
     "is_weekday",
     "is_zero_time",
     "is_zoned_datetime",

utilities/functions.py

@@ -22,7 +22,6 @@ from typing import (
     Literal,
     TypeGuard,
     TypeVar,
-    assert_never,
     cast,
     overload,
     override,
@@ -660,28 +659,6 @@ def is_hashable(obj: Any, /) -> TypeGuard[Hashable]:
 ##
 
 
-def is_instance_not_bool_int(
-    obj: Any, class_or_tuple: TypeLike[Any], /
-) -> TypeGuard[int]:
-    """Check if an instance relationship holds, except bool<int."""
-    match class_or_tuple:
-        case type() as type_:
-            return _is_instance_not_bool_int(obj, type_)
-        case tuple() as types:
-            return any(_is_instance_not_bool_int(obj, p) for p in types)
-        case _ as never:
-            assert_never(never)
-
-
-def _is_instance_not_bool_int(obj: Any, type_: type[Any], /) -> bool:
-    return isinstance(obj, type_) and not (
-        isinstance(obj, bool) and issubclass(type_, int) and not issubclass(type_, bool)
-    )
-
-
-##
-
-
 @overload
 def is_iterable_of(obj: Any, cls: type[_T], /) -> TypeGuard[Iterable[_T]]: ...
 @overload
@@ -796,28 +773,6 @@ def is_string_mapping(obj: Any, /) -> TypeGuard[StrMapping]:
 ##
 
 
-def is_subclass_not_bool_int(cls: type[Any], class_or_tuple: TypeLike[Any], /) -> bool:
-    """Check if a subclass relationship holds, except bool<int."""
-    match class_or_tuple:
-        case type() as parent:
-            return _is_subclass_int_not_bool_one(cls, parent)
-        case tuple() as parents:
-            return any(_is_subclass_int_not_bool_one(cls, p) for p in parents)
-        case _ as never:
-            assert_never(never)
-
-
-def _is_subclass_int_not_bool_one(cls: type[Any], parent: type[Any], /) -> bool:
-    return issubclass(cls, parent) and not (
-        issubclass(cls, bool)
-        and issubclass(parent, int)
-        and not issubclass(parent, bool)
-    )
-
-
-##
-
-
 def is_tuple(obj: Any, /) -> TypeGuard[tuple[Any, ...]]:
     """Check if an object is a tuple or string mapping."""
     return make_isinstance(tuple)(obj)
@@ -1075,7 +1030,6 @@ __all__ = [
     "is_dataclass_class",
     "is_dataclass_instance",
     "is_hashable",
-    "is_instance_not_bool_int",
     "is_iterable_of",
     "is_none",
     "is_not_none",
@@ -1083,7 +1037,6 @@ __all__ = [
     "is_sized",
     "is_sized_not_str",
     "is_string_mapping",
-    "is_subclass_not_bool_int",
     "is_tuple",
     "is_tuple_or_str_mapping",
     "make_isinstance",

utilities/orjson.py

@@ -692,6 +692,7 @@ _LOG_RECORD_DEFAULT_ATTRS = {
     "exc_text",
     "filename",
     "funcName",
+    "getMessage",
     "levelname",
     "levelno",
     "lineno",

utilities/parse.py

@@ -9,12 +9,7 @@ from re import DOTALL
 from types import NoneType
 from typing import TYPE_CHECKING, Any, override
 
-from utilities.datetime import (
-    is_instance_date_not_datetime,
-    is_subclass_date_not_datetime,
-)
 from utilities.enum import ParseEnumError, parse_enum
-from utilities.functions import is_subclass_not_bool_int
 from utilities.iterables import OneEmptyError, OneNonUniqueError, one, one_str
 from utilities.math import ParseNumberError, parse_number
 from utilities.re import ExtractGroupError, extract_group
@@ -36,10 +31,12 @@ from utilities.typing import (
     get_args,
     is_dict_type,
     is_frozenset_type,
+    is_instance_gen,
     is_list_type,
     is_literal_type,
     is_optional_type,
     is_set_type,
+    is_subclass_gen,
     is_tuple_type,
     is_union_type,
 )
@@ -169,12 +166,12 @@ def _parse_object_type(
             raise _ParseObjectParseError(type_=cls, text=text) from None
     if issubclass(cls, str):
         return text
-    if issubclass(cls, bool):
+    if is_subclass_gen(cls, bool):
         try:
             return parse_bool(text)
         except ParseBoolError:
             raise _ParseObjectParseError(type_=cls, text=text) from None
-    if is_subclass_not_bool_int(cls, int):
+    if is_subclass_gen(cls, int):
         try:
             return int(text)
         except ValueError:
@@ -201,14 +198,14 @@ def _parse_object_type(
             return parse_version(text)
         except ParseVersionError:
             raise _ParseObjectParseError(type_=cls, text=text) from None
-    if is_subclass_date_not_datetime(cls):
+    if is_subclass_gen(cls, dt.date):
         from utilities.whenever import ParseDateError, parse_date
 
         try:
             return parse_date(text)
         except ParseDateError:
             raise _ParseObjectParseError(type_=cls, text=text) from None
-    if issubclass(cls, dt.datetime):
+    if is_subclass_gen(cls, dt.datetime):
         from utilities.whenever import ParseDateTimeError, parse_datetime
 
         try:
@@ -473,11 +470,11 @@ def serialize_object(
         obj, bool | int | float | str | Path | Sentinel | Version
     ):
         return str(obj)
-    if is_instance_date_not_datetime(obj):
+    if is_instance_gen(obj, dt.date):
         from utilities.whenever import serialize_date
 
         return serialize_date(obj)
-    if isinstance(obj, dt.datetime):
+    if is_instance_gen(obj, dt.datetime):
         from utilities.whenever import serialize_datetime
 
         return serialize_datetime(obj)

utilities/period.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import datetime as dt
 from dataclasses import dataclass, field
 from functools import cached_property
+from itertools import permutations
 from typing import (
     TYPE_CHECKING,
     Generic,
@@ -15,10 +16,11 @@ from typing import (
     override,
 )
 
-from utilities.datetime import ZERO_TIME, is_instance_date_not_datetime
+from utilities.datetime import ZERO_TIME
 from utilities.functions import get_class_name
 from utilities.iterables import OneUniqueNonUniqueError, always_iterable, one_unique
 from utilities.sentinel import Sentinel, sentinel
+from utilities.typing import is_instance_gen
 from utilities.whenever import (
     serialize_date,
     serialize_local_datetime,
@@ -32,6 +34,7 @@ if TYPE_CHECKING:
     from utilities.iterables import MaybeIterable
     from utilities.types import DateOrDateTime
 
+
 type _DateOrDateTime = Literal["date", "datetime"]
 _TPeriod = TypeVar("_TPeriod", dt.date, dt.datetime)
 
@@ -54,9 +57,11 @@ class Period(Generic[_TPeriod]):
     max_duration: dt.timedelta | None = field(default=None, repr=False, kw_only=True)
 
     def __post_init__(self) -> None:
-        if is_instance_date_not_datetime(
-            self.start
-        ) is not is_instance_date_not_datetime(self.end):
+        if any(
+            is_instance_gen(left, cls) is not is_instance_gen(right, cls)
+            for left, right in permutations([self.start, self.end], 2)
+            for cls in [dt.date, dt.datetime]
+        ):
             raise _PeriodDateAndDateTimeMixedError(start=self.start, end=self.end)
         for date in [self.start, self.end]:
             if isinstance(date, dt.datetime):
@@ -166,7 +171,7 @@ class Period(Generic[_TPeriod]):
     @cached_property
     def kind(self) -> _DateOrDateTime:
         """The kind of the period."""
-        return "date" if is_instance_date_not_datetime(self.start) else "datetime"
+        return "date" if is_instance_gen(self.start, dt.date) else "datetime"
 
     def replace(
         self,

utilities/polars.py

@@ -56,7 +56,6 @@ from polars.exceptions import (
 from polars.testing import assert_frame_equal
 
 from utilities.dataclasses import _YieldFieldsInstance, yield_fields
-from utilities.datetime import is_instance_date_not_datetime
 from utilities.errors import ImpossibleCaseError
 from utilities.functions import (
     EnsureIntError,
@@ -95,6 +94,7 @@ from utilities.typing import (
     get_args,
     get_type_hints,
     is_frozenset_type,
+    is_instance_gen,
     is_list_type,
     is_literal_type,
     is_optional_type,
@@ -945,7 +945,7 @@ def dataclass_to_schema(
             dt.date,
             dt.datetime,
         }:
-            if is_instance_date_not_datetime(field.value):
+            if is_instance_gen(field.value, dt.date):
                 dtype = Date
             else:
                 dtype = _dataclass_to_schema_datetime(field)

utilities/sqlalchemy_polars.py

@@ -28,7 +28,6 @@ from sqlalchemy.exc import DuplicateColumnError
 from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
 
 from utilities.asyncio import timeout_dur
-from utilities.datetime import is_subclass_date_not_datetime
 from utilities.functions import identity
 from utilities.iterables import (
     CheckDuplicatesError,
@@ -49,6 +48,7 @@ from utilities.sqlalchemy import (
     upsert_items,
 )
 from utilities.text import snake_case
+from utilities.typing import is_subclass_gen
 from utilities.zoneinfo import UTC
 
 if TYPE_CHECKING:
@@ -201,14 +201,14 @@ def _insert_dataframe_check_df_and_db_types(
     dtype: PolarsDataType, db_col_type: type, /
 ) -> bool:
     return (
-        ((dtype == pl.Boolean) and issubclass(db_col_type, bool))
-        or ((dtype == Date) and is_subclass_date_not_datetime(db_col_type))
-        or ((dtype == Datetime) and issubclass(db_col_type, dt.datetime))
+        ((dtype == pl.Boolean) and is_subclass_gen(db_col_type, bool))
+        or ((dtype == Date) and is_subclass_gen(db_col_type, dt.date))
+        or ((dtype == Datetime) and is_subclass_gen(db_col_type, dt.datetime))
         or ((dtype == Float64) and issubclass(db_col_type, float))
-        or ((dtype == Int32) and issubclass(db_col_type, int))
-        or ((dtype == Int64) and issubclass(db_col_type, int))
-        or ((dtype == UInt32) and issubclass(db_col_type, int))
-        or ((dtype == UInt64) and issubclass(db_col_type, int))
+        or ((dtype == Int32) and is_subclass_gen(db_col_type, int))
+        or ((dtype == Int64) and is_subclass_gen(db_col_type, int))
+        or ((dtype == UInt32) and is_subclass_gen(db_col_type, int))
+        or ((dtype == UInt64) and is_subclass_gen(db_col_type, int))
         or ((dtype == String) and issubclass(db_col_type, str))
     )
 
@@ -414,15 +414,15 @@ def _select_to_dataframe_map_table_column_type_to_dtype(
     """Map a table column type to a polars type."""
     type_use = type_() if isinstance(type_, type) else type_
     py_type = type_use.python_type
-    if issubclass(py_type, bool):
+    if is_subclass_gen(py_type, bool):
         return pl.Boolean
     if issubclass(py_type, bytes):
         return Binary
     if issubclass(py_type, decimal.Decimal):
         return pl.Decimal
-    if issubclass(py_type, dt.date) and not issubclass(py_type, dt.datetime):
+    if is_subclass_gen(py_type, dt.date):
         return pl.Date
-    if issubclass(py_type, dt.datetime):
+    if is_subclass_gen(py_type, dt.datetime):
         has_tz: bool = type_use.timezone
         return zoned_datetime(time_zone=time_zone) if has_tz else Datetime()
     if issubclass(py_type, dt.time):
@@ -431,7 +431,7 @@ def _select_to_dataframe_map_table_column_type_to_dtype(
         return pl.Duration
     if issubclass(py_type, float):
         return Float64
-    if issubclass(py_type, int):
+    if is_subclass_gen(py_type, int):
         return Int64
     if issubclass(py_type, UUID | str):
         return String

utilities/typing.py

@@ -14,8 +14,10 @@ from typing import (
     Self,
     TypeAliasType,
     TypeGuard,
+    TypeVar,
     Union,  # pyright: ignore[reportDeprecated]
     get_origin,
+    overload,
     override,
 )
 from typing import get_args as _get_args
@@ -27,6 +29,16 @@ from utilities.iterables import unique_everseen
 from utilities.sentinel import Sentinel
 from utilities.types import StrMapping
 
+_T = TypeVar("_T")
+_T1 = TypeVar("_T1")
+_T2 = TypeVar("_T2")
+_T3 = TypeVar("_T3")
+_T4 = TypeVar("_T4")
+_T5 = TypeVar("_T5")
+
+
+##
+
 
 def contains_self(obj: Any, /) -> bool:
     """Check if an annotation contains `Self`."""
@@ -65,6 +77,52 @@ def _get_literal_elements_inner(obj: Any, /) -> list[Any]:
 ##
 
 
+def get_type_classes(obj: Any, /) -> tuple[type[Any], ...]:
+    """Get the type classes from a type/tuple/Union type."""
+    types: Sequence[type[Any]] = []
+    if isinstance(obj, type):
+        types.append(obj)
+    elif isinstance(obj, tuple):
+        for arg in obj:
+            if isinstance(arg, type):
+                types.append(arg)
+            elif isinstance(arg, tuple):
+                types.extend(get_type_classes(arg))
+            elif is_union_type(arg):
+                types.extend(get_union_type_classes(arg))
+            else:
+                raise _GetTypeClassesTupleError(obj=obj, inner=arg)
+    elif is_union_type(obj):
+        types.extend(get_union_type_classes(obj))
+    else:
+        raise _GetTypeClassesTypeError(obj=obj)
+    return tuple(types)
+
+
+@dataclass(kw_only=True, slots=True)
+class GetTypeClassesError(Exception):
+    obj: Any
+
+
+@dataclass(kw_only=True, slots=True)
+class _GetTypeClassesTypeError(GetTypeClassesError):
+    @override
+    def __str__(self) -> str:
+        return f"Object must be a type, tuple or Union type; got {self.obj}"
+
+
+@dataclass(kw_only=True, slots=True)
+class _GetTypeClassesTupleError(GetTypeClassesError):
+    inner: Any
+
+    @override
+    def __str__(self) -> str:
+        return f"Tuple must contain types, tuples or Union types only; got {self.inner}"
+
+
+##
+
+
 def get_type_hints(
     obj: Any,
     /,
@@ -96,8 +154,9 @@ def get_type_hints(
 
 
 def get_union_type_classes(obj: Any, /) -> tuple[type[Any], ...]:
+    """Get the type classes from a Union type."""
     if not is_union_type(obj):
-        raise _GetUnionTypeClassesNotAUnionTypeError(obj=obj)
+        raise _GetUnionTypeClassesUnionTypeError(obj=obj)
     types_: Sequence[type[Any]] = []
     for arg in get_args(obj):
         if isinstance(arg, type):
@@ -105,7 +164,7 @@ def get_union_type_classes(obj: Any, /) -> tuple[type[Any], ...]:
         elif is_union_type(arg):
             types_.extend(get_union_type_classes(arg))
         else:
-            raise _GetUnionTypeClassesNotATypeError(obj=obj, inner=arg)
+            raise _GetUnionTypeClassesInternalTypeError(obj=obj, inner=arg)
     return tuple(types_)
 
 
@@ -115,14 +174,14 @@ class GetUnionTypeClassesError(Exception):
 
 
 @dataclass(kw_only=True, slots=True)
-class _GetUnionTypeClassesNotAUnionTypeError(GetUnionTypeClassesError):
+class _GetUnionTypeClassesUnionTypeError(GetUnionTypeClassesError):
     @override
     def __str__(self) -> str:
         return f"Object must be a Union type; got {self.obj}"
 
 
 @dataclass(kw_only=True, slots=True)
-class _GetUnionTypeClassesNotATypeError(GetUnionTypeClassesError):
+class _GetUnionTypeClassesInternalTypeError(GetUnionTypeClassesError):
     inner: Any
 
     @override
@@ -149,6 +208,50 @@ def is_frozenset_type(obj: Any, /) -> bool:
 ##
 
 
+@overload
+def is_instance_gen(obj: Any, type_: type[_T], /) -> TypeGuard[_T]: ...
+@overload
+def is_instance_gen(obj: Any, type_: tuple[_T1], /) -> TypeGuard[_T1]: ...
+@overload
+def is_instance_gen(obj: Any, type_: tuple[_T1, _T2], /) -> TypeGuard[_T1 | _T2]: ...
+@overload
+def is_instance_gen(
+    obj: Any, type_: tuple[_T1, _T2, _T3], /
+) -> TypeGuard[_T1 | _T2 | _T3]: ...
+@overload
+def is_instance_gen(
+    obj: Any, type_: tuple[_T1, _T2, _T3, _T4], /
+) -> TypeGuard[_T1 | _T2 | _T3 | _T4]: ...
+@overload
+def is_instance_gen(
+    obj: Any, type_: tuple[_T1, _T2, _T3, _T4, _T5], /
+) -> TypeGuard[_T1 | _T2 | _T3 | _T4 | _T5]: ...
+@overload
+def is_instance_gen(obj: Any, type_: Any, /) -> bool: ...
+def is_instance_gen(obj: Any, type_: Any, /) -> bool:
+    """Check if an instance relationship holds, except bool<int."""
+    return any(_is_instance_gen_one(obj, t) for t in get_type_classes(type_))
+
+
+def _is_instance_gen_one(obj: Any, type_: type[_T], /) -> TypeGuard[_T]:
+    return (
+        isinstance(obj, type_)
+        and not (
+            isinstance(obj, bool)
+            and issubclass(type_, int)
+            and not issubclass(type_, bool)
+        )
+        and not (
+            isinstance(obj, dt.datetime)
+            and issubclass(type_, dt.date)
+            and not issubclass(type_, dt.datetime)
+        )
+    )
+
+
+##
+
+
 def is_list_type(obj: Any, /) -> bool:
     """Check if an object is a list type annotation."""
     return _is_annotation_of_type(obj, list)
@@ -222,6 +325,56 @@ def is_set_type(obj: Any, /) -> bool:
 ##
 
 
+@overload
+def is_subclass_gen(cls: type[Any], parent: type[_T], /) -> TypeGuard[type[_T]]: ...
+@overload
+def is_subclass_gen(
+    cls: type[Any], parent: tuple[type[_T1]], /
+) -> TypeGuard[type[_T1]]: ...
+@overload
+def is_subclass_gen(
+    cls: type[Any], parent: tuple[type[_T1], type[_T2]], /
+) -> TypeGuard[type[_T1 | _T2]]: ...
+@overload
+def is_subclass_gen(
+    cls: type[Any], parent: tuple[type[_T1], type[_T2], type[_T3]], /
+) -> TypeGuard[type[_T1 | _T2 | _T3]]: ...
+@overload
+def is_subclass_gen(
+    cls: type[Any], parent: tuple[type[_T1], type[_T2], type[_T3], type[_T4]], /
+) -> TypeGuard[type[_T1 | _T2 | _T3 | _T4]]: ...
+@overload
+def is_subclass_gen(
+    cls: type[Any],
+    parent: tuple[type[_T1], type[_T2], type[_T3], type[_T4], type[_T5]],
+    /,
+) -> TypeGuard[type[_T1 | _T2 | _T3 | _T4 | _T5]]: ...
+@overload
+def is_subclass_gen(cls: type[Any], parent: Any, /) -> bool: ...
+def is_subclass_gen(cls: type[Any], parent: Any, /) -> bool:
+    """Generalized `issubclass`."""
+    return any(_is_subclass_gen_one(cls, p) for p in get_type_classes(parent))
+
+
+def _is_subclass_gen_one(cls: type[Any], parent: type[_T], /) -> TypeGuard[type[_T]]:
+    return (
+        issubclass(cls, parent)
+        and not (
+            issubclass(cls, bool)
+            and issubclass(parent, int)
+            and not issubclass(parent, bool)
+        )
+        and not (
+            issubclass(cls, dt.datetime)
+            and issubclass(parent, dt.date)
+            and not issubclass(parent, dt.datetime)
+        )
+    )
+
+
+##
+
+
 def is_tuple_type(obj: Any, /) -> bool:
     """Check if an object is a tuple type annotation."""
     return _is_annotation_of_type(obj, tuple)
@@ -247,13 +400,16 @@ def _is_annotation_of_type(obj: Any, origin: Any, /) -> bool:
 
 
 __all__ = [
+    "GetTypeClassesError",
     "GetUnionTypeClassesError",
     "contains_self",
     "get_literal_elements",
+    "get_type_classes",
     "get_type_hints",
     "get_union_type_classes",
     "is_dict_type",
     "is_frozenset_type",
+    "is_instance_gen",
     "is_list_type",
     "is_literal_type",
     "is_mapping_type",
@@ -262,6 +418,7 @@ __all__ = [
     "is_optional_type",
     "is_sequence_type",
     "is_set_type",
+    "is_subclass_gen",
     "is_tuple_type",
     "is_union_type",
 ]