lesson 59

ap_12/accounts/admin.py

 from django.contrib import admin
+from accounts.models import User
 
-# Register your models here.
+
+admin.site.register(User)

ap_12/fixtures/auth_dump.json

@@ -3,112 +3,139 @@
     "model": "auth.group",
     "pk": 1,
     "fields": {
-        "name": "moderators",
+        "name": "Moderators",
         "permissions": [
             [
-                "add_group",
-                "auth",
-                "group"
+                "change_article",
+                "webapp",
+                "article"
             ],
             [
-                "change_group",
-                "auth",
-                "group"
+                "view_article",
+                "webapp",
+                "article"
             ],
             [
-                "delete_group",
-                "auth",
-                "group"
+                "change_articletag",
+                "webapp",
+                "articletag"
             ],
             [
-                "view_group",
-                "auth",
-                "group"
+                "view_articletag",
+                "webapp",
+                "articletag"
             ],
             [
-                "add_permission",
-                "auth",
-                "permission"
+                "change_comment",
+                "webapp",
+                "comment"
             ],
             [
-                "change_permission",
-                "auth",
-                "permission"
+                "view_comment",
+                "webapp",
+                "comment"
             ],
             [
-                "delete_permission",
-                "auth",
-                "permission"
+                "change_tag",
+                "webapp",
+                "tag"
+            ],
+            [
+                "view_tag",
+                "webapp",
+                "tag"
             ]
         ]
     }
 },
 {
-    "model": "auth.user",
+    "model": "accounts.user",
     "pk": 1,
     "fields": {
-        "password": "pbkdf2_sha256$260000$lq4ksTNG7r59t43pWudHUh$x7l77+SRwPzLJGjrmRiF7Q2f0ZoOsoa0M1jmnjaYOp4=",
-        "last_login": "2024-02-08T14:08:46.334Z",
+        "password": "pbkdf2_sha256$260000$8UtRanhkO5mewMnDbwLFAG$I7SXHyc1AkRQ75IHXi7mCEs5rZ96WA/jtQPlwqPw4wU=",
+        "last_login": "2024-02-26T14:40:30.384Z",
         "is_superuser": true,
         "username": "admin",
-        "first_name": "john",
-        "last_name": "doe",
+        "first_name": "",
+        "last_name": "",
         "email": "admin@gmail.com",
         "is_staff": true,
         "is_active": true,
-        "date_joined": "2023-12-18T14:33:11Z",
-        "groups": [
-            [
-                "moderators"
-            ]
-        ],
+        "date_joined": "2024-02-26T13:48:18Z",
+        "groups": [],
         "user_permissions": [
             [
-                "add_logentry",
-                "admin",
-                "logentry"
+                "add_article",
+                "webapp",
+                "article"
             ],
             [
-                "change_logentry",
-                "admin",
-                "logentry"
+                "change_article",
+                "webapp",
+                "article"
             ],
             [
-                "delete_logentry",
-                "admin",
-                "logentry"
+                "delete_article",
+                "webapp",
+                "article"
             ],
             [
-                "view_logentry",
-                "admin",
-                "logentry"
-            ],
-            [
-                "add_group",
-                "auth",
-                "group"
-            ],
-            [
-                "change_group",
-                "auth",
-                "group"
-            ],
+                "view_article",
+                "webapp",
+                "article"
+            ]
+        ]
+    }
+},
+{
+    "model": "accounts.user",
+    "pk": 4,
+    "fields": {
+        "password": "pbkdf2_sha256$260000$k60ituf6lev5ZiilBXiiOz$og/LlRwDzSHvn6NehUOb+mA4oySMvSHxnFahP3X+XVs=",
+        "last_login": "2024-02-26T14:44:12.429Z",
+        "is_superuser": false,
+        "username": "john",
+        "first_name": "john",
+        "last_name": "doe",
+        "email": "johndoe@gmail.com",
+        "is_staff": false,
+        "is_active": true,
+        "date_joined": "2024-02-26T14:07:22Z",
+        "groups": [
             [
-                "delete_group",
-                "auth",
-                "group"
+                "Moderators"
+            ]
         ],
+        "user_permissions": [
             [
-                "view_group",
-                "auth",
-                "group"
+                "add_article",
+                "webapp",
+                "article"
             ],
             [
-                "add_permission",
-                "auth",
-                "permission"
+                "delete_article",
+                "webapp",
+                "article"
             ]
         ]
     }
+},
+{
+    "model": "accounts.user",
+    "pk": 5,
+    "fields": {
+        "password": "pbkdf2_sha256$260000$TWe6vsT6TMp0WiTk2Y9iYp$8UztS9GRERsEgobGp04eXnH92vhPcEUqFc0MBKdWOEM=",
+        "last_login": "2024-02-26T14:47:09.814Z",
+        "is_superuser": false,
+        "username": "jane",
+        "first_name": "jane",
+        "last_name": "doe",
+        "email": "janedoe@gmail.com",
+        "is_staff": false,
+        "is_active": true,
+        "date_joined": "2024-02-26T14:07:36.807Z",
+        "groups": [],
+        "user_permissions": []
+    }
 }
 ]

