dycw-utilities 0.136.2__py3-none-any.whl → 0.136.4__py3-none-any.whl

{dycw_utilities-0.136.2.dist-info → dycw_utilities-0.136.4.dist-info}/METADATA

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

{dycw_utilities-0.136.2.dist-info → dycw_utilities-0.136.4.dist-info}/RECORD

@@ -1,8 +1,7 @@
-utilities/__init__.py,sha256=ESUyKsoR2YasUqFOJys4BrLjrFP0GRipV9h5dDOv4jk,60
+utilities/__init__.py,sha256=dsiNRoGm7Boxbp7neOt74yIhZx6QM31tZ2CTGJY7K20,60
 utilities/aiolimiter.py,sha256=mD0wEiqMgwpty4XTbawFpnkkmJS6R4JRsVXFUaoitSU,628
 utilities/altair.py,sha256=HeZBVUocjkrTNwwKrClppsIqgNFF-ykv05HfZSoHYno,9104
-utilities/arq.py,sha256=S-sfBfY-E1ErRKf4sSXt2YyCjKvu-pBlOECDfjBebRA,6399
-utilities/asyncio.py,sha256=kr0S02DrAgpjPU3ySmhD806S5x_7wFUFeV7-zWMrinY,38249
+utilities/asyncio.py,sha256=dcGeKQzjLBXxKzZkVIk5oZsFXEcynVbRB9iNB5XEDZk,38526
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
 utilities/atools.py,sha256=9im2g8OCf-Iynqa8bAv8N0Ycj9QvrJmGO7yLCZEdgII,986
 utilities/cachetools.py,sha256=v1-9sXHLdOLiwmkq6NB0OUbxeKBuVVN6wmAWefWoaHI,2744
@@ -89,7 +88,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=A-yoOqBqrcVD1yDINDsTFDw7dq9-zgUGn_f8CxVUQJs,23332
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=oEH-nL3t4h9uawyZqWDtNtDAl6M-CLpLYGI_nI6DulM,1971
-dycw_utilities-0.136.2.dist-info/METADATA,sha256=RP6OhX1-oe-Warr3bsEEICiTheRrgAPd9_Mz3Kd0afY,1637
-dycw_utilities-0.136.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.136.2.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.136.2.dist-info/RECORD,,
+dycw_utilities-0.136.4.dist-info/METADATA,sha256=2ONGjvk-6Ada8oSDMVX6w1cxOniaJPZpMI-Spv4Lhzs,1637
+dycw_utilities-0.136.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.136.4.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.136.4.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.136.2"
+__version__ = "0.136.4"

utilities/asyncio.py

@@ -220,6 +220,7 @@ class EnhancedQueue[T](Queue[T]):
 class EnhancedTaskGroup(TaskGroup):
     """Task group with enhanced features."""
 
+    _max_tasks: int | None
     _semaphore: Semaphore | None
     _timeout: TimeDelta | None
     _error: MaybeType[BaseException]
@@ -237,7 +238,11 @@ class EnhancedTaskGroup(TaskGroup):
         debug: MaybeCallableBool = False,
     ) -> None:
         super().__init__()
-        self._semaphore = None if max_tasks is None else Semaphore(max_tasks)
+        self._max_tasks = max_tasks
+        if (max_tasks is None) or (max_tasks <= 0):
+            self._semaphore = None
+        else:
+            self._semaphore = Semaphore(max_tasks)
         self._timeout = timeout
         self._error = error
         self._debug = debug
@@ -257,7 +262,7 @@ class EnhancedTaskGroup(TaskGroup):
         tb: TracebackType | None,
     ) -> None:
         _ = await self._stack.__aexit__(et, exc, tb)
-        match to_bool(bool_=self._debug):
+        match self._is_debug():
             case True:
                 with suppress(Exception):
                     _ = await super().__aexit__(et, exc, tb)
@@ -293,7 +298,7 @@ class EnhancedTaskGroup(TaskGroup):
         name: str | None = None,
         context: Context | None = None,
     ) -> T | Task[T]:
