인프런 커뮤니티 질문&답변

from django.db import models


class Post(models.Model):
    title = models.CharField(max_length=200)
    content = models.TextField()


class Comment(models.Model):
    post = models.ForeignKey(Post, on_delete=models.CASCADE)
    content = models.TextField()

import itertools
from typing import Iterator, Tuple

from django.core.management import BaseCommand
from django.db import transaction

from blog.models import Post, Comment


class Command(BaseCommand):

    help = "test"

    def handle(self, *args, **options):

        generate_post_instances = (
            Post(title=title, content=content)
            for title, content in self.generate_post_row()
        )

        with transaction.atomic():
            for chunks in self.get_chunks(generate_post_instances, chunk_size=1000):
                Post.objects.bulk_create(chunks, ignore_conflicts=True)

            for post_instance in Post.objects.all():

                generate_comment_instance = (
                    Comment(post=post_instance, content=content)
                    for content in self.generate_comment_row()
                )

                for chunks in self.get_chunks(
                    generate_comment_instance, chunk_size=1000
                ):
                    Comment.objects.bulk_create(chunks, ignore_conflicts=True)

    @classmethod
    def generate_post_row(cls, size: int = 1000) -> Iterator[Tuple[str, str]]:
        for i in range(1, size + 1):
            yield f"Test post {i}", f"Test content {i}"

    @classmethod
    def generate_comment_row(cls, size: int = 100) -> Iterator[str]:
        for i in range(1, size + 1):
            yield f"Test comment {i}"

    @classmethod
    def get_chunks(cls, iterable: Iterator, chunk_size: int = 100) -> Iterator:
        iterator = iterable if hasattr(iterable, "__next__") else iter(iterable)
        for first in iterator:
            yield itertools.chain([first], itertools.islice(iterator, chunk_size - 1))

안녕하세요.
기존 강의에서 다루시던 대용량 데이터 처리 관련 코드를 참조하여 좀 더 이해하고자 커스텀해서 예시코드를 작성하게 되었습니다.

제너레이터, bulk_create 메서드에 대해서 질문이 있습니다.

1. 제너레이터는 yield 한 다음 반복문을 이용해 호출합니다. 제너레이트를 호출 한다면 호출할 때 반복문이 여러개 겹치게 되는데 이때 시간복잡도는 어떻게 되는건가요?

[generate_post_row 호출 > generate_post_instances 호출 > get_chunks > 호출] 이때 반복문을 많이 사용하게 되는데 어떤식으로 처리가 되는지 궁금합니다.

2. transaction.atomic() 에서 먼저 Post을 bulk_create로 생성한 뒤 해당 Post의 데이터들을 Comment 인스턴스에 넘겨주어 bulk_create를 사용하였는데. 혹시 더 나은 방법이 있을까요?