Fixed state class.

Signed-off-by: Pavel Kirilin <win10@list.ru>

Author: s3rius

taskiq/state.py

@@ -1,23 +1,34 @@
+from collections import UserDict
 from typing import Any
 
 
-class TaskiqState(dict[str, Any]):
+class TaskiqState(UserDict[str, Any]):
     """
     State class.
 
     This class is used to store useful variables
     for later use.
     """
 
+    def __init__(self) -> None:
+        self.__dict__["data"] = {}
+
     def __getattr__(self, name: str) -> Any:
         try:
-            return super().__getitem__(name)  # noqa: WPS613
+            return self.__dict__["data"][name]
         except KeyError:
             cls_name = self.__class__.__name__
             raise AttributeError(f"'{cls_name}' object has no attribute '{name}'")
 
     def __setattr__(self, name: str, value: Any) -> None:
-        super().__setitem__(name, value)  # noqa: WPS613
+        self[name] = value
 
     def __delattr__(self, name: str) -> None:  # noqa: WPS603
-        return super().__delitem__(name)  # noqa: WPS613
+        try:
+            del self[name]  # noqa: WPS420
+        except KeyError:
+            cls_name = self.__class__.__name__
+            raise AttributeError(f"'{cls_name}' object has no attribute '{name}'")
+
+    def __str__(self) -> str:
+        return "TaskiqState(%s)" % super().__str__()

taskiq/tests/test_state.py

@@ -0,0 +1,58 @@
+from taskiq.state import TaskiqState
+
+
+def test_state_set() -> None:
+    """Tests that you can sel values as dict items."""
+    state = TaskiqState()
+    state["a"] = 1
+
+    assert state["a"] == 1
+
+
+def test_state_get() -> None:
+    """Tests that you can get values as dict items."""
+    state = TaskiqState()
+
+    state["a"] = 1
+
+    assert state["a"] == 1
+
+
+def test_state_del() -> None:
+    """Tests that you can del values as dict items."""
+    state = TaskiqState()
+
+    state["a"] = 1
+
+    del state["a"]  # noqa: WPS420
+
+    assert state.get("a") is None
+
+
+def test_state_set_attr() -> None:
+    """Tests that you can set values by attribute."""
+    state = TaskiqState()
+
+    state.a = 1
+
+    assert state["a"] == 1
+
+
+def test_state_get_attr() -> None:
+    """Tests that you can get values by attribute."""
+    state = TaskiqState()
+
+    state["a"] = 1
+
+    assert state.a == 1
+
+
+def test_state_del_attr() -> None:
+    """Tests that you can delete values by attribute."""
+    state = TaskiqState()
+
+    state["a"] = 1
+
+    del state.a  # noqa: WPS420
+
+    assert state.get("a") is None