-        match to_bool(bool_=self._debug):
+        match self._is_debug():
             case True:
                 return await coro
             case False:
@@ -301,6 +306,11 @@ class EnhancedTaskGroup(TaskGroup):
             case _ as never:
                 assert_never(never)
 
+    def _is_debug(self) -> bool:
+        return to_bool(bool_=self._debug) or (
+            (self._max_tasks is not None) and (self._max_tasks <= 0)
+        )
+
     async def _wrap_with_semaphore[T](
         self, semaphore: Semaphore, coroutine: _CoroutineLike[T], /
     ) -> T:

utilities/arq.py

@@ -1,216 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from functools import wraps
-from itertools import chain
-from typing import TYPE_CHECKING, Any, Self, cast, override
-
-from arq.constants import default_queue_name, expires_extra_ms
-from arq.cron import cron
-
-from utilities.dataclasses import replace_non_sentinel
-from utilities.sentinel import Sentinel, sentinel
-
-if TYPE_CHECKING:
-    from collections.abc import Callable, Iterable, Sequence
-    from datetime import datetime, timedelta, timezone
-
-    from arq.connections import ArqRedis, RedisSettings
-    from arq.cron import CronJob
-    from arq.jobs import Deserializer, Job, Serializer
-    from arq.typing import (
-        OptionType,
-        SecondsTimedelta,
-        StartupShutdown,
-        WeekdayOptionType,
-        WorkerCoroutine,
-    )
-    from arq.worker import Function
-
-    from utilities.types import Coro, StrMapping
-
-
-def cron_raw(
-    coroutine: Callable[..., Coro[Any]],
-    /,
-    *,
-    name: str | None = None,
-    month: OptionType = None,
-    day: OptionType = None,
-    weekday: WeekdayOptionType = None,
-    hour: OptionType = None,
-    minute: OptionType = None,
-    second: OptionType = 0,
-    microsecond: int = 123_456,
-    run_at_startup: bool = False,
-    unique: bool = True,
-    job_id: str | None = None,
-    timeout: SecondsTimedelta | None = None,
-    keep_result: float | None = 0,
-    keep_result_forever: bool | None = False,
-    max_tries: int | None = 1,
-    args: Iterable[Any] | None = None,
-    kwargs: StrMapping | None = None,
-) -> CronJob:
-    """Create a cron job with a raw coroutine function."""
-    lifted = _lift_cron(
-        coroutine, *(() if args is None else args), **({} if kwargs is None else kwargs)
-    )
-    return cron(
-        lifted,
-        name=name,
-        month=month,
-        day=day,
-        weekday=weekday,
-        hour=hour,
-        minute=minute,
-        second=second,
-        microsecond=microsecond,
-        run_at_startup=run_at_startup,
-        unique=unique,
-        job_id=job_id,
-        timeout=timeout,
-        keep_result=keep_result,
-        keep_result_forever=keep_result_forever,
-        max_tries=max_tries,
-    )
-
-
-def _lift_cron[**P, T](
-    func: Callable[P, Coro[T]], *args: P.args, **kwargs: P.kwargs
-) -> WorkerCoroutine:
-    """Lift a coroutine function & call arg/kwargs for `cron`."""
-
-    @wraps(func)
-    async def wrapped(ctx: StrMapping, /) -> T:
-        _ = ctx
-        return await func(*args, **kwargs)
-
-    return cast("Any", wrapped)
-
-
-##
-
-
-@dataclass(kw_only=True, slots=True)
-class _JobEnqueuer:
-    """Enqueuer of jobs."""
-
-    job_id: str | None = None
-    queue_name: str | None = None
-    defer_until: datetime | None = None
-    defer_by: int | float | timedelta | None = None
-    expires: int | float | timedelta | None = None
-    job_try: int | None = None
-
-    async def __call__[**P, T](
-        self,
-        redis: ArqRedis,
-        function: Callable[P, Coro[T]],
-        *args: P.args,
-        **kwargs: P.kwargs,
-    ) -> Job | None:
-        return await redis.enqueue_job(  # skipif-ci-and-not-linux
-            function.__name__,
-            *args,
-            _job_id=self.job_id,
-            _queue_name=self.queue_name,
-            _defer_until=self.defer_until,
-            _defer_by=self.defer_by,
-            _expires=self.expires,
-            _job_try=self.job_try,
-            **kwargs,
-        )
-
-    def settings(
-        self,
-        *,
-        job_id: str | None | Sentinel = sentinel,
-        queue_name: str | None | Sentinel = sentinel,
-        defer_until: datetime | None | Sentinel = sentinel,
-        defer_by: float | timedelta | None | Sentinel = sentinel,
-        expires: float | timedelta | None | Sentinel = sentinel,
-        job_try: int | None | Sentinel = sentinel,
-    ) -> Self:
-        """Replace elements of the enqueuer."""
-        return replace_non_sentinel(  # skipif-ci-and-not-linux
-            self,
-            job_id=job_id,
-            queue_name=queue_name,
-            defer_until=defer_until,
-            defer_by=defer_by,
-            expires=expires,
-            job_try=job_try,
-        )
-
-
-job_enqueuer = _JobEnqueuer()
-
-
-##
-
-
-class _WorkerMeta(type):
-    @override
-    def __new__(
-        mcs: type[_WorkerMeta],
-        name: str,
-        bases: tuple[type, ...],
-        namespace: dict[str, Any],
-        /,
-    ) -> type[Worker]:
-        cls = cast("type[Worker]", super().__new__(mcs, name, bases, namespace))
-        cls.functions = tuple(chain(cls.functions, map(cls._lift, cls.functions_raw)))
-        return cls
-
-    @classmethod
-    def _lift[**P, T](cls, func: Callable[P, Coro[T]]) -> WorkerCoroutine:
-        """Lift a coroutine function to accept the required `ctx` argument."""
-
-        @wraps(func)
-        async def wrapped(ctx: StrMapping, *args: P.args, **kwargs: P.kwargs) -> T:
-            _ = ctx
-            return await func(*args, **kwargs)
-
-        return cast("Any", wrapped)
-
-
-@dataclass(kw_only=True)
-class Worker(metaclass=_WorkerMeta):
-    """Base class for all workers."""
-
-    functions: Sequence[Function | WorkerCoroutine] = ()
-    functions_raw: Sequence[Callable[..., Coro[Any]]] = ()
-    queue_name: str | None = default_queue_name
-    cron_jobs: Sequence[CronJob] | None = None
-    redis_settings: RedisSettings | None = None
-    redis_pool: ArqRedis | None = None
-    burst: bool = False
-    on_startup: StartupShutdown | None = None
-    on_shutdown: StartupShutdown | None = None
-    on_job_start: StartupShutdown | None = None
-    on_job_end: StartupShutdown | None = None
-    after_job_end: StartupShutdown | None = None
-    handle_signals: bool = True
-    job_completion_wait: int = 0
-    max_jobs: int = 10
-    job_timeout: SecondsTimedelta = 300
-    keep_result: SecondsTimedelta = 3600
-    keep_result_forever: bool = False
-    poll_delay: SecondsTimedelta = 0.5
-    queue_read_limit: int | None = None
-    max_tries: int = 5
-    health_check_interval: SecondsTimedelta = 3600
-    health_check_key: str | None = None
-    ctx: dict[Any, Any] | None = None
-    retry_jobs: bool = True
-    allow_abort_jobs: bool = False
-    max_burst_jobs: int = -1
-    job_serializer: Serializer | None = None
-    job_deserializer: Deserializer | None = None
-    expires_extra_ms: int = expires_extra_ms
-    timezone: timezone | None = None
-    log_results: bool = True
-
-
-__all__ = ["Worker", "cron", "job_enqueuer"]