diff --git a/oauth2_provider/models.py b/oauth2_provider/models.py index a76db37c0..b43c7f4d3 100644 --- a/oauth2_provider/models.py +++ b/oauth2_provider/models.py @@ -532,7 +532,7 @@ def revoke(self): self = list(token)[0] with suppress(access_token_model.DoesNotExist): - access_token_model.objects.get(id=self.access_token_id).revoke() + access_token_model.objects.get(pk=self.access_token_id).revoke() self.access_token = None self.revoked = timezone.now() @@ -712,9 +712,9 @@ def batch_delete(queryset, query): current_no = start_no = queryset.count() while current_no: - flat_queryset = queryset.values_list("id", flat=True)[:CLEAR_EXPIRED_TOKENS_BATCH_SIZE] + flat_queryset = queryset.values_list("pk", flat=True)[:CLEAR_EXPIRED_TOKENS_BATCH_SIZE] batch_length = flat_queryset.count() - queryset.model.objects.filter(id__in=list(flat_queryset)).delete() + queryset.model.objects.filter(pk__in=list(flat_queryset)).delete() logger.debug(f"{batch_length} tokens deleted, {current_no - batch_length} left") queryset = queryset.model.objects.filter(query) time.sleep(CLEAR_EXPIRED_TOKENS_BATCH_INTERVAL)