Dependencies were updated.

.pre-commit-config.yaml

@@ -29,6 +29,7 @@ repos:
         pass_filenames: false
         types: [python]
         args:
+          - "check"
           - "--fix"
           - "taskiq"
           - "tests"
@@ -37,4 +38,8 @@ repos:
         name: Validate types with MyPy
         entry: poetry run mypy
         language: system
+        pass_filenames: false
         types: [python]
+        args:
+          - ./taskiq
+          - ./tests

pyproject.toml

@@ -41,16 +41,16 @@ pyzmq = { version = "^23.2.0", optional = true, markers = "python_version < '3.1
 # For speed
 uvloop = { version = ">=0.16.0,<1", optional = true, markers = "sys_platform != 'win32'" }
 # For hot-reload.
-watchdog = { version = "^2.1.9", optional = true }
+watchdog = { version = "^4", optional = true }
 gitignore-parser = { version = "^0", optional = true }
 pytz = "*"
-orjson = { version = "^3.9.9", optional = true }
+orjson = { version = "^3", optional = true }
 msgpack = { version = "^1.0.7", optional = true }
-cbor2 = { version = "^5.4.6", optional = true }
+cbor2 = { version = "^5", optional = true }
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1.2"
-ruff = "^0.0.291"
+ruff = "^0"
 black = { version = "^22.6.0", allow-prereleases = true }
 mypy = "^1"
 pre-commit = "^2.20.0"
@@ -120,7 +120,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.ruff]
 # List of enabled rulsets.
 # See https://docs.astral.sh/ruff/rules/ for more information.
