From bb34581fd1c5b47e0fad9ce39435acd01bfd7da6 Mon Sep 17 00:00:00 2001
From: kennhh <133614589+kennhh@users.noreply.github.com>
Date: Fri, 14 Jul 2023 01:53:05 -0400
Subject: [PATCH] feat: add args, kwargs to task (#1478)

* modified:   interactions/models/internal/tasks/task.py

* reset to set_last_call_time

Signed-off-by: kennhh <133614589+kennhh@users.noreply.github.com>

---------

Signed-off-by: kennhh <133614589+kennhh@users.noreply.github.com>
---
 interactions/models/internal/tasks/task.py | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/interactions/models/internal/tasks/task.py b/interactions/models/internal/tasks/task.py
index 24bda8f2b..54c68551a 100644
--- a/interactions/models/internal/tasks/task.py
+++ b/interactions/models/internal/tasks/task.py
@@ -75,25 +75,25 @@ def on_error(self, error: Exception) -> None:
         self.on_error_sentry_hook(error)
         interactions.Client.default_error_handler("Task", error)
 
-    async def __call__(self) -> None:
+    async def __call__(self, *args, **kwargs) -> None:
         try:
             if inspect.iscoroutinefunction(self.callback):
-                val = await self.callback()
+                val = await self.callback(*args, **kwargs)
             else:
-                val = self.callback()
+                val = self.callback(*args, **kwargs)
 
             if isinstance(val, BaseTrigger):
                 self.reschedule(val)
         except Exception as e:
             self.on_error(e)
 
-    def _fire(self, fire_time: datetime) -> None:
+    def _fire(self, fire_time: datetime, *args, **kwargs) -> None:
         """Called when the task is being fired."""
         self.trigger.set_last_call_time(fire_time)
-        _ = asyncio.create_task(self())
+        _ = asyncio.create_task(self(*args, **kwargs))
         self.iteration += 1
 
-    async def _task_loop(self) -> None:
+    async def _task_loop(self, *args, **kwargs) -> None:
         """The main task loop to fire the task at the specified time based on triggers configured."""
         while not self._stop.is_set():
             fire_time = self.trigger.next_fire()
@@ -106,14 +106,14 @@ async def _task_loop(self) -> None:
             if future in done:
                 return None
 
-            self._fire(fire_time)
+            self._fire(fire_time, *args, **kwargs)
 
-    def start(self) -> None:
+    def start(self, *args, **kwargs) -> None:
         """Start this task."""
         try:
             self.trigger.reschedule()
             self._stop.clear()
-            self.task = asyncio.create_task(self._task_loop())
+            self.task = asyncio.create_task(self._task_loop(*args, **kwargs))
         except RuntimeError:
             get_logger().error(
                 "Unable to start task without a running event loop! We recommend starting tasks within an `on_startup` event."
@@ -125,10 +125,10 @@ def stop(self) -> None:
         if self.task:
             self.task.cancel()
 
-    def restart(self) -> None:
+    def restart(self, *args, **kwargs) -> None:
         """Restart this task."""
         self.stop()
-        self.start()
+        self.start(*args, **kwargs)
 
     def reschedule(self, trigger: BaseTrigger) -> None:
         """