render_messages: Tighten function signatures with generic QuerySet.

Signed-off-by: Zixuan James Li <p359101898@gmail.com>
This commit is contained in:
Zixuan James Li
2022-06-15 17:58:40 -04:00
committed by Tim Abbott
parent 32ad110af2
commit 8f80376501

View File

@@ -9,7 +9,7 @@ from zerver.lib.message import render_markdown
from zerver.models import Message
def queryset_iterator(queryset: QuerySet, chunksize: int = 5000) -> Iterator[Any]:
def queryset_iterator(queryset: QuerySet[Message], chunksize: int = 5000) -> Iterator[Message]:
queryset = queryset.order_by("id")
while queryset.exists():
for row in queryset[:chunksize]: