relocate class CeleryBeat

Author: SKairinos

codeforlife/settings/third_party.py

@@ -7,7 +7,7 @@
 
 import typing as t
 
-from ..types import CeleryBeat
+from ..tasks import CeleryBeat
 from .custom import REDIS_URL, SERVICE_SITE_URL
 from .django import ENV
 

codeforlife/tasks.py

@@ -10,6 +10,33 @@
 from celery import shared_task as _shared_task
 from django.conf import settings
 
+from .types import Args, KwArgs
+
+if t.TYPE_CHECKING:
+    from celery.schedules import crontab, solar
+
+
+class CeleryBeat(t.Dict[str, t.Any]):
+    """A Celery beat schedule.
+
+    https://docs.celeryq.dev/en/v5.4.0/userguide/periodic-tasks.html
+    """
+
+    def __init__(
+        self,
+        task: str,
+        schedule: t.Union[int, "crontab", "solar"],
+        args: t.Optional[Args] = None,
+        kwargs: t.Optional[KwArgs] = None,
+    ):
+        super().__init__()
+        self["task"] = task
+        self["schedule"] = schedule
+        if args:
+            self["args"] = args
+        if kwargs:
+            self["kwargs"] = kwargs
+
 
 def shared_task(*args, **kwargs):
     """

codeforlife/types.py

@@ -7,10 +7,6 @@
 
 import typing as t
 
-if t.TYPE_CHECKING:
-    from celery.schedules import crontab, solar
-
-
 CookieSamesite = t.Optional[t.Literal["Lax", "Strict", "None", False]]
 
 Env = t.Literal["local", "development", "staging", "production"]
@@ -37,18 +33,6 @@
 ]
 
 
-class CeleryBeat(t.NamedTuple):
-    """A Celery beat schedule.
-
-    https://docs.celeryq.dev/en/v5.4.0/userguide/periodic-tasks.html
-    """
-
-    task: str
-    schedule: t.Union[int, "crontab", "solar"]
-    args: t.Optional[Args] = tuple()
-    kwargs: t.Optional[KwArgs] = {}
-
-
 def get_arg(cls: t.Type[t.Any], index: int, orig_base: int = 0):
     """Get a type arg from a class.