dycw-utilities 0.135.0__py3-none-any.whl → 0.178.1__py3-none-any.whl

utilities/zoneinfo.py

@@ -7,7 +7,8 @@ from zoneinfo import ZoneInfo
 
 from whenever import ZonedDateTime
 
-from utilities.tzlocal import LOCAL_TIME_ZONE
+from utilities.types import TIME_ZONES
+from utilities.tzlocal import LOCAL_TIME_ZONE, LOCAL_TIME_ZONE_NAME
 
 if TYPE_CHECKING:
     from utilities.types import TimeZone, TimeZoneLike
@@ -19,57 +20,114 @@ UTC = ZoneInfo("UTC")
 ##
 
 
-def ensure_time_zone(obj: TimeZoneLike, /) -> ZoneInfo:
-    """Ensure the object is a time zone."""
+def to_zone_info(obj: TimeZoneLike, /) -> ZoneInfo:
+    """Convert to a time-zone."""
     match obj:
         case ZoneInfo() as zone_info:
             return zone_info
-        case ZonedDateTime() as datetime:
-            return ZoneInfo(datetime.tz)
-        case "local":
+        case ZonedDateTime() as date_time:
+            return ZoneInfo(date_time.tz)
+        case "local" | "localtime":
             return LOCAL_TIME_ZONE
         case str() as key:
             return ZoneInfo(key)
         case dt.tzinfo() as tzinfo:
             if tzinfo is dt.UTC:
                 return UTC
-            raise _EnsureTimeZoneInvalidTZInfoError(time_zone=obj)
-        case dt.datetime() as datetime:
-            if datetime.tzinfo is None:
-                raise _EnsureTimeZonePlainDateTimeError(datetime=datetime)
-            return ensure_time_zone(datetime.tzinfo)
-        case _ as never:
+            raise _ToZoneInfoInvalidTZInfoError(time_zone=obj)
+        case dt.datetime() as date_time:
+            if date_time.tzinfo is None:
+                raise _ToZoneInfoPlainDateTimeError(date_time=date_time)
+            return to_zone_info(date_time.tzinfo)
+        case never:
             assert_never(never)
 
 
 @dataclass(kw_only=True, slots=True)
-class EnsureTimeZoneError(Exception): ...
+class ToTimeZoneError(Exception): ...
 
 
 @dataclass(kw_only=True, slots=True)
-class _EnsureTimeZoneInvalidTZInfoError(EnsureTimeZoneError):
+class _ToZoneInfoInvalidTZInfoError(ToTimeZoneError):
     time_zone: dt.tzinfo
 
     @override
     def __str__(self) -> str:
-        return f"Unsupported time zone: {self.time_zone}"
+        return f"Invalid time-zone: {self.time_zone}"
 
 
 @dataclass(kw_only=True, slots=True)
-class _EnsureTimeZonePlainDateTimeError(EnsureTimeZoneError):
-    datetime: dt.datetime
+class _ToZoneInfoPlainDateTimeError(ToTimeZoneError):
+    date_time: dt.datetime
 
     @override
     def __str__(self) -> str:
-        return f"Plain datetime: {self.datetime}"
+        return f"Plain date-time: {self.date_time}"
 
 
 ##
 
 
