dycw-utilities 0.114.3__py3-none-any.whl → 0.114.5__py3-none-any.whl

{dycw_utilities-0.114.3.dist-info → dycw_utilities-0.114.5.dist-info}/METADATA

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

{dycw_utilities-0.114.3.dist-info → dycw_utilities-0.114.5.dist-info}/RECORD

@@ -1,7 +1,7 @@
-utilities/__init__.py,sha256=SWDkfIOyhgygmyWsgThn0Boll445H7CW7YAsVtIjpMY,60
+utilities/__init__.py,sha256=Ee7tBy_kvLSTrwmTaOWL0kL-0AqH0DWS7zd_7gJdAj0,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/astor.py,sha256=xuDUkjq0-b6fhtwjhbnebzbqQZAjMSHR1IIS5uOodVg,777
-utilities/asyncio.py,sha256=Hysj5gHlizHUhBo3D9Czo5wciJxNoO5HrhXZ4hbC-sk,21478
+utilities/asyncio.py,sha256=d_z20q4pFhjMBruepHKjvbWVFF-89VpP6_o9ZFQIe9k,21508
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
 utilities/atools.py,sha256=IYMuFSFGSKyuQmqD6v5IUtDlz8PPw0Sr87Cub_gRU3M,1168
 utilities/cachetools.py,sha256=C1zqOg7BYz0IfQFK8e3qaDDgEZxDpo47F15RTfJM37Q,2910
@@ -29,7 +29,7 @@ utilities/ipython.py,sha256=V2oMYHvEKvlNBzxDXdLvKi48oUq2SclRg5xasjaXStw,763
 utilities/iterables.py,sha256=prKXBdF5QfLTGC-q4567DwO8xzUng_Z-2a4wBkMqyDo,45360
 utilities/jupyter.py,sha256=ft5JA7fBxXKzP-L9W8f2-wbF0QeYc_2uLQNFDVk4Z-M,2917
 utilities/lightweight_charts.py,sha256=vyVOzarYhBIOZj2xDhqdbP85qbSKUjdc6Au91rc1W4M,2814
-utilities/logging.py,sha256=55LVJYd2BKzeTTQr4tG7uSNMDVEgUO56wK4s-rldjrM,25326
+utilities/logging.py,sha256=gwo3pusPjnWO1ollrtn1VKYyRAQJTue4SkCbMeNvec4,25715
 utilities/loguru.py,sha256=MEMQVWrdECxk1e3FxGzmOf21vWT9j8CAir98SEXFKPA,3809
 utilities/luigi.py,sha256=fpH9MbxJDuo6-k9iCXRayFRtiVbUtibCJKugf7ygpv0,5988
 utilities/math.py,sha256=-mQgbah-dPJwOEWf3SonrFoVZ2AVxMgpeQ3dfVa-oJA,26764
@@ -64,7 +64,7 @@ utilities/rich.py,sha256=t50MwwVBsoOLxzmeVFSVpjno4OW6Ufum32skXbV8-Bs,1911
 utilities/scipy.py,sha256=X6ROnHwiUhAmPhM0jkfEh0-Fd9iRvwiqtCQMOLmOQF8,945
 utilities/sentinel.py,sha256=3jIwgpMekWgDAxPDA_hXMP2St43cPhciKN3LWiZ7kv0,1248
 utilities/shelve.py,sha256=HZsMwK4tcIfg3sh0gApx4-yjQnrY4o3V3ZRimvRhoW0,738
-utilities/slack_sdk.py,sha256=Gbla983KulSSXnNyzaXgYQLKoq84KvLH8SdhxU-jQ0Q,4126
+utilities/slack_sdk.py,sha256=wPqn9F5AMXgmkp3zgIrBMllLt2SDCCnBNNyi-ag3yzw,5555
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
 utilities/sqlalchemy.py,sha256=585hWuuXVTKTnyn0Pfd9JI6jp-hmKW6pLKGYMjXjytM,36959
 utilities/sqlalchemy_polars.py,sha256=wjJpoUo-yO9E2ujpG_06vV5r2OdvBiQ4yvV6wKCa2Tk,15605
@@ -76,7 +76,7 @@ utilities/tenacity.py,sha256=1PUvODiBVgeqIh7G5TRt5WWMSqjLYkEqP53itT97WQc,4914
 utilities/text.py,sha256=Fo12N4aA7k2rnb4W4vH9iiDh88Q5_nvRssTkfNsvVM8,10965
 utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
 utilities/timer.py,sha256=Rkc49KSpHuC8s7vUxGO9DU55U9I6yDKnchsQqrUCVBs,4075
