mirrors/takahe
1
1
Fork 0
mirror of https://github.com/jointakahe/takahe.git synced 2024-09-29 05:41:55 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix home timeline pagination

Browse source
This commit is contained in:
Dan Watson 2023-03-09 21:46:57 -05:00 committed by GitHub
parent 3bd01b2b3d
commit 6e8149675c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
api/pagination.py
View file

@ -4,10 +4,11 @@ from collections.abc import Callable
from typing import Any, Generic, Protocol, TypeVar
from django.db import models
from django.db.models.expressions import Case, F, When
from django.http import HttpRequest
from hatchway.http import ApiResponse
from activities.models import PostInteraction
from activities.models import PostInteraction, TimelineEvent
T = TypeVar("T")
@ -247,27 +248,22 @@ class MastodonPaginator:
The home timeline requires special handling where we mix Posts and
PostInteractions together.
"""
queryset = queryset.annotate(
event_id=Case(
When(type=TimelineEvent.Types.post, then=F("subject_post_id")),
default=F("subject_post_interaction"),
)
)
if max_id and not max_id.startswith("interaction"):
queryset = queryset.filter(
models.Q(subject_post_id__lt=max_id)
| models.Q(subject_post_interaction_id__lt=max_id)
)
queryset = queryset.filter(event_id__lt=max_id)
if since_id and not since_id.startswith("interaction"):
queryset = queryset.filter(
models.Q(subject_post_id__gt=since_id)
| models.Q(subject_post_interaction_id__gt=since_id)
)
queryset = queryset.filter(event_id__gt=since_id)
if min_id and not min_id.startswith("interaction"):
# Min ID requires items _immediately_ newer than specified, so we
# invert the ordering to accommodate
queryset = queryset.filter(
models.Q(subject_post_id__gt=min_id)
| models.Q(subject_post_interaction_id__gt=min_id)
).order_by("id")
queryset = queryset.filter(event_id__gt=min_id).order_by("event_id")
else:
queryset = queryset.order_by("-id")
queryset = queryset.order_by("-event_id")
limit = min(limit or self.default_limit, self.max_limit)
return PaginationResult(