-def get_time_zone_name(time_zone: TimeZoneLike, /) -> TimeZone:
-    """Get the name of a time zone."""
-    return cast("TimeZone", ensure_time_zone(time_zone).key)
+def to_time_zone_name(obj: TimeZoneLike, /) -> TimeZone:
+    """Convert to a time zone name."""
+    match obj:
+        case ZoneInfo() as zone_info:
+            return cast("TimeZone", zone_info.key)
+        case ZonedDateTime() as date_time:
+            return cast("TimeZone", date_time.tz)
+        case "local" | "localtime":
+            return LOCAL_TIME_ZONE_NAME
+        case str() as time_zone:
+            if time_zone in TIME_ZONES:
+                return time_zone
+            raise _ToTimeZoneNameInvalidKeyError(time_zone=time_zone)
+        case dt.tzinfo() as tzinfo:
+            if tzinfo is dt.UTC:
+                return cast("TimeZone", UTC.key)
+            raise _ToTimeZoneNameInvalidTZInfoError(time_zone=obj)
+        case dt.datetime() as date_time:
+            if date_time.tzinfo is None:
+                raise _ToTimeZoneNamePlainDateTimeError(date_time=date_time)
+            return to_time_zone_name(date_time.tzinfo)
+        case never:
+            assert_never(never)
+
+
+@dataclass(kw_only=True, slots=True)
+class ToTimeZoneNameError(Exception): ...
+
+
+@dataclass(kw_only=True, slots=True)
+class _ToTimeZoneNameInvalidKeyError(ToTimeZoneNameError):
+    time_zone: str
+
+    @override
+    def __str__(self) -> str:
+        return f"Invalid time-zone: {self.time_zone!r}"
+
+
+@dataclass(kw_only=True, slots=True)
+class _ToTimeZoneNameInvalidTZInfoError(ToTimeZoneNameError):
+    time_zone: dt.tzinfo
+
+    @override
+    def __str__(self) -> str:
+        return f"Invalid time-zone: {self.time_zone}"
+
+
+@dataclass(kw_only=True, slots=True)
+class _ToTimeZoneNamePlainDateTimeError(ToTimeZoneNameError):
+    date_time: dt.datetime
+
+    @override
+    def __str__(self) -> str:
+        return f"Plain date-time: {self.date_time}"
 
 
-__all__ = ["UTC", "EnsureTimeZoneError", "ensure_time_zone", "get_time_zone_name"]
+__all__ = [
+    "UTC",
+    "ToTimeZoneError",
+    "ToTimeZoneNameError",
+    "to_time_zone_name",
+    "to_zone_info",
+]

dycw_utilities-0.135.0.dist-info/METADATA