-utilities/traceback.py,sha256=secexUnBsecfWV4ZuqP1W4pGF3prOeO1CRyJK-8zQDU,27402
+utilities/traceback.py,sha256=p9WATV-e4_5AW6SvyRBiU-MY8XnEFcKgrFNUoFzalXI,27521
 utilities/types.py,sha256=FhE1b8v_s_IlmXucwY7jAMWAq9cfpzyKssQwgwb3jnM,18267
 utilities/typing.py,sha256=H6ysJkI830aRwLsMKz0SZIw4cpcsm7d6KhQOwr-SDh0,13817
 utilities/tzdata.py,sha256=yCf70NICwAeazN3_JcXhWvRqCy06XJNQ42j7r6gw3HY,1217
@@ -87,7 +87,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=iLRP_-8CZtBpHKbGZGu-kjSMg1ZubJ-VSmgSy7Eudxw,17787
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-Xm57PMMwDTYpxJdkiJG13wnbwK--I7XItBh5WVhD-o,1874
-dycw_utilities-0.114.3.dist-info/METADATA,sha256=vNqkObcQLSh6xKu-ehOh26STTT5syHO9ImD4L7O3dPc,12943
-dycw_utilities-0.114.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.114.3.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.114.3.dist-info/RECORD,,
+dycw_utilities-0.114.5.dist-info/METADATA,sha256=n2XeSZhLo9Msjq6ct0xGXRAtvWEh-YVamV95mGjMh3Y,12943
+dycw_utilities-0.114.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.114.5.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.114.5.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.114.3"
+__version__ = "0.114.5"

utilities/asyncio.py

@@ -326,7 +326,7 @@ class ExceptionProcessor(QueueProcessor[Exception | type[Exception]]):
 ##
 
 
-@dataclass(kw_only=True)
+@dataclass(kw_only=True, unsafe_hash=True)
 class InfiniteLooper(ABC, Generic[THashable]):
     """An infinite loop which can throw exceptions by setting events."""
 
@@ -334,7 +334,7 @@ class InfiniteLooper(ABC, Generic[THashable]):
     sleep_restart: Duration = MINUTE
     logger: str | None = None
     _events: Mapping[THashable, Event] = field(
-        default_factory=dict, init=False, repr=False
+        default_factory=dict, init=False, repr=False, hash=False
     )
 
     def __post_init__(self) -> None:

utilities/logging.py

