Modify the get_locked method in DBTaskResultQuerySet to allow retriev…

…al of multiple locked jobs at once.

django_tasks/backends/database/models.py

@@ -1,13 +1,13 @@
 import datetime
 import logging
 import uuid
-from typing import TYPE_CHECKING, Any, Generic, Optional, TypeVar
+from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
 import django
 from django.conf import settings
 from django.core.exceptions import SuspiciousOperation
 from django.db import models
-from django.db.models import F, Q
+from django.db.models import F, Q, QuerySet
 from django.db.models.constraints import CheckConstraint
 from django.utils import timezone
 from django.utils.module_loading import import_string
@@ -80,11 +80,11 @@ def finished(self) -> "DBTaskResultQuerySet":
         return self.failed() | self.succeeded()
 
     @retry()
-    def get_locked(self) -> Optional["DBTaskResult"]:
+    def get_locked(self, size: int = 1) -> QuerySet["DBTaskResult"]:
         """
         Get a job, locking the row and accounting for deadlocks.
         """
-        return self.select_for_update(skip_locked=True).first()
+        return self.select_for_update(skip_locked=True)[:size]
 
 
 class DBTaskResult(GenericBase[P, T], models.Model):