chore: enable ruff for taskiq/compat.py

Author: danfimov

pyproject.toml

@@ -177,6 +177,9 @@ lint.mccabe = { max-complexity = 10 }
 line-length = 88
 
 [tool.ruff.lint.per-file-ignores]
+"taskiq/compat.py" = [
+    "D103",   # Missing docstring in public function
+]
 "tests/*" = [
 	"S101",   # Use of assert detected
 	"S301",   # Use of pickle detected

taskiq/compat.py

@@ -1,38 +1,38 @@
-# flake8: noqa
+from collections.abc import Hashable
 from functools import lru_cache
 from importlib.metadata import version
-from typing import Any, Dict, Hashable, Optional, Type, TypeVar, Union
+from typing import Any, TypeVar
 
 import pydantic
 from packaging.version import Version, parse
 
 PYDANTIC_VER = parse(version("pydantic"))
 
 Model = TypeVar("Model", bound="pydantic.BaseModel")
-IS_PYDANTIC2 = PYDANTIC_VER >= Version("2.0")
+IS_PYDANTIC2 = Version("2.0") <= PYDANTIC_VER
 
 if IS_PYDANTIC2:
     T = TypeVar("T", bound=Hashable)
 
-    @lru_cache()
-    def create_type_adapter(annot: Type[T]) -> pydantic.TypeAdapter[T]:
+    @lru_cache
+    def create_type_adapter(annot: type[T]) -> pydantic.TypeAdapter[T]:
         return pydantic.TypeAdapter(annot)
 
-    def parse_obj_as(annot: Type[T], obj: Any) -> T:
+    def parse_obj_as(annot: type[T], obj: Any) -> T:
         return create_type_adapter(annot).validate_python(obj)
 
     def model_validate(
-        model_class: Type[Model],
-        message: Dict[str, Any],
+        model_class: type[Model],
+        message: dict[str, Any],
     ) -> Model:
         return model_class.model_validate(message)
 
-    def model_dump(instance: Model) -> Dict[str, Any]:
+    def model_dump(instance: Model) -> dict[str, Any]:
         return instance.model_dump(mode="json")
 
     def model_validate_json(
-        model_class: Type[Model],
-        message: Union[str, bytes, bytearray],
+        model_class: type[Model],
+        message: str | bytes | bytearray,
     ) -> Model:
         return model_class.model_validate_json(message)
 
@@ -41,7 +41,7 @@ def model_dump_json(instance: Model) -> str:
 
     def model_copy(
         instance: Model,
-        update: Optional[Dict[str, Any]] = None,
+        update: dict[str, Any] | None = None,
         deep: bool = False,
     ) -> Model:
         return instance.model_copy(update=update, deep=deep)
@@ -52,17 +52,17 @@ def model_copy(
     parse_obj_as = pydantic.parse_obj_as  # type: ignore
 
     def model_validate(
-        model_class: Type[Model],
-        message: Dict[str, Any],
+        model_class: type[Model],
+        message: dict[str, Any],
     ) -> Model:
         return model_class.parse_obj(message)
 
-    def model_dump(instance: Model) -> Dict[str, Any]:
+    def model_dump(instance: Model) -> dict[str, Any]:
         return instance.dict()
 
     def model_validate_json(
-        model_class: Type[Model],
-        message: Union[str, bytes, bytearray],
+        model_class: type[Model],
+        message: str | bytes | bytearray,
     ) -> Model:
         return model_class.parse_raw(message)  # type: ignore[arg-type]
 
@@ -71,7 +71,7 @@ def model_dump_json(instance: Model) -> str:
 
     def model_copy(
         instance: Model,
-        update: Optional[Dict[str, Any]] = None,
+        update: dict[str, Any] | None = None,
         deep: bool = False,
     ) -> Model:
         return instance.copy(update=update, deep=deep)