dycw-utilities 0.175.17__py3-none-any.whl → 0.185.8__py3-none-any.whl

utilities/traceback.py

@@ -13,28 +13,22 @@ from sys import stderr
 from traceback import TracebackException
 from typing import TYPE_CHECKING, override
 
-from utilities.atomicwrites import writer
-from utilities.errors import repr_error
-from utilities.iterables import OneEmptyError, one
-from utilities.pathlib import module_path, to_path
-from utilities.reprlib import (
+from utilities.constants import (
+    LOCAL_TIME_ZONE_NAME,
     RICH_EXPAND_ALL,
     RICH_INDENT_SIZE,
     RICH_MAX_DEPTH,
     RICH_MAX_LENGTH,
     RICH_MAX_STRING,
     RICH_MAX_WIDTH,
-    yield_mapping_repr,
 )
+from utilities.core import OneEmptyError, get_now, get_now_local, one, write_text
+from utilities.errors import repr_error
+from utilities.pathlib import module_path, to_path
+from utilities.reprlib import yield_mapping_repr
 from utilities.text import to_bool
-from utilities.tzlocal import LOCAL_TIME_ZONE_NAME
-from utilities.version import to_version
-from utilities.whenever import (
-    format_compact,
-    get_now,
-    get_now_local,
-    to_zoned_date_time,
-)
+from utilities.version import to_version3
+from utilities.whenever import format_compact, to_zoned_date_time
 
 if TYPE_CHECKING:
     from collections.abc import Callable, Iterator
@@ -48,7 +42,7 @@ if TYPE_CHECKING:
         MaybeCallableZonedDateTimeLike,
         PathLike,
     )
-    from utilities.version import MaybeCallableVersionLike
+    from utilities.version import MaybeCallableVersion3Like
 
 
 ##
