Move logic from AsyncResult.from_publisher to Task._get_result

src/aiotaskq/task.py

@@ -45,22 +45,6 @@ def __init__(
         self.result = result
         self.error = error
 
-    @classmethod
-    async def from_publisher(cls, task_id: str) -> "AsyncResult":
-        """Return the result of the task once finished."""
-        from aiotaskq.serde import Serialization  # pylint: disable=import-outside-toplevel
-
-        pubsub_ = PubSub.get(url=Config.broker_url(), poll_interval_s=0.01)
-        async with pubsub_ as pubsub:  # pylint: disable=not-async-context-manager
-            await pubsub.subscribe(Constants.results_channel_template().format(task_id=task_id))
-            message: PollResponse = await pubsub.poll()
-
-        logger.debug("Message: %s", message)
-
-        result_serialized: bytes = message["data"]
-        result_: "AsyncResult" = Serialization.deserialize(cls, result_serialized)
-        return result_
-
     def get(self) -> RT | Exception:
         """Return the result of the task once finished."""
         if self.error is not None:
@@ -197,8 +181,19 @@ async def publish(self) -> None:
             await pubsub.publish(Constants.tasks_channel(), message=message)
 
     async def _get_result(self) -> RT:
+        from aiotaskq.serde import Serialization  # pylint: disable=import-outside-toplevel
+
         logger.debug("Retrieving result for task [task_id=%s]", self.id)
-        async_result: AsyncResult[RT] = await AsyncResult.from_publisher(task_id=self.id)
+        pubsub_ = PubSub.get(url=Config.broker_url(), poll_interval_s=0.01)
+        async with pubsub_ as pubsub:  # pylint: disable=not-async-context-manager
+            await pubsub.subscribe(Constants.results_channel_template().format(task_id=self.id))
+            message: PollResponse = await pubsub.poll()
+
+        logger.debug("Message: %s", message)
+
+        result_serialized: bytes = message["data"]
+        async_result: AsyncResult[RT] = Serialization.deserialize(AsyncResult, result_serialized)
+
         result: RT | Exception = async_result.get()
         if isinstance(result, Exception):
             raise result