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

올려주신 코드에서 ManyToMany 관계를 맺기 전에 pk(즉, id) 값이 필요하기 때문에 bulk_create를 수행하고 tag_set.add 매서드를 호출한다고 강의에서 설명해주셨습니다.

def create_posts(orig_post_list):
    """포스팅 생성"""

    category_dict = {category.name: category for category in Category.objects.all()}
    tag_dict = {tag.name: tag for tag in Tag.objects.all()}

    user_list = list(User.objects.all())

    post_list = []
    for orig_post in orig_post_list:
        post = Post(
            category=category_dict[orig_post["category_name"]],
            author=choice(user_list),
            title=orig_post["title"],
            status=choice([Post.Status.DRAFT, Post.Status.PUBLISHED]),
            content=orig_post["content"],
        )
        post._tag_list = orig_post["tag_list"]
        post.slugify()
        post_list.append(post)

    if post_list:
        print(f"{len(post_list)} 개의 포스팅 생성")
        Post.objects.bulk_create(post_list, batch_size=1000)

        for post in post_list:
            _tag_list = [tag_dict[tag_name] for tag_name in post._tag_list]
            post.tag_set.add(*_tag_list)

그러나 해당 코드를 실행 시 여전히 tag_set.add를 위해서는 id가 필요하다고 에러가 발생합니다.

사용 중인 DB는 mysql 이며, 일단 임시 변통으로 bulk_create 전에 id를 직접 할당하여 해결하였습니다.

for idx, orig_post in enumerate(orig_post_list):
    post.id = idx + 1 # id는 0이 될 수 없음  

그러나 이는 --clear argument를 주었기 때문에 가능한 행위였습니다.

수정하기 위해서 하기와 같은 방법이 적절할까요? 그리고 왜 제 경우만 에러가 발생하는 걸까요?

    if post_list:
        print(f"{len(post_list)} 개의 포스팅 생성")
        Post.objects.bulk_create(post_list, batch_size=1000)

        tag_lists = [t for t in orig_post["tag_list"]]
        post_list = Post.objects.all() # id 로드를 위함
      
        for post, tag_list in zip(post_list, tag_lists):
            _tag_list = [tag_dict[tag_name] for tag_name in tag_list]
            post.tag_set.add(*_tag_list)