from django.contrib import admin, messages from django.contrib.admin import register from django.contrib.auth.admin import UserAdmin from django.shortcuts import redirect from .models import Article, Comment, Page, User admin.site.register(User, UserAdmin) @register(Article) class ArticleAdmin(admin.ModelAdmin): ordering = ["status", "-published_at"] list_display = [ "title", "status", "author", "created_at", "published_at", "updated_at", "views_count", ] list_display_links = ["title"] list_filter = ["status"] date_hierarchy = "created_at" fieldsets = [ ( "Metadata", { "fields": ( ("title", "slug"), ("author", "status"), ("published_at", "created_at", "updated_at"), "views_count", ) }, ), ("Content", {"fields": ("content",)}), ] readonly_fields = [ "created_at", "updated_at", "views_count", "status", "published_at", ] prepopulated_fields = {"slug": ("title",)} change_form_template = "articles/article_change_form.html" def publish(self, request, queryset): if not request.user.has_perm("articles.change_article"): messages.warning(request, "You're not allowed to do this.") return for article in queryset: article.publish() messages.success(request, f"{len(queryset)} articles published.") publish.short_description = "Publish selected articles" def unpublish(self, request, queryset): if not request.user.has_perm("articles.change_article"): messages.warning(request, "You're not allowed to do this.") return for article in queryset: article.unpublish() messages.success(request, f"{len(queryset)} articles unpublished.") unpublish.short_description = "Unpublish selected articles" actions = [publish, unpublish] class Media: css = {"all": ("admin_articles.css",)} def response_post_save_add(self, request, obj: Article): if "_preview" in request.POST: return redirect("article-detail", slug=obj.slug) return super().response_post_save_add(request, obj) def response_change(self, request, obj: Article): if "_preview" in request.POST: obj.save() return redirect("article-detail", slug=obj.slug) if "_publish" in request.POST: obj.publish() messages.success(request, "Item has been published") return redirect(".") if "_unpublish" in request.POST: obj.unpublish() messages.success(request, "Item has been unpublished") return redirect(".") return super().response_change(request, obj) @register(Page) class PageAdmin(ArticleAdmin): list_display = ["position"] + ArticleAdmin.list_display fieldsets = [ ( "Metadata", { "fields": ( ("title", "slug", "position"), ("author", "status"), ("published_at", "created_at", "updated_at"), "views_count", ) }, ), ("Content", {"fields": ("content",)}), ] @register(Comment) class CommentAdmin(admin.ModelAdmin): list_display = ( "username", "email", "content", "article", "created_at", "status", "user_notified", ) list_filter = ("status",) search_fields = ("username", "email", "content") actions = ["approve_comments", "reject_comments"] def approve_comments(self, request, queryset): count = queryset.update(status=Comment.APPROVED, user_notified=False) messages.success(request, f"Approved {count} message(s).") def reject_comments(self, request, queryset): count = queryset.update(status=Comment.REJECTED, user_notified=False) messages.success(request, f"Rejected {count} message(s).")