Add handling for job types.

scpwiki · Jan 29, 2024 · 2d3dcee · 2d3dcee
1 parent 8609af2
commit 2d3dcee
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 3 deletions.
diff --git a/yellowstone/job/__init__.py b/yellowstone/job/__init__.py
@@ -14,7 +14,13 @@
 
 from ..exception import UnknownJobError
 from ..types import Json
-from . import get_user, get_user_avatar, index_site_members
+from . import (
+    get_user,
+    get_user_avatar,
+    index_forum_categories,
+    index_forum_threads,
+    index_site_members,
+)
 from .get_user import GetUserJob
 from .get_user_avatar import GetUserAvatarJob
 from .index_forum_threads import ForumThreadsJob
@@ -116,6 +122,16 @@ def process(self, core: "BackupDispatcher", job: JobDict) -> None:
                         core,
                         cast(index_site_members.SiteMemberJob, data),
                     )
+                case JobType.INDEX_FORUM_CATEGORIES:
+                    index_forum_categories.run(
+                        core,
+                        cast(index_forum_categories.ForumCategoriesJob, data),
+                    )
+                case JobType.INDEX_FORUM_THREADS:
+                    index_forum_threads.run(
+                        core,
+                        cast(index_forum_threads.ForumThreadsJob, data),
+                    )
                 case JobType.FETCH_USER:
                     get_user.run(
                         core,

diff --git a/yellowstone/job/index_forum_categories.py b/yellowstone/job/index_forum_categories.py
@@ -26,7 +26,7 @@ class ForumCategoryProgressRow(TypedDict):
     last_post_id: Optional[int]
 
 
-def run(core: "BackupDispatcher", *, data: ForumCategoriesJob) -> None:
+def run(core: "BackupDispatcher", data: ForumCategoriesJob) -> None:
     site_slug = data["site_slug"]
 
     # Clear out previous forum groups

diff --git a/yellowstone/job/index_forum_threads.py b/yellowstone/job/index_forum_threads.py
@@ -19,6 +19,6 @@ class ForumThreadsJob(TypedDict):
     category_id: int
 
 
-def run(core: "BackupDispatcher", *, data: ForumThreadsJob) -> None:
+def run(core: "BackupDispatcher", data: ForumThreadsJob) -> None:
     # TODO
     pass