@@ -60,7 +54,7 @@ def format_exception_stack(
     *,
     header: bool = False,
     start: MaybeCallableZonedDateTimeLike = get_now,
-    version: MaybeCallableVersionLike | None = None,
+    version: MaybeCallableVersion3Like | None = None,
     capture_locals: bool = False,
     max_width: int = RICH_MAX_WIDTH,
     indent_size: int = RICH_INDENT_SIZE,
@@ -91,7 +85,7 @@ def format_exception_stack(
 def _yield_header_lines(
     *,
     start: MaybeCallableZonedDateTimeLike = get_now,
-    version: MaybeCallableVersionLike | None = None,
+    version: MaybeCallableVersion3Like | None = None,
 ) -> Iterator[str]:
     """Yield the header lines."""
     now = get_now_local()
@@ -102,7 +96,7 @@ def _yield_header_lines(
     yield f"User       | {getuser()}"
     yield f"Host       | {gethostname()}"
     yield f"Process ID | {getpid()}"
-    version_use = "" if version is None else to_version(version)
+    version_use = "" if version is None else to_version3(version)
     yield f"Version    | {version_use}"
     yield ""
 
@@ -197,7 +191,7 @@ def _trim_path(path: PathLike, pattern: str, /) -> Path | None:
 def make_except_hook(
     *,
     start: MaybeCallableZonedDateTimeLike = get_now,
-    version: MaybeCallableVersionLike | None = None,
+    version: MaybeCallableVersion3Like | None = None,
     path: MaybeCallablePathLike | None = None,
     path_max_age: Delta | None = None,
     max_width: int = RICH_MAX_WIDTH,
@@ -236,7 +230,7 @@ def _make_except_hook_inner(
     /,
     *,
     start: MaybeCallableZonedDateTimeLike = get_now,
-    version: MaybeCallableVersionLike | None = None,
+    version: MaybeCallableVersion3Like | None = None,
     path: MaybeCallablePathLike | None = None,
     path_max_age: Delta | None = None,
     max_width: int = RICH_MAX_WIDTH,
@@ -272,8 +266,7 @@ def _make_except_hook_inner(
             max_depth=max_depth,
             expand_all=expand_all,
         )
-        with writer(path_log, overwrite=True) as temp:
-            _ = temp.write_text(full)
+        write_text(path_log, full, overwrite=True)
         if path_max_age is not None:
             _make_except_hook_purge(path, path_max_age)
     if slack_url is not None:  # pragma: no cover

utilities/types.py

@@ -11,6 +11,7 @@ from re import Pattern
 from types import TracebackType
 from typing import (
     Any,
+    BinaryIO,
     ClassVar,
     Literal,
     Protocol,
@@ -65,6 +66,7 @@ type OpenMode = Literal[
 type MaybeCallable[T] = T | Callable[[], T]
 type MaybeStr[T] = T | str
 type MaybeType[T] = T | type[T]
+type StrDict = dict[str, Any]
 type StrMapping = Mapping[str, Any]
 type StrStrMapping = Mapping[str, str]
 type TupleOrStrMapping = tuple[Any, ...] | StrMapping
@@ -89,6 +91,11 @@ class SupportsKeysAndGetItem(Protocol[_T, _T_co]):
     def __getitem__(self, key: _T, /) -> _T_co: ...  # pragma: no cover
 
 
+# compression
+type Compression = Literal["bz2", "gzip", "lzma"]
+type PathToBinaryIO = Callable[[PathLike], BinaryIO]
+
+
 # concurrent
 type Parallelism = Literal["processes", "threads"]
 
@@ -98,7 +105,7 @@ type Parallelism = Literal["processes", "threads"]
 class Dataclass(Protocol):
     """Protocol for `dataclass` classes."""
 
-    __dataclass_fields__: ClassVar[dict[str, Any]]
+    __dataclass_fields__: ClassVar[StrDict]
 
 
 # datetime
@@ -127,7 +134,7 @@ type SequenceLT[T] = list[T] | tuple[T, ...]
 # iterables - dervied
 type MaybeSequence[T] = T | SequenceLT[T]
 type SequenceStr = SequenceLT[str]
-type CollectionStr = dict[str, Any] | frozenset[str] | set[str] | SequenceStr
+type CollectionStr = StrDict | frozenset[str] | set[str] | SequenceStr
 # iterables - maybe str
 type MaybeCollectionStr = str | CollectionStr
 type MaybeSequenceStr = str | SequenceStr
@@ -213,16 +220,26 @@ class SupportsRound(Protocol[_T_co]):
     def __round__(self, ndigits: int, /) -> _T_co: ...
 
 
+# os
+type CopyOrMove = Literal["copy", "move"]
+type IntOrAll = int | Literal["all"]
+
+
 # parse
 type ParseObjectExtra = Mapping[Any, Callable[[str], Any]]
 type SerializeObjectExtra = Mapping[Any, Callable[[Any], str]]
 
 
 # pathlib
+type FileOrDir = Literal["file", "dir"]
 type PathLike = MaybeStr[Path]
 type MaybeCallablePathLike = MaybeCallable[PathLike]
 
 
+# platform
+type System = Literal["windows", "mac", "linux"]
+
+
 # random
 type Seed = int | float | str | bytes | bytearray | Random
 
@@ -232,13 +249,17 @@ type PatternLike = MaybeStr[Pattern[str]]
 
 
 # retry
-type Retry = tuple[int, Delta | None]
+type Retry = tuple[int, Duration | None]
 
 
 # text
 type MaybeCallableStr = MaybeCallable[str]
 
 
+# time
+type Duration = Number | TimeDelta
+
+
 # traceback
 type ExcInfo = tuple[type[BaseException], BaseException, TracebackType]
 type OptExcInfo = ExcInfo | tuple[None, None, None]
@@ -249,6 +270,11 @@ type UUIDLike = MaybeStr[UUID]
 type MaybeCallableUUIDLike = MaybeCallable[UUIDLike | Seed]
 
 
+# warnings
+type FilterWarningsAction = Literal[
+    "error", "ignore", "always", "default", "module", "once"
+]
+
 # whenever
 type DateDeltaLike = MaybeStr[DateDelta]
 type DateLike = MaybeStr[Date]
@@ -292,6 +318,8 @@ type TimeZoneLike = (
 
 __all__ = [
     "TIME_ZONES",
+    "Compression",
+    "CopyOrMove",
     "Coro",
     "Dataclass",
     "DateDeltaLike",
@@ -300,11 +328,15 @@ __all__ = [
     "DateTimeDeltaLike",
     "DateTimeRoundMode",
     "Delta",
+    "Duration",
     "EnumLike",
     "ExcInfo",
     "ExceptionTypeLike",
+    "FileOrDir",
+    "FilterWarningsAction",
     "IPv4AddressLike",
     "IPv6AddressLike",
+    "IntOrAll",
     "LogLevel",
     "LoggerLike",
     "MathRoundMode",
@@ -334,6 +366,7 @@ __all__ = [
     "Parallelism",
     "ParseObjectExtra",
     "PathLike",
+    "PathToBinaryIO",
     "PatternLike",
     "PlainDateTimeLike",
     "Retry",
@@ -341,6 +374,7 @@ __all__ = [
     "SequenceStr",
     "SerializeObjectExtra",
     "Sign",
+    "StrDict",
     "StrMapping",
     "StrStrMapping",
     "SupportsAbs",
@@ -356,6 +390,7 @@ __all__ = [
     "SupportsLT",
     "SupportsRichComparison",
     "SupportsRound",
+    "System",
     "TimeDeltaLike",
     "TimeLike",
     "TimeOrDateTimeDelta",

utilities/typing.py

@@ -19,7 +19,6 @@ from typing import (
     TypeGuard,
     Union,  # pyright: ignore[reportDeprecated]
     _TypedDictMeta,  # pyright: ignore[reportAttributeAccessIssue]
-    cast,
     get_origin,
     overload,
     override,
@@ -40,10 +39,11 @@ from whenever import (
     ZonedDateTime,
 )
 
-from utilities.iterables import unique_everseen
-from utilities.sentinel import Sentinel
+from utilities.constants import Sentinel
+from utilities.core import unique_everseen
 from utilities.types import (
     Dataclass,
+    StrDict,
     StrMapping,
     StrStrMapping,
     TupleOrStrMapping,
@@ -145,7 +145,7 @@ def get_type_hints(
     globalns: StrMapping | None = None,
     localns: StrMapping | None = None,
     warn_name_errors: bool = False,
-) -> dict[str, Any]:
+) -> StrDict:
     """Get the type hints of an object."""
     _ = {
         Date,
@@ -170,7 +170,7 @@ def get_type_hints(
     }
     globalns_use = globals() | ({} if globalns is None else dict(globalns))
     localns_use = {} if localns is None else dict(localns)
-    result: dict[str, Any] = obj.__annotations__
+    result: StrDict = obj.__annotations__
     result = result | dict(get_forward_ref_args(obj))
     try:
         hints = _get_type_hints(
@@ -404,11 +404,8 @@ def _is_instance_typed_dict[T: _TypedDictMeta](
     localns: StrMapping | None = None,
     warn_name_errors: bool = False,
 ) -> TypeGuard[T]:
-    if not isinstance(obj, dict):
+    if not is_str_dict(obj):
         return False
-    if not all(isinstance(k, str) for k in obj):
-        return False
-    obj = cast("dict[str, Any]", obj)
     hints = get_type_hints(
         type_, globalns=globalns, localns=localns, warn_name_errors=warn_name_errors
     )
@@ -749,7 +746,15 @@ def is_set_type(obj: Any, /) -> bool:
 ##
 
 
-def is_string_mapping(obj: Any, /) -> TypeGuard[StrMapping]:
+def is_str_dict(obj: Any, /) -> TypeGuard[StrDict]:
+    """Check if an object is a string mapping."""
+    return isinstance(obj, Mapping) and is_iterable_of(obj, str)
+
+
+##
+
+
+def is_str_mapping(obj: Any, /) -> TypeGuard[StrMapping]:
     """Check if an object is a string mapping."""
     return isinstance(obj, Mapping) and is_iterable_of(obj, str)
 
@@ -852,7 +857,7 @@ def is_tuple(obj: Any, /) -> TypeGuard[tuple[Any, ...]]:
 
 def is_tuple_or_str_mapping(obj: Any, /) -> TypeGuard[TupleOrStrMapping]:
     """Check if an object is a tuple or string mapping."""
-    return is_tuple(obj) or is_string_mapping(obj)
+    return is_tuple(obj) or is_str_mapping(obj)
 
 
 ##
@@ -943,7 +948,8 @@ __all__ = [
     "is_sequence_of_tuple_or_str_mapping",
     "is_sequence_type",
     "is_set_type",
-    "is_string_mapping",
+    "is_str_dict",
+    "is_str_mapping",
     "is_subclass_gen",
     "is_tuple",
     "is_tuple_or_str_mapping",

utilities/uuid.py

@@ -5,8 +5,8 @@ from random import Random
 from typing import TYPE_CHECKING, assert_never, overload
 from uuid import UUID, uuid4
 
+from utilities.constants import Sentinel
 from utilities.random import get_state
-from utilities.sentinel import Sentinel
 
 if TYPE_CHECKING:
     from utilities.types import MaybeCallableUUIDLike, Seed