Django: Custom User Model과 1:N 관계 확장하기

01. Custom User Model

Custom User Model을 사용하는 이유

• Django의 기본 User Model은 대부분의 프로젝트에서 확장성이 부족할 수 있습니다.
• Custom User Model을 사용하면 필요에 따라 유저 모델을 자유롭게 확장하고, 프로젝트 요구사항에 맞출 수 있습니다.
• Django는 AUTH_USER_MODEL 설정을 통해 기본 User Model을 대체할 수 있습니다.

Custom User Model 구현하기

1. accounts/models.py

from django.db import models
from django.contrib.auth.models import AbstractUser

class User(AbstractUser):
    pass

• Django 기본 AbstractUser 클래스를 상속받아 Custom User Model을 생성합니다.

2. settings.py

...
AUTH_USER_MODEL = 'accounts.User'
...

• AUTH_USER_MODEL 설정을 통해 Custom User Model을 사용하도록 지정합니다.

3. accounts/admin.py

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import User

admin.site.register(User, UserAdmin)

• Django Admin에서 Custom User Model을 관리할 수 있도록 등록합니다.

4. forms.py

from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth import get_user_model

class CustomUserCreationForm(UserCreationForm):
    class Meta:
        model = get_user_model()
        fields = UserCreationForm.Meta.fields + ('필요시 추가 필드',)

• Django의 기본 UserCreationForm을 확장하여 Custom User Model을 지원합니다.

5. 마이그레이션 적용

• 데이터베이스를 초기화하고 마이그레이션을 다시 생성합니다.

$ python manage.py makemigrations accounts
$ python manage.py migrate

02. 1:N 관계 확장하기

User(1) - Article(N) 확장하기

1. articles/models.py

from django.db import models
from django.conf import settings

class Article(models.Model):
    title = models.CharField(max_length=200)
    content = models.TextField()
    author = models.ForeignKey(
        settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="articles"
    )
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

• author 필드를 통해 Custom User Model과 1:N 관계를 설정합니다.

2. articles/views.py

from django.shortcuts import render, redirect
from .models import Article
from .forms import ArticleForm

# 게시글 생성
def create(request):
    if request.method == "POST":
        form = ArticleForm(request.POST, request.FILES)
        if form.is_valid():
            article = form.save(commit=False)
            article.author = request.user
            article.save()
            return redirect("articles:article_detail", article.pk)
    else:
        form = ArticleForm()
    context = {"form": form}
    return render(request, "articles/create.html", context)

• 게시글 작성 시 현재 로그인한 사용자를 author 필드에 저장합니다.

3. 댓글 기능 확장

댓글 모델 추가 (comments/models.py)

from django.db import models
from django.conf import settings
from articles.models import Article

class Comment(models.Model):
    content = models.TextField()
    author = models.ForeignKey(
        settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="comments"
    )
    article = models.ForeignKey(
        Article, on_delete=models.CASCADE, related_name="comments"
    )
    created_at = models.DateTimeField(auto_now_add=True)

댓글 작성 (comments/views.py)

from django.shortcuts import redirect
from .models import Comment
from .forms import CommentForm

# 댓글 생성
def add_comment(request, article_id):
    if request.method == "POST" and request.user.is_authenticated:
        form = CommentForm(request.POST)
        if form.is_valid():
            comment = form.save(commit=False)
            comment.author = request.user
            comment.article_id = article_id
            comment.save()
            return redirect("articles:article_detail", article_id)
    else:
        return redirect("accounts:login")

댓글 목록에서 작성자 노출 (articles/templates/articles/detail.html)

<div>
    <h3>댓글</h3>
    {% for comment in article.comments.all %}
        <p>{{ comment.content }} - {{ comment.author }}</p>
        {% if comment.author == request.user %}
            <a href="{% url 'comments:delete_comment' comment.id %}">댓글 삭제</a>
        {% endif %}
    {% endfor %}
</div>

추가

1. 게시글 기능 확장

• 글 목록에서 작성자 표시:

{% for article in articles %}
    <p>{{ article.title }} - 작성자: {{ article.author }}</p>
{% endfor %}

• 게시글 삭제 및 수정 제한:

{% if article.author == request.user %}
    <a href="{% url 'articles:update' article.id %}">수정</a>
    <a href="{% url 'articles:delete' article.id %}">삭제</a>
{% endif %}

2. 댓글 기능 확장

• 댓글 작성자 노출:

{% for comment in comments %}
    <p>{{ comment.content }} - 작성자: {{ comment.author }}</p>
    {% if comment.author == request.user %}
        <a href="{% url 'comments:delete' comment.id %}">삭제</a>
    {% endif %}
{% endfor %}

• 비로그인 사용자 댓글 제한:

{% if user.is_authenticated %}
    <form method="POST" action="{% url 'comments:add' article.id %}">
        {% csrf_token %}
        {{ form.as_p }}
        <button type="submit">댓글 작성</button>
    </form>
{% else %}
    <p>댓글 작성은 로그인 후 가능합니다.</p>
{% endif %}

이 코드를 사용하면 Custom User Model 및 1:N 관계 확장을 통해 게시글과 댓글 작성자를 관리하고, 각 작성자별로 수정/삭제 권한을 제어할 수 있습니다.