ap_12/templates/articles/detail.html

@@ -19,14 +19,19 @@
     </ul>
 
     <div class="d-flex">
+
+        {% if perms.webapp.change_article or article.author == user %}
             <a href="{% url 'articles_update' id=article.id %}" class="btn btn-primary me-3">
                 Update
             </a>
+        {% endif %}
             
+        {% if perms.webapp.delete_article or article.author == user %}
             <form action="{% url 'articles_delete' id=article.id %}" method="post" onsubmit="return confirm('Delete?')">
                 {% csrf_token %}
                 <button class="btn btn-danger">Delete</button>
             </form>
+        {% endif %}
     </div>
 
     <hr>

ap_12/templates/base.html

@@ -19,9 +19,11 @@
             <div class="collapse navbar-collapse flex-grow-0" id="navbarNav">
                 <ul class="navbar-nav">
                 
+                {% if perms.webapp.add_article %}
                     <li class="nav-item">
                         <a class="nav-link" aria-current="page" href="{% url 'articles_add' %}">Create</a>
                     </li>
+                {% endif %}
 
                 <form class="d-flex">
                     <input class="form-control me-2" name="search" type="search" placeholder="Search" aria-label="Search">

ap_12/webapp/admin.py

@@ -3,6 +3,7 @@ from webapp import models
 
 
 admin.site.register(models.Comment)
+admin.site.register(models.Tag)
 
 
 @admin.register(models.Article)

ap_12/webapp/forms.py

@@ -7,7 +7,7 @@ from webapp import models
 class ArticleForm(forms.ModelForm):
     class Meta:
         model = models.Article
-        fields = ('title', 'text', 'author', 'tags')
+        fields = ('title', 'text', 'tags')
         
         widgets = {
             'title': widgets.TextInput(attrs={
@@ -18,10 +18,6 @@ class ArticleForm(forms.ModelForm):
                 'class': 'form-control mb-3',
                 'placeholder': 'Write some text',
             }),
-            'author': widgets.TextInput(attrs={
-                'class': 'form-control mb-3',
-                'placeholder': 'Author',
-            }),
             'tags': widgets.CheckboxSelectMultiple({
                 'class': 'list-unstyled ps-0 mb-3',
             })
@@ -41,15 +37,11 @@ class SearchForm(forms.Form):
 class CommentForm(forms.ModelForm):
     class Meta:
         model = models.Comment
-        fields = ('text', 'author')
+        fields = ('text',)
         
         widgets = {
             'text': widgets.Textarea(attrs={
                 'class': 'form-control mb-3',
                 'placeholder': 'Write some text',
             }),
-            'author': widgets.TextInput(attrs={
-                'class': 'form-control mb-3',
-                'placeholder': 'Author'
-            })
         }

ap_12/webapp/migrations/0010_alter_article_author.py

+# Generated by Django 3.2.19 on 2024-02-26 14:35
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('webapp', '0009_alter_article_title'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='article',
+            name='author',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='articles', to=settings.AUTH_USER_MODEL, verbose_name='Автор'),
+        ),
+    ]

ap_12/webapp/migrations/0011_alter_article_options.py

+# Generated by Django 3.2.19 on 2024-02-26 14:51
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('webapp', '0010_alter_article_author'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='article',
+            options={'permissions': [('can_eat_pizza', 'Может съесть пиццу')], 'verbose_name': 'статья', 'verbose_name_plural': 'статьи'},
+        ),
+    ]

ap_12/webapp/models.py

 from django.db import models
 from django.core.validators import MinLengthValidator
