code refactor

Author: vishesh10

tasktiger/_internal.py

@@ -27,6 +27,11 @@
     from .task import Task
     from .tasktiger import TaskTiger
 
+# Constants pertaining to Redis keys
+TASK = "task"
+EXECUTIONS = "executions"
+EXECUTIONS_COUNT = "executions_count"
+
 # Task states (represented by different queues)
 # Note some client code may rely on the string values (e.g. get_queue_stats).
 QUEUED = "queued"

tasktiger/constants.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING
 
-from .constants import EXECUTIONS, TASK
+from ._internal import EXECUTIONS, TASK
 from .utils import redis_glob_escape
 
 if TYPE_CHECKING:

tasktiger/migrations.py

@@ -3,8 +3,15 @@
 
 from redis import Redis
 
-from ._internal import ACTIVE, ERROR, QUEUED, SCHEDULED
-from .constants import EXECUTIONS, EXECUTIONS_COUNT, TASK
+from ._internal import (
+    ACTIVE,
+    ERROR,
+    EXECUTIONS,
+    EXECUTIONS_COUNT,
+    QUEUED,
+    SCHEDULED,
+    TASK,
+)
 
 try:
     from redis.commands.core import Script

tasktiger/redis_scripts.py

@@ -20,8 +20,11 @@
 
 from ._internal import (
     ERROR,
+    EXECUTIONS,
+    EXECUTIONS_COUNT,
     QUEUED,
     SCHEDULED,
+    TASK,
     g,
     gen_id,
     gen_unique_id,
@@ -30,7 +33,6 @@
     serialize_func_name,
     serialize_retry_method,
 )
-from .constants import EXECUTIONS, EXECUTIONS_COUNT, TASK
 from .exceptions import QueueFullException, TaskImportError, TaskNotFound
 from .runner import BaseRunner, get_runner_class
 from .types import RetryStrategy

tasktiger/task.py

@@ -36,8 +36,11 @@
 from ._internal import (
     ACTIVE,
     ERROR,
+    EXECUTIONS,
+    EXECUTIONS_COUNT,
     QUEUED,
     SCHEDULED,
+    TASK,
     dotted_parts,
     g,
     g_fork_lock,
@@ -47,7 +50,6 @@
     serialize_func_name,
     serialize_retry_method,
 )
-from .constants import EXECUTIONS, EXECUTIONS_COUNT, TASK
 from .exceptions import (
     RetryException,
     StopRetry,