Skip to content

Commit

Permalink
Add job for continuing at last offset page.
Browse files Browse the repository at this point in the history
  • Loading branch information
emmiegit committed Jan 22, 2024
1 parent 6aaa042 commit 2ad2386
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 7 deletions.
7 changes: 1 addition & 6 deletions yellowstone/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,12 +68,7 @@ def queue_all_sites(self) -> None:
logger.info("Queueing site start jobs for '%s'", site_slug)
# XXX add_index_site_pages_job(site_slug)
# XXX add_index_site_forums_job(site_slug)
self.job.index_site_members(
{
"site_slug": site_slug,
"offset": START_MEMBER_OFFSET,
},
)
self.job.index_site_members_continue(site_slug)

def process_all_jobs(self) -> None:
logger.info("Processing all jobs in queue")
Expand Down
10 changes: 10 additions & 0 deletions yellowstone/job/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,16 @@ def index_site_forums(self, data: None) -> None:
def index_site_members(self, data: SiteMemberJob) -> None:
self.add_raw(JobType.INDEX_SITE_MEMBERS, cast(Json, data))

def index_site_members_continue(self, site_slug: str) -> None:
# Reads the last member page, and continues from there
offset = self.database.get_last_number_offset(site_slug=site_slug)
self.index_site_members(
{
"site_slug": site_slug,
"offset": offset,
},
)

def fetch_user(self, data: GetUserJob) -> None:
self.add_raw(JobType.FETCH_USER, cast(Json, data))

Expand Down
2 changes: 1 addition & 1 deletion yellowstone/job/index_site_members.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def run(core: "BackupDispatcher", data: SiteMemberJob) -> None:
site_slug = data["site_slug"]
offset = data["offset"]

assert offset >= START_OFFSET, "Offset cannot be zero or negative"
assert offset >= 1, "Offset cannot be zero or negative"
site_id = core.site_id_cache[site_slug]
logger.info(
"Retrieving page %d of site members from '%s' (%d)",
Expand Down

0 comments on commit 2ad2386

Please sign in to comment.