[Django tutorial] 사용자 인증

사용자가 작성하는 게시글

현재 로그인만 되어 있으면 누구라도 게시글에 사용자의 이름을 남겨서 작성할 수 있습니다. 여기서 몇 가지 문제가 있습니다.

1. 사용자의 이름이 겹친다면?
2. 작성한 글을 수정하고싶은데 이름이 같다면?

위 문제들이 있어 작성한글을 사용자의 이름만으로 구별하는 것은 좋은 방법이 아닙니다.

 

author이라는 필드를 수정해서 User모델의 pk값인 id값을 저장하도록 하면 실제로 저장된 사용자는 User모델에 저장된 사람에 한정되게 됩니다. - 마이그레이션은 아직 하지 않습니다!

# bbs/models.py

from django.db import models

class Article(models.Model):
    title= models.CharField('제목', max_length=126, null=False)
    content= models.TextField('내용', null=False)
    # author= models.CharField('작성자', max_length=16, null=False)
    author = models.PositiveIntegerField('작성자')
    created_at= models.DateTimeField('작성일', auto_now_add=True)

    def __str__(self):
        return '[{}] {}'.format(self.id, self.title)

이렇게하면 실제 데이터가 게시글에 등록될 때 등록하는 사용자의 id값이 저장됩니다.

1. 해당 게시글을 수정할 때는 로그인된 사용자의 id값을 author 필드와 비교해서 동일한 경우에만 수정이 됩니다. 
2. Aritcle모델에서 자기가 작성한 게시글만 필터링해서 볼 수 있습니다.
3. 게시글 목록에서도 author 값으로 User테이블을 검색해서 작성자의 이름을 표시할 수 있습니다.

 

이렇게 모델의 필드값을 다른 모델의 id로 사용할 떄의 장점이 있는 반면 단점도 분명히 존재합니다. 일반적으로 RDB(관계형 데이터 베이스)에서 이런 경우 모델과 모델에 제약조건을 추가하여 두 모델 간의 관계형 데이터베이스에 주입하고, 데이터베이스가 알아서 무결성을 보장하도록 설정합니다. Foreign Key라고 하는 외래키가 그것입니다.

 

 

외래키 (Foreign Key)

데이터 베이스에서는 외래키를 지정해서 테이블 간의 관계를 표시하는 것이 가능합니다. ForeignKey이라는 필드 타입을 제공하는데 sql의 그 ForeigmKey를 생성하는 기능을 제공합니다. 

 

ForeignKey 필드는 연결되어지는 모델과 해당 모델이 1:N관계가 됩니다. 하나의 Aritcle은 1명의 User가 작성을 할 수 있고, 1명의 User은 다수의 Ariticle을 작성할 수 있으니 User과 Aritcle은 1:N 관계가 됩니다. 

# bbs/models.py

from django.db import models
from django.utils import timezone

class Article(models.Model):
    title      = models.CharField('제목', max_length=126, null=False)
    content    = models.TextField('내용', null=False)
    # author     = models.CharField('작성자', max_length=16, null=False)
    author     = models.ForeignKey('user.User', related_name='articles', on_delete=models.CASCADE)
    created_at = models.DateTimeField('작성일', default=timezone.now)

    def __str__(self):
        return '[{}] {}'.format(self.id, self.title)

 

 

1. 모델 수정

ForeignKey 필드를 가장 전형적인 방식으로 선언했습니다.

1. 젓번째 모델의 인자로 ForeignKey로 연결한 모델의 식별자를 전달합니다.
2. 두 번째 인자 related_name는 ForeignKey에 설정되지는 모델에서 Article객체를 참조해야 할 때 사용하는 속성으로 사용됩니다.
   • 특정 사용자가 작성한 모든 게시글을 검색할때 user.articles.all()이라고 하면 User모델에 정의한 적 없는 article이라는 속성으로 추가되며 연관된 Article을 모두 검색할 수 있게 됩니다.
   • related_name를 설정해주지 않았다면 기본적으로 클래스이름 + '_set'로 related_name로 추가됩니다.
   • related_name를 추가해 주는 두 모델 사이에서 ForeignKey가 두 개 이상 존재하는 경우가 있습니다. 
