django batch queryset

batch_queryset.py

def batch_queryset(queryset, batchsize=2000):
    queryset = queryset.order_by("pk")
    total = queryset.count()
    with transaction.atomic():
        for start in range(0, total, batchsize):
            end = min(start + batchsize, total)
            yield queryset[start: end]