-select = [
+lint.select = [
     "E",   # Error
     "F",   # Pyflakes
     "W",   # Pycodestyle
@@ -147,7 +147,7 @@ select = [
     "PL",  # PyLint checks
     "RUF", # Specific to Ruff checks
 ]
-ignore = [
+lint.ignore = [
     "D105",    # Missing docstring in magic method
     "D107",    # Missing docstring in __init__
     "D212",    # Multi-line docstring summary should start at the first line
@@ -161,10 +161,10 @@ ignore = [
     "D106",    # Missing docstring in public nested class
 ]
 exclude = [".venv/"]
-mccabe = { max-complexity = 10 }
+lint.mccabe = { max-complexity = 10 }
 line-length = 88
 
-[tool.ruff.per-file-ignores]
+[tool.ruff.lint.per-file-ignores]
 "tests/*" = [
     "S101",   # Use of assert detected
     "S301",   # Use of pickle detected
@@ -174,12 +174,12 @@ line-length = 88
     "D101",   # Missing docstring in public class
 ]
 
-[tool.ruff.pydocstyle]
+[tool.ruff.lint.pydocstyle]
 convention = "pep257"
 ignore-decorators = ["typing.overload"]
 
-[tool.ruff.pylint]
+[tool.ruff.lint.pylint]
 allow-magic-value-types = ["int", "str", "float"]
 
-[tool.ruff.flake8-bugbear]
+[tool.ruff.lint.flake8-bugbear]
 extend-immutable-calls = ["taskiq_dependencies.Depends", "taskiq.TaskiqDepends"]

taskiq/abc/broker.py

@@ -335,9 +335,9 @@ def inner(
                     ),
                 )
 
-                self._register_task(decorated_task.task_name, decorated_task)
+                self._register_task(decorated_task.task_name, decorated_task)  # type: ignore
 
-                return decorated_task
+                return decorated_task  # type: ignore
 
             return inner
 
@@ -417,7 +417,7 @@ def add_event_handler(
 
     def with_result_backend(
         self,
-        result_backend: "AsyncResultBackend[_T]",
+        result_backend: "AsyncResultBackend[Any]",
     ) -> "Self":  # pragma: no cover
         """
         Set a result backend and return updated broker.

taskiq/kicker.py

@@ -228,6 +228,10 @@ def _prepare_arg(cls, arg: Any) -> Any:
         if isinstance(arg, BaseModel):
             arg = model_dump(arg)
         if is_dataclass(arg):
+            if isinstance(arg, type):
+                raise ValueError(
+                    f"Cannot serialize types. The {arg} is not serializable.",
+                )
             arg = asdict(arg)
         return arg
 

taskiq/labels.py

@@ -34,7 +34,7 @@ def prepare_label(label_value: Any) -> Tuple[str, int]:
     var_type = type(label_value)
     if var_type in (int, str, float, bool):
         return str(label_value), LabelType[var_type.__name__.upper()].value
-    if var_type == bytes:
+    if var_type is bytes:
         return base64.b64encode(label_value).decode(), LabelType.BYTES.value
     return str(label_value), LabelType.ANY.value
 

taskiq/serializers/cbor_serializer.py

@@ -1,5 +1,5 @@
 import datetime
-from typing import Any, Callable, Optional
+from typing import Any, Callable, Dict, Optional
 
 from taskiq.abc.serializer import TaskiqSerializer
 
@@ -26,8 +26,10 @@ def __init__(
         date_as_datetime: bool = True,
         string_referencing: bool = True,
         # Decoder options
-        tag_hook: "Optional[Callable[[cbor2.CborDecoder, Any], Any]]" = None,
-        object_hook: Optional[Callable[[Any], Any]] = None,
+        tag_hook: Optional[Callable[["cbor2.CBORDecoder", Any], Any]] = None,
+        object_hook: Optional[
+            Callable[["cbor2.CBORDecoder", Dict[Any, Any]], Any]
+        ] = None,
     ) -> None:
         if cbor2 is None:
             raise ImportError("cbor2 is not installed")
@@ -43,7 +45,7 @@ def __init__(
 
     def dumpb(self, value: Any) -> bytes:
         """Dump value to bytes."""
-        return cbor2.dumps(
+        return cbor2.dumps(  # type: ignore
             value,
             datetime_as_timestamp=self.datetime_as_timestamp,
             timezone=self.timezone,
@@ -56,7 +58,7 @@ def dumpb(self, value: Any) -> bytes:
 
     def loadb(self, value: bytes) -> Any:
         """Load value from bytes."""
-        return cbor2.loads(
+        return cbor2.loads(  # type: ignore
             value,
             tag_hook=self.tag_hook,
             object_hook=self.object_hook,

taskiq/state.py

@@ -1,7 +1,7 @@
 from collections import UserDict
 from typing import TYPE_CHECKING, Any
 
-if TYPE_CHECKING:  # pragma: no cover  # noqa: SIM108
+if TYPE_CHECKING:  # pragma: no cover
     _Base = UserDict[str, Any]
 else:
     _Base = UserDict

tests/receiver/test_receiver.py

@@ -186,7 +186,7 @@ def test_func() -> None:
     assert result.return_value is None
     assert result.is_err
     assert len(_TestMiddleware.found_exceptions) == 1
-    assert _TestMiddleware.found_exceptions[0].__class__ == ValueError
+    assert _TestMiddleware.found_exceptions[0].__class__ is ValueError
 
 
 @pytest.mark.anyio

tests/test_serialization.py

@@ -278,7 +278,6 @@ def test_exception_to_python_when_type_error() -> None:
 
 def test_json_context() -> None:
     error1 = ValueError("Context")
-    ValueError("Cause")
     error3 = ValueError("Error")
 
     try:
@@ -336,7 +335,6 @@ def test_json_cause() -> None:
 
 def test_pickle_context() -> None:
     error1 = ValueError("Context")
-    ValueError("Cause")
     error3 = ValueError("Error")
 
     try:

tests/utils.py

@@ -2,6 +2,7 @@
 from typing import AsyncGenerator
 
 from taskiq import AsyncBroker, BrokerMessage
+from taskiq.acks import AckableMessage
 
 
 class AsyncQueueBroker(AsyncBroker):
@@ -24,10 +25,8 @@ async def wait_tasks(self) -> None:
         """Small method to wait for all tasks to be processed."""
         await self.queue.join()
 
-    async def listen(self) -> AsyncGenerator[bytes, None]:
+    async def listen(self) -> AsyncGenerator[AckableMessage, None]:
         """This method returns all tasks from queue."""
         while True:
             task = await self.queue.get()
-            yield task
-            # Notify that task is done.
-            self.queue.task_done()
+            yield AckableMessage(data=task, ack=self.queue.task_done)