django bulk inserter

django-bulk-insert.py

class ModelBulkCreator:
    def __init__(self, cap=100):
        self.Model: Model = None
        self.cap = cap
        self._objects = []

    def add(self, obj):
        if not self.Model:
            self.Model = type(obj)
        self._objects.append(obj)
        if len(self._objects) >= self.cap:
            self.create_all()

    def create_all(self):
        if self.Model:
            log.info(f"Bulk inserting objects: Num - {len(self._objects)}")
            self.Model.objects.bulk_create(self._objects)
            self._objects.clear()