@@ -1,39 +0,0 @@
-Metadata-Version: 2.4
-Name: dycw-utilities
-Version: 0.135.0
-Author-email: Derek Wan <d.wan@icloud.com>
-License-File: LICENSE
-Requires-Python: >=3.12
-Requires-Dist: atomicwrites<1.5,>=1.4.1
-Requires-Dist: typing-extensions<4.15,>=4.14.0
-Requires-Dist: tzlocal<5.4,>=5.3.1
-Requires-Dist: whenever<0.9,>=0.8.5
-Provides-Extra: logging
-Requires-Dist: coloredlogs<15.1,>=15.0.1; extra == 'logging'
-Provides-Extra: test
-Requires-Dist: dycw-pytest-only<2.2,>=2.1.1; extra == 'test'
-Requires-Dist: hypothesis<6.136,>=6.135.2; extra == 'test'
-Requires-Dist: pytest-asyncio<1.1,>=1.0.0; extra == 'test'
-Requires-Dist: pytest-cov<6.2,>=6.1.1; extra == 'test'
-Requires-Dist: pytest-instafail<0.6,>=0.5.0; extra == 'test'
-Requires-Dist: pytest-lazy-fixtures<1.2,>=1.1.4; extra == 'test'
-Requires-Dist: pytest-randomly<3.17,>=3.16.0; extra == 'test'
-Requires-Dist: pytest-regressions<2.9,>=2.8.0; extra == 'test'
-Requires-Dist: pytest-rerunfailures<16,>=15.1; extra == 'test'
-Requires-Dist: pytest-rng<1.1,>=1.0.0; extra == 'test'
-Requires-Dist: pytest-timeout<2.5,>=2.4.0; extra == 'test'
-Requires-Dist: pytest-xdist<3.8,>=3.7.0; extra == 'test'
-Requires-Dist: pytest<8.4,>=8.3.5; extra == 'test'
-Description-Content-Type: text/markdown
-
-[![PyPI version](https://badge.fury.io/py/dycw-utilities.svg)](https://badge.fury.io/py/dycw-utilities)
-
-# `dycw-utilities`
-
-[All the Python functions I don't want to write twice.](https://github.com/nvim-lua/plenary.nvim)
-
-## Installation
-
-- `pip install dycw-utilities`
-
-or with [extras](https://github.com/dycw/python-utilities/blob/master/pyproject.toml).

dycw_utilities-0.135.0.dist-info/RECORD

@@ -1,96 +0,0 @@
-utilities/__init__.py,sha256=ZNWx_k_4ER3p4DVbJX8DEVVB_Qth1GiAAheIk1f0wkY,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=qqoV2xXrZoOw1ck9yHOh8mxlnoh96Huwd5xGmJcu3fQ,38165
-utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
-utilities/atools.py,sha256=9im2g8OCf-Iynqa8bAv8N0Ycj9QvrJmGO7yLCZEdgII,986
-utilities/cachetools.py,sha256=v1-9sXHLdOLiwmkq6NB0OUbxeKBuVVN6wmAWefWoaHI,2744
-utilities/click.py,sha256=rXCbuBi6xqRK2-2e0hXQ2vYBWJzT9vPY6KkRIu73_uI,16049
-utilities/concurrent.py,sha256=ZdhcNeBl1-HaAPY3h7bZ5ccuYdfdq2ATHplvZdnzlhk,2858
-utilities/contextlib.py,sha256=lpaLJBy3X0UGLWjM98jkQZZq8so4fRmoK-Bheq0uOW4,1027
-utilities/contextvars.py,sha256=RsSGGrbQqqZ67rOydnM7WWIsM2lIE31UHJLejnHJPWY,505
-utilities/cryptography.py,sha256=_CiK_K6c_-uQuUhsUNjNjTL-nqxAh4_1zTfS11Xe120,972
-utilities/cvxpy.py,sha256=Rv1-fD-XYerosCavRF8Pohop2DBkU3AlFaGTfD8AEAA,13776
-utilities/dataclasses.py,sha256=1-REGxtzCo2BbGkjLw5idjbTaiItpOiWvMNbVkJrFXM,32663
-utilities/enum.py,sha256=IEPMGiNJBcofGPuoGZErf4bMNSBE4SLs32nvm2r81h0,5791
-utilities/errors.py,sha256=nC7ZYtxxDBMfrTHtT_MByBfup_wfGQFRo3eDt-0ZPe8,1045
-utilities/eventkit.py,sha256=cV76NIHKDPyrhnZRuhjCXse5CQqgRGXZ7p7Gon4Mu2g,12646
-utilities/fastapi.py,sha256=E8T2J1-N_RbpkN4czthU6NPIxAZDzxy-k_WGJaxeJ48,2671
-utilities/fpdf2.py,sha256=PmPj8ugr_SlxFEpw-9OsI8--mteLQ4MaXv_Cbmf7XXs,1852
-utilities/functions.py,sha256=Teqm7ylOqEcBLHWhRwjKqouUfc6nv_6qO6fILpnOPyA,27937
-utilities/functools.py,sha256=I00ru2gQPakZw2SHVeKIKXfTv741655s6HI0lUoE0D4,1552
-utilities/getpass.py,sha256=DfN5UgMAtFCqS3dSfFHUfqIMZX2shXvwphOz_6J6f6A,103
-utilities/git.py,sha256=oi7-_l5e9haSANSCvQw25ufYGoNahuUPHAZ6114s3JQ,1191
-utilities/hashlib.py,sha256=SVTgtguur0P4elppvzOBbLEjVM3Pea0eWB61yg2ilxo,309
-utilities/http.py,sha256=WcahTcKYRtZ04WXQoWt5EGCgFPcyHD3EJdlMfxvDt-0,946
-utilities/hypothesis.py,sha256=_OFFvb8rVvMgXo04zaIFYw4p32QGrVHzJsR8pL3G8l4,36195
-utilities/importlib.py,sha256=mV1xT_O_zt_GnZZ36tl3xOmMaN_3jErDWY54fX39F6Y,429
-utilities/inflect.py,sha256=DbqB5Q9FbRGJ1NbvEiZBirRMxCxgrz91zy5jCO9ZIs0,347
-utilities/ipython.py,sha256=V2oMYHvEKvlNBzxDXdLvKi48oUq2SclRg5xasjaXStw,763
-utilities/iterables.py,sha256=wlcm0PS2fKG-H0r0k367NOLrryMbfXLlwzUeAmBSvv8,43392
-utilities/jupyter.py,sha256=ft5JA7fBxXKzP-L9W8f2-wbF0QeYc_2uLQNFDVk4Z-M,2917
-utilities/libcst.py,sha256=Jto5ppzRzsxn4AD32IS8n0lbgLYXwsVJB6EY8giNZyY,4974
-utilities/lightweight_charts.py,sha256=JrkrAZMo6JID2Eoc9QCc05Y_pK4l2zsApIhmii1z2Ig,2764
-utilities/logging.py,sha256=j0xS7bNdZcMAobWSRahpg_d7GWewd_99oXvexrjWm6k,17841
-utilities/luigi.py,sha256=wK7cB3y8NXeSa8d6r_yTKRmjMguNmIPmy52yg10vPaI,4774
-utilities/math.py,sha256=_6vrDyjtaqE_OFE-F2DNWrDG_J_kMl3nFAJsok9v_bY,26862
-utilities/memory_profiler.py,sha256=XzN56jDCa5aqXS_DxEjb_K4L6aIWh_5zyKi6OhcIxw0,853
-utilities/modules.py,sha256=iuvLluJya-hvl1Q25-Jk3dLgx2Es3ck4SjJiEkAlVTs,3195
-utilities/more_itertools.py,sha256=EX4P5JTc3p9fF3Ia3ZzPy9CEUe6OZcFqrrVplZ4pMaQ,10937
-utilities/numpy.py,sha256=Xn23sA2ZbVNqwUYEgNJD3XBYH6IbCri_WkHSNhg3NkY,26122
-utilities/operator.py,sha256=fZM2DBZdjtU8Lh9Z-eHC4ktNr0vO3JfX8hXgIYwdvBI,3826
-utilities/optuna.py,sha256=C-fhWYiXHVPo1l8QctYkFJ4DyhbSrGorzP1dJb_qvd8,1933
-utilities/orjson.py,sha256=y5ynSGhQjX7vikfvsHh_AklLnH7gjvsSkIC3h5tnx98,36474
-utilities/os.py,sha256=D_FyyT-6TtqiN9KSS7c9g1fnUtgxmyMtzAjmYLkk46A,3587
-utilities/parse.py,sha256=bCZW1bBfXM2j-yLGq0TGUshOnbL8V-U8amGu_OZTE_I,17907
-utilities/pathlib.py,sha256=jCFPZm4rBKylEva9wDVTwQlTTVKMepu92WrTpoGa438,3248
-utilities/period.py,sha256=QZsy2tCcoh0LGr70bkvlB5iWqEx6hdit3-iuaWJYXXc,4689
-utilities/pickle.py,sha256=MBT2xZCsv0pH868IXLGKnlcqNx2IRVKYNpRcqiQQqxw,653
-utilities/platform.py,sha256=5uCKRf_ij7ukJDcbnNfhY2ay9fbrpiNLRO1t2QvcwqQ,2825
-utilities/polars.py,sha256=xCeB-pLkezOoQJvgxTdz2rNlaehdhek-HmDztXWn2j0,63266
-utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
-utilities/pottery.py,sha256=RN3XwOEsVAPXvEfsRPmn3ZSKgTzK_c182PNrtksq-bg,3429
-utilities/pqdm.py,sha256=BTsYPtbKQWwX-iXF4qCkfPG7DPxIB54J989n83bXrIo,3092
-utilities/psutil.py,sha256=0j4YxtVb8VjaaKKiHg6UEK95SUPkEcENgPtLgPJsNv0,3760
-utilities/pudb.py,sha256=Dte7P9J0VWrWTNYvwxZX6kgaEtdTt5ZnnmOAUTjltqU,1771
-utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-utilities/pydantic.py,sha256=CmxCi4sukeHM3JGjJ1Rbp8UAvcx4MZapLg10mFYJ-nk,1771
-utilities/pyinstrument.py,sha256=_Rfq6Gg4NKV2NF0dRYOpK2IRyyePxI7-3UmHIQLYrlQ,852
-utilities/pytest.py,sha256=vv5ZpePZS6pBxl5jY3XQR5SnAhgd4f0sSMyRsvHWg60,8086
-utilities/pytest_regressions.py,sha256=YI55B7EtLjhz7zPJZ6NK9bWrxrKCKabWZJe1cwcbA5o,5082
-utilities/python_dotenv.py,sha256=dYooRYwqrvhSoZWuiVbCiKUWiS-M5b5yv2zDWGYPEvI,3209
-utilities/random.py,sha256=YWYzWxQDeyJRiuHGnO1OxF6dDucpq7qc1tH_ealwCRg,4130
-utilities/re.py,sha256=6qxeV0rQZaBDKWcB7apSBmxtg_XzoGY-EdegTkMn-ZY,4578
-utilities/redis.py,sha256=FuVyxF70BMnlnbQdvdv09Nj7xNuWmws4s94Bb-tIMNc,35415
-utilities/reprlib.py,sha256=ssYTcBW-TeRh3fhCJv57sopTZHF5FrPyyUg9yp5XBlo,3953
-utilities/scipy.py,sha256=wZJM7fEgBAkLSYYvSmsg5ac-QuwAI0BGqHVetw1_Hb0,947
-utilities/sentinel.py,sha256=3jIwgpMekWgDAxPDA_hXMP2St43cPhciKN3LWiZ7kv0,1248
-utilities/shelve.py,sha256=HZsMwK4tcIfg3sh0gApx4-yjQnrY4o3V3ZRimvRhoW0,738
-utilities/slack_sdk.py,sha256=RrB34gOj4TzBFh1vzMy6wL_ajzIG-2c9kiS6c6LzMFM,4233
-utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
-utilities/sqlalchemy.py,sha256=5Cx-ioYb099xDUDZuXrXp5tFU90HnCvu3jUu6wkGxrg,38001
-utilities/sqlalchemy_polars.py,sha256=5TwuUPORZ1Nz3qbI6L6pzeThtEv8Ws4aQW8cm3MG-v0,14293
-utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
-utilities/streamlit.py,sha256=U9PJBaKP1IdSykKhPZhIzSPTZsmLsnwbEPZWzNhJPKk,2955
-utilities/string.py,sha256=XmU-s04qIV_tODnKl2pQiwmHaxzgOqRKU-RyzdrfvSE,620
-utilities/tempfile.py,sha256=VqmZJAhTJ1OaVywFzk5eqROV8iJbW9XQ_QYAV0bpdRo,1384
-utilities/text.py,sha256=ymBFlP_cA8OgNnZRVNs7FAh7OG8HxE6YkiLEMZv5g_A,11297
-utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
-utilities/timer.py,sha256=oXfTii6ymu57niP0BDGZjFD55LEHi2a19kqZKiTgaFQ,2588
-utilities/traceback.py,sha256=i-790AQbTrDA8MiYyOcYPFpm48I558VR_kL_7x4ypfY,8503
-utilities/typed_settings.py,sha256=niNxgvVKrTy1c0j0D0Jp2tO3m231f4yJJ-38ni-xY3Q,3739
-utilities/types.py,sha256=DrFWZQqpkekHdnr2-G2eBaxGX8FoND6ON2XFRvhm8Jo,17177
-utilities/typing.py,sha256=Z-_XDaWyT_6wIo3qfNK-hvRlzxP2Jxa9PgXzm5rDYRA,13790
-utilities/tzdata.py,sha256=fgNVj66yUbCSI_-vrRVzSD3gtf-L_8IEJEPjP_Jel5Y,266
-utilities/tzlocal.py,sha256=KyCXEgCTjqGFx-389JdTuhMRUaT06U1RCMdWoED-qro,728
-utilities/uuid.py,sha256=32p7DGHGM2Btx6PcBvCZvERSWbpupMXqx6FppPoSoTU,612
-utilities/version.py,sha256=ufhJMmI6KPs1-3wBI71aj5wCukd3sP_m11usLe88DNA,5117
-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.135.0.dist-info/METADATA,sha256=USiapXRoQLG5Wl1CftVIV6nKma65HtH3nIoU7n8OMl8,1584
-dycw_utilities-0.135.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.135.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.135.0.dist-info/RECORD,,

dycw_utilities-0.135.0.dist-info/WHEEL

@@ -1,4 +0,0 @@
-Wheel-Version: 1.0
-Generator: hatchling 1.27.0
-Root-Is-Purelib: true
-Tag: py3-none-any

dycw_utilities-0.135.0.dist-info/licenses/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2024 Derek Wan
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.

utilities/aiolimiter.py

@@ -1,25 +0,0 @@
-from __future__ import annotations
-
-from asyncio import get_running_loop
-from typing import TYPE_CHECKING
-
-from aiolimiter import AsyncLimiter
-
-if TYPE_CHECKING:
-    from collections.abc import Hashable
-
-_LIMITERS: dict[tuple[int, Hashable], AsyncLimiter] = {}
-
-
-def get_async_limiter(key: Hashable, /, *, rate: float = 1.0) -> AsyncLimiter:
-    """Get a loop-aware rate limiter."""
-    id_ = id(get_running_loop())
-    full = (id_, key)
-    try:
-        return _LIMITERS[full]
-    except KeyError:
-        limiter = _LIMITERS[full] = AsyncLimiter(1.0, time_period=rate)
-        return limiter
-
-
-__all__ = ["get_async_limiter"]

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"]