-
+from django.contrib.auth import get_user_model
 
 class Article(models.Model):
     title = models.CharField(
@@ -11,9 +11,9 @@ class Article(models.Model):
     text = models.TextField(
         null=False, blank=False, verbose_name='Текст',
     )
-    author = models.CharField(
-        max_length=40, null=False, blank=False, default='Неизвестный автор', 
-        verbose_name='Автор',
+    author = models.ForeignKey(
+        get_user_model(), on_delete=models.CASCADE, 
+        related_name='articles', verbose_name='Автор'
     )
     created_at = models.DateTimeField(auto_now_add=True, verbose_name='Дата создания')
     updated_at = models.DateTimeField(auto_now=True, verbose_name='Дата обновления')
@@ -24,6 +24,9 @@ class Article(models.Model):
     class Meta:
         verbose_name = 'статья'
         verbose_name_plural = 'статьи'
+        permissions = [
+            ('can_eat_pizza', 'Может съесть пиццу'),
+        ]
 
     def __str__(self):
         return f'{self.id} {self.title}'

ap_12/webapp/views/articles.py

@@ -4,16 +4,18 @@ from django.http.response import HttpResponse as HttpResponse
 from django.shortcuts import get_object_or_404, redirect, render
 from django.urls import reverse, reverse_lazy
 from django.views import generic
-from django.contrib.auth.mixins import LoginRequiredMixin
+from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin, UserPassesTestMixin
+from django.core.exceptions import PermissionDenied
 
 from webapp.forms import ArticleForm, CommentForm
 from webapp.models import Article
 
 
-class ArticleDetailView(generic.DetailView):
+class ArticleDetailView(PermissionRequiredMixin, generic.DetailView):
     pk_url_kwarg = 'id'
     model = Article
     template_name = 'articles/detail.html'
+    permission_required = 'webapp.can_eat_pizza'
     
     def get_context_data(self, **kwargs):
         comments = self.object.comments.order_by('-created_at')
@@ -24,71 +26,32 @@ class ArticleDetailView(generic.DetailView):
         )
 
 
-class ArticleCreateView(LoginRequiredMixin, generic.CreateView):
+class ArticleCreateView(PermissionRequiredMixin, generic.CreateView):
     model = Article
     template_name = 'articles/article.html'
     form_class = ArticleForm
-
-    def get_success_url(self):
-        return reverse('articles_detail', kwargs={'id': self.object.id})
-
-
-class UpdateView(generic.View):
-    template_name = None
-    model = None
-    form_class = None
-    context_object_name = None
-    pk_url_kwarg = None
-    success_url = None
-    
-    def dispatch(self, request, *args, **kwargs):
-        self.object = self.get_object()
-        return super().dispatch(request, *args, **kwargs)
-
-    def get(self, request, *args, **kwargs):
-        form = self.form_class(instance=self.object)
-        return self.render_to_response(form=form)
-    
-    def post(self, request, *args, **kwargs):
-        form = self.form_class(instance=self.object, data=request.POST)
-        
-        if form.is_valid():
-            return self.form_valid(form)
-
-        return self.form_invalid(form)
-
-    def get_object(self):
-        return get_object_or_404(self.model, id=self.kwargs.get(self.pk_url_kwarg))
-    
-    def get_context_data(self, **kwargs):
-        kwargs[self.context_object_name] = self.object
-        return kwargs
-
-    def get_success_url(self):
-        return self.success_url
+    permission_required = 'webapp.add_article'
+    permission_denied_message = 'У вас нет прав'
     
     def form_valid(self, form):
-        self.object = form.save()
-        return redirect(self.get_success_url())
+        form.instance.author = self.request.user
+        return super().form_valid(form)
 
-    def form_invalid(self, form):
-        return self.render_to_response(form=form)
-
-    def render_to_response(self, **kwargs):
-        return render(
-            self.request, 
-            self.template_name, 
-            context=self.get_context_data(**kwargs),
-        )
+    def get_success_url(self):
+        return reverse('articles_detail', kwargs={'id': self.object.id})
 
 
-class ArticleUpdateView(generic.UpdateView):
+class ArticleUpdateView(PermissionRequiredMixin, generic.UpdateView):
     template_name = 'articles/article.html'
     model = Article
     form_class = ArticleForm
     context_object_name = 'article'
     pk_url_kwarg = 'id'
     extra_context = {'btn_text': 'Update'}
+    permission_required = 'webapp.change_article'
+
+    def has_permission(self) -> bool:
+        return super().has_permission() or self.get_object().author == self.request.user
 
     def get_success_url(self):
         return reverse('articles_detail', kwargs={'id': self.object.id})

ap_12/webapp/views/comments.py

 from django.shortcuts import redirect
 from django.views import generic
 from django.urls import reverse
+from django.contrib.auth.mixins import LoginRequiredMixin
 
 from webapp.models import Comment
 from webapp.forms import CommentForm
-from webapp.views.articles import UpdateView, DeleteView
 
 
-class CommentCreateView(generic.CreateView):
+class CommentCreateView(LoginRequiredMixin, generic.CreateView):
     model = Comment
     template_name = 'comments/comment.html'
     form_class = CommentForm
@@ -18,6 +18,7 @@ class CommentCreateView(generic.CreateView):
 
         comment = form.save(commit=False)
         comment.article_id = article_id
+        comment.author = self.request.user
         comment.save()
 
         return redirect('articles_detail', id=article_id)