61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
from django.contrib import admin, messages
|
|
from django.contrib.admin import register
|
|
from django.contrib.auth.admin import UserAdmin
|
|
|
|
from .models import Article, User
|
|
|
|
admin.site.register(User, UserAdmin)
|
|
|
|
|
|
@register(Article)
|
|
class ArticleAdmin(admin.ModelAdmin):
|
|
list_display = [
|
|
"title",
|
|
"status",
|
|
"author",
|
|
"created_at",
|
|
"published_at",
|
|
"updated_at",
|
|
]
|
|
list_display_links = ["title"]
|
|
list_filter = ["status"]
|
|
date_hierarchy = "created_at"
|
|
fieldsets = [
|
|
(
|
|
"Metadata",
|
|
{
|
|
"fields": (
|
|
"title",
|
|
("author", "status"),
|
|
("published_at", "created_at", "updated_at"),
|
|
"views_count",
|
|
)
|
|
},
|
|
),
|
|
("Content", {"fields": ("content",)}),
|
|
]
|
|
readonly_fields = ["created_at", "updated_at", "views_count"]
|
|
|
|
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(save=False)
|
|
Article.objects.bulk_update(queryset, ["published_at", "status"])
|
|
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(save=False)
|
|
Article.objects.bulk_update(queryset, ["published_at", "status"])
|
|
messages.success(request, f"{len(queryset)} articles unpublished.")
|
|
|
|
unpublish.short_description = "Unpublish selected articles"
|
|
actions = [publish, unpublish]
|