@@ -587,19 +587,19 @@ def setup_logging(
 
     # console
     if console_level is not None:  # skipif-ci-and-windows
-        console_low_handler = StreamHandler(stream=stdout)
-        add_filters(console_low_handler, lambda x: x.levelno < ERROR)
-        add_filters(console_low_handler, *console_filters)
-        add_filters(console_low_handler, *filters)
-        console_low_handler.setFormatter(console_formatter)
-        console_low_handler.setLevel(console_level)
-        logger_use.addHandler(console_low_handler)
-
-        console_high_handler = StreamHandler(stream=stdout)
-        add_filters(console_high_handler, *console_filters)
-        add_filters(console_high_handler, *filters)
+        console_low_or_no_exc_handler = StreamHandler(stream=stdout)
+        add_filters(console_low_or_no_exc_handler, _console_low_or_no_exc_filter)
+        add_filters(console_low_or_no_exc_handler, *console_filters)
+        add_filters(console_low_or_no_exc_handler, *filters)
+        console_low_or_no_exc_handler.setFormatter(console_formatter)
+        console_low_or_no_exc_handler.setLevel(console_level)
+        logger_use.addHandler(console_low_or_no_exc_handler)
+
+        console_high_and_exc_handler = StreamHandler(stream=stdout)
+        add_filters(console_high_and_exc_handler, *console_filters)
+        add_filters(console_high_and_exc_handler, *filters)
         _ = RichTracebackFormatter.create_and_set(
-            console_high_handler,
+            console_high_and_exc_handler,
             version=formatter_version,
             max_width=formatter_max_width,
             indent_size=formatter_indent_size,
@@ -610,10 +610,10 @@ def setup_logging(
             detail=True,
             post=_ansi_wrap_red,
         )
-        console_high_handler.setLevel(
+        console_high_and_exc_handler.setLevel(
             max(get_logging_level_number(console_level), ERROR)
         )
-        logger_use.addHandler(console_high_handler)
+        logger_use.addHandler(console_high_and_exc_handler)
 
     # debug & info
     directory = resolve_path(path=files_dir)  # skipif-ci-and-windows
@@ -640,7 +640,7 @@ def setup_logging(
     standalone_file_handler = StandaloneFileHandler(  # skipif-ci-and-windows
         level=ERROR, path=directory.joinpath("errors")
     )
-    add_filters(standalone_file_handler, lambda x: x.exc_info is not None)
+    add_filters(standalone_file_handler, _standalone_file_filter)
     standalone_file_handler.setFormatter(
         RichTracebackFormatter(
             version=formatter_version,
@@ -660,6 +660,16 @@ def setup_logging(
         extra(logger_use)
 
 
+def _console_low_or_no_exc_filter(record: LogRecord, /) -> bool:
+    return (record.levelno < ERROR) or (
+        (record.levelno >= ERROR) and (record.exc_info is None)
+    )
+
+
+def _standalone_file_filter(record: LogRecord, /) -> bool:
+    return record.exc_info is not None
+
+
 ##
 
 

utilities/slack_sdk.py

@@ -9,7 +9,12 @@ from typing import TYPE_CHECKING, override
 
 from slack_sdk.webhook.async_client import AsyncWebhookClient
 
-from utilities.asyncio import QueueProcessor, sleep_dur, timeout_dur
+from utilities.asyncio import (
+    InfiniteQueueLooper,
+    QueueProcessor,
+    sleep_dur,
+    timeout_dur,
+)
 from utilities.datetime import MINUTE, SECOND, datetime_duration_to_float
 from utilities.functools import cache
 from utilities.math import safe_round
@@ -95,6 +100,49 @@ class SlackHandler(Handler, QueueProcessor[str]):
             await sleep_dur(duration=self.sleep)
 
 
+@dataclass(init=False, unsafe_hash=True)
+class SlackHandlerIQL(Handler, InfiniteQueueLooper[None, str]):
+    """Handler for sending messages to Slack."""
+
+    @override
+    def __init__(
+        self,
+        url: str,
+        /,
+        *,
+        level: int = NOTSET,
+        sleep_core: Duration = _SLEEP,
+        sleep_restart: Duration = _SLEEP,
+        queue_type: type[Queue[str]] = Queue,
+        sender: Callable[[str, str], Coroutine1[None]] = _send_adapter,
+        timeout: Duration = _TIMEOUT,
+    ) -> None:
+        InfiniteQueueLooper.__init__(  # InfiniteQueueLooper first
+            self, queue_type=queue_type
+        )
+        InfiniteQueueLooper.__post_init__(self)
+        Handler.__init__(self, level=level)
+        self.url = url
+        self.sender = sender
+        self.timeout = timeout
+        self.sleep_core = sleep_core
+        self.sleep_restart = sleep_restart
+
+    @override
+    def emit(self, record: LogRecord) -> None:
+        try:
+            self.put_items_nowait(self.format(record))
+        except Exception:  # noqa: BLE001  # pragma: no cover
+            self.handleError(record)
+
+    @override
+    async def _process_items(self, *items: str) -> None:
+        """Process the first item."""
+        text = "\n".join(items)
+        async with timeout_dur(duration=self.timeout):
+            await self.sender(self.url, text)
+
+
 ##
 
 
@@ -128,4 +176,4 @@ def _get_client(url: str, /, *, timeout: Duration = _TIMEOUT) -> AsyncWebhookCli
     return AsyncWebhookClient(url, timeout=timeout_use)
 
 
-__all__ = ["SendToSlackError", "SlackHandler", "send_to_slack"]
+__all__ = ["SendToSlackError", "SlackHandler", "SlackHandlerIQL", "send_to_slack"]

utilities/traceback.py

@@ -60,6 +60,9 @@ _CALL_ARGS = "_CALL_ARGS"
 _INDENT = 4 * " "
 
 
+##
+
+
 class RichTracebackFormatter(Formatter):
     """Formatter for rich tracebacks."""
 
@@ -162,10 +165,17 @@ class RichTracebackFormatter(Formatter):
             detail=detail,
             post=post,
         )
-        handler.addFilter(lambda r: r.exc_info is not None)
+        handler.addFilter(cls._has_exc_info)
         handler.setFormatter(formatter)
         return formatter
 
+    @classmethod
+    def _has_exc_info(cls, record: LogRecord, /) -> bool:
+        return record.exc_info is not None
+
+
+##
+
 
 @dataclass(repr=False, kw_only=True, slots=True)
 class _CallArgs:
@@ -470,6 +480,9 @@ class _Frame:
         return indent("\n".join(lines), depth * _INDENT)
 
 
+##
+
+
 def get_rich_traceback(
     error: TBaseException,
     /,