3. on_delete속성은 ForeignKey가 연결되는 모델의 데이터가 삭제될 경우 해당 ForeignKey필드의 값을 어떻게 할지 에 대한 설정입니다. - 기본 값은 CASCADE입니다. (참고)

 

마이그레이션을 해보면 아래와 같은 오류가 뜹니다.

(venv) PS D:\DjangoTutorial\DjangoTutorial> py manage.py makemigrations
Migrations for 'bbs':
  bbs\migrations\0003_alter_article_author.py
    - Alter field author on article
(venv) PS D:\DjangoTutorial\DjangoTutorial> py manage.py migrate       
Operations to perform:
  Apply all migrations: admin, auth, bbs, contenttypes, sessions, user
Running migrations:
  Applying bbs.0003_alter_article_author...Traceback (most recent call last):
  File "D:\DjangoTutorial\DjangoTutorial\manage.py", line 22, in <module>
    main()
  File "D:\DjangoTutorial\DjangoTutorial\manage.py", line 18, in main   
    execute_from_command_line(sys.argv)
  File "D:\DjangoTutorial\venv\lib\site-packages\django\core\management\__init__.py", line 419, in execute_from_command_line
    utility.execute()
  File "D:\DjangoTutorial\venv\lib\site-packages\django\core\management\__init__.py", line 413, in execute
    self.fetch_command(subcommand).run_from_argv(self.argv)
  File "D:\DjangoTutorial\venv\lib\site-packages\django\core\management\base.py", line 354, in run_from_argv
    self.execute(*args, **cmd_options)
  File "D:\DjangoTutorial\venv\lib\site-packages\django\core\management\base.py", line 398, in execute
    output = self.handle(*args, **options)
  File "D:\DjangoTutorial\venv\lib\site-packages\django\core\management\base.py", line 89, in wrapped
    res = handle_func(*args, **kwargs)
  File "D:\DjangoTutorial\venv\lib\site-packages\django\core\management\commands\migrate.py", line 244, in handle
    post_migrate_state = executor.migrate(
  File "D:\DjangoTutorial\venv\lib\site-packages\django\db\migrations\executor.py", line 117, in migrate
    state = self._migrate_all_forwards(state, plan, full_plan, fake=fake, fake_initial=fake_initial)
  File "D:\DjangoTutorial\venv\lib\site-packages\django\db\migrations\executor.py", line 147, in _migrate_all_forwards
    state = self.apply_migration(state, migration, fake=fake, fake_initial=fake_initial)
  File "D:\DjangoTutorial\venv\lib\site-packages\django\db\migrations\executor.py", line 226, in apply_migration
    with self.connection.schema_editor(atomic=migration.atomic) as schema_editor:
  File "D:\DjangoTutorial\venv\lib\site-packages\django\db\backends\sqlite3\schema.py", line 35, in __exit__
    self.connection.check_constraints()
  File "D:\DjangoTutorial\venv\lib\site-packages\django\db\backends\sqlite3\base.py", line 353, in check_constraints
    raise IntegrityError(
django.db.utils.IntegrityError: The row in table 'bbs_article' with primary key '1' has an invalid foreign key: bbs_article.author_id contains a value 'jongs' that does not have a corresponding value in user_user.id.(venv) PS D:\DjangoTutorial\DjangoTutorial>

맨 밑에 오류 메시지를 보면

django.db.utils.IntegrityError: 기본 키가 '1'인 테이블 'bbs_article'의 행에 잘못된 외래 키가 있습니다. bbs_article.author_id에는 user_user.id에 해당 값이 없는 'jongs' 값이 포함되어 있습니다.

이라는 오류를 확인할 수 있는데 Foreignkey 필드로 생성된 필드에 이미 문자열 데이터가 들어가 있으니 User 모델의 id의 자료형이 맞지 않아 오류가 발생합니다. 

 

우선 데이터베이스가 변경되었는지 오류로 인해 롤백이 되었는지 확인해 봅니다.

 

 - 저는 잘 되지 않아서 결국 db를 초기화 후 진행하였습니다.

 

 

2. 뷰 수정

지금까지는 author이 문자열이었지만 이제는 숫자형이기 때문에 문자열이 저장되거나 문자열로 처리를 할 경우 오류가 발생할 수 있습니다.

 

뷰를 수정해 주는데 현재 사용 중인 뷰 중에서 author을 사용하는 뷰는 ArticleCreateUpdateView입니다. Post로 입력받은 author를 처리하지 않도록 수정하고 저장할 때 로그인한 사용자의 인스턴스를 article.author에 저장해 주도록 수정합니다. 

# bbs/views.py
class ArticleCreateUpdateView(LoginRequiredMixin, TemplateView):
    login_url = settings.LOGIN_URL
    template_name = 'article_update.html'
    queryset = Article.objects.all()
    pk_url_kwargs = 'article_id'
    
    def get_object(self, queryset=None):
        queryset = queryset or self.queryset
        pk = self.kwargs.get(self.pk_url_kwargs)
        article = queryset.filter(pk=pk).first()

        if pk:
            if not article:
                raise Http404('invalid pk')
            elif article.author != self.request.user:                             # 작성자가 수정하려는 사용자와 다른 경우 
                raise Http404('invalid user')
        return article
    def get(self, request, *args, **kwargs):
        article = self.get_object()

        ctx = {
            'article': article,
        }
        return self.render_to_response(ctx)

    def post(self, request, *args, **kwargs):
        action = request.POST.get('action')
        now = datetime.now()

        post_data = {key: request.POST.get(key) for key in ('title', 'content')}
        post_data['author'] = self.request.user
        post_data['created_at'] = now.strftime('%Y-%m-%d %H:%M:%S')
        print(post_data)
        for key in post_data:
            if not post_data[key]:
                messages.error(self.request, '{} 값이 존재하지 않습니다.'.format(key), extra_tags='danger')

        if len(messages.get_messages(request)) == 0:
            if action == 'create':
                article = Article.objects.create(**post_data)
                messages.success(self.request, '게시글이 저장되었습니다.')
            elif action == 'update':
                article = self.get_object()
                for key, value in post_data.items():
                    setattr(article, key, value)
                article.save()
                messages.success(self.request, '게시글이 수정되었습니다.')
            else:
                messages.error(self.request, '알 수 없는 요청입니다.', extra_tags='danger')

            return HttpResponseRedirect('/article/') # 정상적인 저장이 완료되면 '/articles/'로 이동됨

        ctx = {
            'article': self.get_object() if action == 'update' else None
        }
        return self.render_to_response(ctx)

 

 

3. 템플릿 수정

템플릿에서는 크게 두 가지를 수정해 주면 됩니다. User모델에 __str__함수가 미리 정의되어 있고, article.author라고 출력하더라도 이메일만 출력이 되어 기존의 article.author템플릿 변수를 수정할 필요는 없지만 <User: 0000@email.com>이라고 출력되는 것이 불편하기 때문에 article.author.email이라고 변경해줘야 합니다. article_update.html에서 작성자 입력란만 삭제해 주면 됩니다.

{% extends 'base.html' %}

{% block title %}
{% if article %}
<title>게시글 수정 - {{ article.pk }}. {{ article.title }}</title>
{% else %}
<title>게시글 작성</title>
{% endif %}
{% endblock title %}

{% block content %}
<form action="." method="post" class="form-horizontal">
{% csrf_token %}
<input type="hidden" name="action" value="{% if article %}update{% else %}create{% endif %}">
<table class="table table-striped table-bordered">
	<tr>
		<th>번호</th>
		<td>{{ article.pk }}</td>
	</tr>
	<tr>
		<th>제목</th>
		<td><input type="text" class="form-control" name="title" value="{{ article.title }}"></td>
	</tr>
	<tr>
		<th>내용</th>
		<td><textarea rows="10" class="form-control" name="content">{{ article.content }}</textarea></td>
	</tr>
    <!-- <tr> -->
		<!-- <th>작성자</th> -->
		<!-- <td><input type="text" class="form-control" name="author" value="{{ article.author }}"></td> -->
	<!-- </tr> -->
    <tr>
		<th>작성일</th>
		<td>{{ article.created_at | date:"Y-m-d H:i" }}</td>
	</tr>
</table>

<button class="btn btn-primary" type="submit">게시글 저장</button>
</form>
{% endblock content %}