Merge pull request #19 from vvanglro/feat/package_version

Author: s3rius

.pre-commit-config.yaml

@@ -6,6 +6,7 @@ repos:
     hooks:
       - id: check-ast
       - id: trailing-whitespace
+        exclude: 'README.md'
       - id: check-toml
       - id: end-of-file-fixer
 

pyproject.toml

@@ -22,6 +22,7 @@ keywords = ["taskiq", "tasks", "distributed", "async", "aio-pika"]
 python = "^3.7"
 taskiq = ">=0.6.0,<1"
 aio-pika = "^9.0"
+importlib-metadata = {version = "^4.0.0", python = "<3.8"}
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.0"

taskiq_aio_pika/__init__.py

@@ -1,4 +1,13 @@
 """Taskiq integration with aio-pika."""
 from taskiq_aio_pika.broker import AioPikaBroker
 
+try:
+    # Python 3.8+
+    from importlib.metadata import version  # noqa: WPS433
+except ImportError:
+    # Python 3.7
+    from importlib_metadata import version  # noqa: WPS433, WPS440
+
+__version__ = version("taskiq-aio-pika")
+
 __all__ = ["AioPikaBroker"]

taskiq_aio_pika/broker.py

@@ -226,21 +226,20 @@ async def kick(self, message: BrokerMessage) -> None:
         if self.write_channel is None:
             raise ValueError("Please run startup before kicking.")
 
-        message_base_params: dict[str, Any] = {
+        message_base_params: Dict[str, Any] = {
             "body": message.message,
             "headers": {
                 "task_id": message.task_id,
                 "task_name": message.task_name,
                 **message.labels,
             },
             "delivery_mode": DeliveryMode.PERSISTENT,
+            "priority": parse_val(
+                int,
+                message.labels.get("priority"),
+            ),
         }
 
-        message_base_params["priority"] = parse_val(
-            int,
-            message.labels.get("priority"),
-        )
-
         delay: Optional[int] = parse_val(int, message.labels.get("delay"))
         rmq_message: Message = Message(**message_base_params)