Skip to content

Commit 7f47d0a

Browse files
committed
use pk instead of id in clear expired tokens
1 parent 0a5ffdb commit 7f47d0a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

oauth2_provider/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -712,9 +712,9 @@ def batch_delete(queryset, query):
712712
current_no = start_no = queryset.count()
713713

714714
while current_no:
715-
flat_queryset = queryset.values_list("id", flat=True)[:CLEAR_EXPIRED_TOKENS_BATCH_SIZE]
715+
flat_queryset = queryset.values_list("pk", flat=True)[:CLEAR_EXPIRED_TOKENS_BATCH_SIZE]
716716
batch_length = flat_queryset.count()
717-
queryset.model.objects.filter(id__in=list(flat_queryset)).delete()
717+
queryset.model.objects.filter(pk__in=list(flat_queryset)).delete()
718718
logger.debug(f"{batch_length} tokens deleted, {current_no - batch_length} left")
719719
queryset = queryset.model.objects.filter(query)
720720
time.sleep(CLEAR_EXPIRED_TOKENS_BATCH_INTERVAL)

0 commit comments

Comments
 (0)