diff --git a/yellowstone/job/__init__.py b/yellowstone/job/__init__.py index 0dbfa84..21b57f2 100644 --- 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 index 511122a..1944807 100644 --- 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 index fd1f82b..2a945be 100644 --- 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