feat: add package version (#145)

Author: vvanglro

taskiq/__init__.py

@@ -31,7 +31,16 @@
 from taskiq.state import TaskiqState
 from taskiq.task import AsyncTaskiqTask
 
+try:
+    # Python 3.8+
+    from importlib.metadata import version  # noqa: WPS433
+except ImportError:
+    # Python 3.7
+    from importlib_metadata import version  # noqa: WPS433
+
+__version__ = version("taskiq")
 __all__ = [
+    "__version__",
     "gather",
     "Context",
     "AsyncBroker",

taskiq/__main__.py

@@ -2,8 +2,9 @@
 import sys
 from typing import Dict
 
-from importlib_metadata import entry_points, version
+from importlib_metadata import entry_points
 
+from taskiq import __version__
 from taskiq.abc.cmd import TaskiqCMD
 
 
@@ -60,7 +61,7 @@ def main() -> None:  # noqa: WPS210  # pragma: no cover
     args, _ = parser.parse_known_args()
 
     if args.version:
-        print(version("taskiq"))  # noqa: WPS421
+        print(__version__)  # noqa: WPS421
         return
 
     if args.subcommand is None:

tests/cli/worker/test_log_collector.py

@@ -9,8 +9,8 @@ def test_log_collector_std_success() -> None:
     """Tests that stdout and stderr calls are collected correctly."""
     log = StringIO()
     with log_collector(log, "%(message)s"):
-        print("log1")  # noqa: WPS421
-        print("log2", file=sys.stderr)  # noqa: WPS421
+        print("log1")
+        print("log2", file=sys.stderr)
     assert log.getvalue() == "log1\nlog2\n"
 
 

tests/receiver/test_receiver.py

@@ -162,7 +162,7 @@ async def test_callback_success() -> None:
 
     @broker.task
     async def my_task() -> int:
-        nonlocal called_times  # noqa: WPS420
+        nonlocal called_times
         called_times += 1
         return 1
 
@@ -415,7 +415,7 @@ async def test_callback_semaphore() -> None:
 
     @broker.task
     async def task_sem() -> int:
-        nonlocal sem_num  # noqa: WPS420
+        nonlocal sem_num
         sem_num += 1
         await asyncio.sleep(1)
         return 1

tests/test_state.py

@@ -24,7 +24,7 @@ def test_state_del() -> None:
 
     state["a"] = 1
 
-    del state["a"]  # noqa: WPS420
+    del state["a"]
 
     assert state.get("a") is None
 
@@ -53,6 +53,6 @@ def test_state_del_attr() -> None:
 
     state["a"] = 1
 
-    del state.a  # noqa: WPS420
+    del state.a
 
     assert state.get("a") is None