Django에서 자기 참조 외래 키를 사용하는 방법

2024-07-27

자기 참조 외래 키 정의

자기 참조 외래 키를 정의하려면 다음과 같이 ForeignKey 필드를 모델에 추가합니다.

from django.db import models

class MyModel(models.Model):
    # ... 다른 필드들 ...
    parent = models.ForeignKey('self', on_delete=models.CASCADE, null=True, blank=True)

이 코드는 MyModel이라는 모델을 정의하며, 이 모델에는 parent라는 이름의 ForeignKey 필드가 있습니다. 이 필드는 같은 MyModel 모델의 다른 인스턴스를 참조합니다. on_delete 매개변수는 부모 인스턴스가 삭제될 때 자식 인스턴스가 어떻게 처리되는지 제어합니다. null 및 blank 매개변수는 필드가 NULL 값을 허용할지 여부를 제어합니다.

자기 참조 외래 키를 사용하면 다음과 같이 모델 인스턴스 간의 관계를 만들 수 있습니다.

parent_instance = MyModel.objects.get(pk=1)
child_instance = MyModel.objects.create(
    # ... 다른 필드 데이터 ...
    parent=parent_instance
)

이 코드는 pk가 1인 MyModel 인스턴스를 parent_instance 변수에 저장합니다. 그런 다음 새 MyModel 인스턴스를 만들고 parent 필드를 parent_instance로 설정합니다.

다음은 자기 참조 외래 키를 사용하는 몇 가지 일반적인 예시입니다.

계층 구조 모델링: 조직 구조 또는 파일 시스템 계층 구조와 같은 계층 구조를 모델링하는 데 사용할 수 있습니다.
트리 구조 모델링: 댓글 시스템 또는 포럼 스레드와 같은 트리 구조를 모델링하는 데 사용할 수 있습니다.
순환 관계 모델링: 서로 참조하는 두 개의 모델 간의 순환 관계를 모델링하는 데 사용할 수 있습니다.

주의 사항

자기 참조 외래 키를 사용할 때는 다음과 같은 몇 가지 주의 사항을 기억해야 합니다.

무한 루프 방지: 자기 참조 외래 키를 사용하면 모델에서 무한 루프가 발생할 수 있습니다. 이를 방지하려면 Django의 prevent_loop 유효성 검사기를 사용해야 합니다.
삭제 캐스케이드 주의: on_delete=models.CASCADE를 사용하면 부모 인스턴스가 삭제될 때 자식 인스턴스도 삭제됩니다. 이는 원하지 않는 데이터 손실로 이어질 수 있으므로 주의해서 사용해야 합니다.

Django 자기 참조 외래 키 예제 코드

models.py

from django.db import models

class Category(models.Model):
    name = models.CharField(max_length=255)
    parent = models.ForeignKey('self', on_delete=models.CASCADE, null=True, blank=True)

이 코드는 Category라는 모델을 정의합니다. 이 모델에는 name 필드와 parent라는 자기 참조 외래 키 필드가 있습니다. parent 필드는 같은 Category 모델의 다른 인스턴스를 참조할 수 있습니다.

views.py

from django.shortcuts import render

from .models import Category

def create_category(request):
    if request.method == 'POST':
        name = request.POST['name']
        parent_id = request.POST.get('parent')

        if parent_id:
            parent = Category.objects.get(pk=parent_id)
        else:
            parent = None

        category = Category.objects.create(
            name=name,
            parent=parent
        )

        return render(request, 'category_created.html', {'category': category})
    else:
        categories = Category.objects.all()
        return render(request, 'create_category.html', {'categories': categories})

이 코드는 카테고리를 만들고 부모 카테고리에 할당하는 데 사용되는 create_category 뷰 함수를 정의합니다.

templates/create_category.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Create Category</title>
</head>
<body>
    <h1>Create Category</h1>
    <form method="post">
        {% csrf_token %}
        <label for="name">Name:</label>
        <input type="text" id="name" name="name">

        <label for="parent">Parent:</label>
        <select id="parent" name="parent">
            <option value="">None</option>
            {% for category in categories %}
                <option value="{{ category.pk }}">{{ category.name }}</option>
            {% endfor %}
        </select>

        <button type="submit">Create</button>
    </form>
</body>
</html>

이 템플릿은 사용자가 카테고리 이름과 부모 카테고리를 선택할 수 있는 양식을 제공합니다.

templates/category_created.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Category Created</title>
</head>
<body>
    <h1>Category Created</h1>
    <p>Name: {{ category.name }}</p>
    {% if category.parent %}
        <p>Parent: {{ category.parent.name }}</p>
    {% else %}
        <p>Parent: None</p>
    {% endif %}
</body>
</html>

이 템플릿은 새로 생성된 카테고리의 이름과 부모 카테고리(있는 경우)를 표시합니다.

이 코드는 Django에서 자기 참조 외래 키를 사용하는 기본적인 방법을 보여줍니다. 실제 응용 프로그램에서는 더 복잡한 로직이 필요할 수 있습니다.

추가 예제

계층 구조 모델링: 조직 구조 또는 파일 시스템 계층 구조와 같은 계층 구조를 모델링하는 데 사용할 수 있습니다. 예를 들어 다음과 같은 코드를 사용하여 직원 계층 구조를 모델링할 수 있습니다.

class Employee(models.Model):
    name = models.CharField(max_length=255)
    manager = models.ForeignKey('self', on_delete=models.CASCADE, null=True, blank=True)

Django에서 자기 참조 외래 키 대체 방법

다대다 관계 사용

자기 참조 외래 키 대신 다대다 관계를 사용할 수 있습니다. 이를 위해 다음과 같이 두 개의 모델을 정의하고 ManyToManyField를 사용하여 연결해야 합니다.

from django.db import models

class Category(models.Model):
    name = models.CharField(max_length=255)

class ParentCategory(models.Model):
    category = models.ForeignKey(Category, on_delete=models.CASCADE)
    parent = models.ForeignKey(Category, on_delete=models.CASCADE, related_name='children')

이 코드는 Category라는 두 개의 모델을 정의합니다. ParentCategory 모델에는 category 필드와 parent 필드가 있습니다. category 필드는 자식 카테고리를 참조하고 parent 필드는 부모 카테고리를 참조합니다.

중첩 모델 사용

from django.db import models
from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType

class Category(models.Model):
    name = models.CharField(max_length=255)
    parent_content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    parent_object_id = models.PositiveIntegerField()
    parent = GenericForeignKey('parent_content_type', 'parent_object_id')

이 코드는 Category라는 모델을 정의합니다. 이 모델에는 parent_content_type 필드와 parent_object_id 필드가 있습니다. parent_content_type 필드는 부모 객체의 콘텐츠 유형을 참조하고 parent_object_id 필드는 부모 객체의 ID를 참조합니다. parent GenericForeignKey는 두 필드를 함께 사용하여 부모 객체를 참조합니다.

추가 고려 사항

자기 참조 외래 키를 대체 방법으로 사용하기 전에 다음 사항을 고려해야 합니다.