DjangoBlog/src/blog/article_views.py

from django.contrib.auth.decorators import login_required
from django.contrib.auth.mixins import LoginRequiredMixin
from django.shortcuts import redirect, get_object_or_404
from django.views.generic.edit import CreateView, UpdateView
from django.urls import reverse_lazy
from .models import Article, Category, Tag
from .forms import ArticleForm


class ArticleCreateView(LoginRequiredMixin, CreateView):
    """创建文章视图"""
    model = Article
    form_class = ArticleForm
    template_name = 'blog/article_create.html'
    
    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['all_categories'] = Category.objects.all()
        context['all_tags'] = Tag.objects.all()
        return context

    def form_valid(self, form):
        # 先不保存表单，先处理分类
        form.instance.author = self.request.user
        
        # 处理分类
        category_name = form.cleaned_data['category']
        category, created = Category.objects.get_or_create(
            name=category_name,
            defaults={'parent_category': None}
        )
        # 直接设置分类对象，而不是名称
        form.instance.category = category
        
        # 先保存文章实例，以便可以添加多对多关系
        # 但在保存前，先从表单中移除category字段，避免表单尝试保存它
        category_temp = form.cleaned_data.pop('category')
        tags_temp = form.cleaned_data.pop('tags')
        
        response = super().form_valid(form)
        
        # 处理标签
        if tags_temp:
            tag_names = [tag.strip() for tag in tags_temp.split(',') if tag.strip()]
            for tag_name in tag_names:
                tag, created = Tag.objects.get_or_create(name=tag_name)
                form.instance.tags.add(tag)
        
        return response

    def get_success_url(self):
        return reverse_lazy('blog:detailbyid', kwargs={
            'article_id': self.object.id,
            'year': self.object.creation_time.year,
            'month': self.object.creation_time.month,
            'day': self.object.creation_time.day
        })


class ArticleUpdateView(LoginRequiredMixin, UpdateView):
    """更新文章视图"""
    model = Article
    form_class = ArticleForm
    template_name = 'blog/article_edit.html'
    pk_url_kwarg = 'article_id'
    
    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['all_categories'] = Category.objects.all()
        context['all_tags'] = Tag.objects.all()
        return context
    
    def get_initial(self):
        initial = super().get_initial()
        # 设置分类初始值
        if self.object.category:
            initial['category'] = self.object.category.name
        
        # 设置标签初始值
        if self.object.tags.exists():
            tag_names = [tag.name for tag in self.object.tags.all()]
            initial['tags'] = ', '.join(tag_names)
        
        return initial

    def dispatch(self, request, *args, **kwargs):
        obj = self.get_object()
        # 只有文章作者或管理员可以编辑
        if obj.author != request.user and not request.user.is_superuser:
            return redirect('blog:detailbyid',
                           article_id=obj.id,
                           year=obj.creation_time.year,
                           month=obj.creation_time.month,
                           day=obj.creation_time.day)
        return super().dispatch(request, *args, **kwargs)

    def form_valid(self, form):
        # 先不保存表单，先处理分类
        form.instance.author = self.request.user
        
        # 处理分类
        category_name = form.cleaned_data['category']
        category, created = Category.objects.get_or_create(
            name=category_name,
            defaults={'parent_category': None}
        )
        # 直接设置分类对象，而不是名称
        form.instance.category = category
        
        # 先保存文章实例，以便可以添加多对多关系
        # 但在保存前，先从表单中移除category字段，避免表单尝试保存它
        category_temp = form.cleaned_data.pop('category')
        tags_temp = form.cleaned_data.pop('tags')
        
        response = super().form_valid(form)
        
        # 处理标签
        if tags_temp:
            tag_names = [tag.strip() for tag in tags_temp.split(',') if tag.strip()]
            form.instance.tags.clear()  # 清除现有标签
            for tag_name in tag_names:
                tag, created = Tag.objects.get_or_create(name=tag_name)
                form.instance.tags.add(tag)
        else:
            form.instance.tags.clear()  # 如果没有标签，清除所有标签
        
        return response

    def get_success_url(self):
        return reverse_lazy('blog:detailbyid', kwargs={
            'article_id': self.object.id,
            'year': self.object.creation_time.year,
            'month': self.object.creation_time.month,
            'day': self.object.creation_time.day
        })