152 lines
4.9 KiB
Python
152 lines
4.9 KiB
Python
from typing import cast
|
|
|
|
from django.contrib import admin, messages
|
|
from django.contrib.admin import register
|
|
from django.contrib.auth.admin import UserAdmin
|
|
from django.db.models import QuerySet
|
|
from django.http import HttpRequest, HttpResponse, HttpResponseRedirect
|
|
from django.shortcuts import redirect
|
|
|
|
from .models import Article, Tag, User
|
|
|
|
admin.site.register(User, UserAdmin)
|
|
|
|
|
|
@register(Article)
|
|
class ArticleAdmin(admin.ModelAdmin):
|
|
ordering = ["status", "-published_at"]
|
|
list_display = [
|
|
"title",
|
|
"status",
|
|
"keywords",
|
|
"created_at",
|
|
"published_at",
|
|
"updated_at",
|
|
"views_count",
|
|
"has_code",
|
|
"is_home",
|
|
"has_custom_css",
|
|
"read_time",
|
|
]
|
|
list_display_links = ["title"]
|
|
list_filter = [
|
|
"status",
|
|
"has_code",
|
|
"status",
|
|
"is_home",
|
|
]
|
|
date_hierarchy = "created_at"
|
|
fieldsets = [
|
|
(
|
|
"Metadata",
|
|
{
|
|
"fields": [
|
|
("title", "slug"),
|
|
("author", "tags"),
|
|
("status", "published_at"),
|
|
("created_at", "updated_at"),
|
|
("views_count", "read_time"),
|
|
("has_code", "has_custom_css"),
|
|
]
|
|
},
|
|
),
|
|
(
|
|
"Content",
|
|
{"fields": ("content",)},
|
|
),
|
|
(
|
|
"Custom CSS",
|
|
{
|
|
"fields": ("custom_css",),
|
|
"classes": ("collapse",),
|
|
},
|
|
),
|
|
]
|
|
readonly_fields = [
|
|
"created_at",
|
|
"updated_at",
|
|
"views_count",
|
|
"status",
|
|
"published_at",
|
|
"read_time",
|
|
"has_custom_css",
|
|
]
|
|
prepopulated_fields = {"slug": ("title",)}
|
|
change_form_template = "articles/article_change_form.html"
|
|
search_fields = ["title", "content", "tags__name"]
|
|
autocomplete_fields = ["tags"]
|
|
show_full_result_count = False
|
|
|
|
def get_queryset(self, request: HttpRequest) -> QuerySet:
|
|
queryset = super().get_queryset(request)
|
|
queryset = queryset.prefetch_related("tags")
|
|
return queryset
|
|
|
|
@admin.action(description="Publish selected articles")
|
|
def publish(self, request: HttpRequest, queryset: QuerySet) -> None:
|
|
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.")
|
|
|
|
@admin.action(description="Unpublish selected articles")
|
|
def unpublish(self, request: HttpRequest, queryset: QuerySet) -> None:
|
|
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.")
|
|
|
|
@admin.action(description="Refresh draft key of selected articles")
|
|
def refresh_draft_key(self, request: HttpRequest, queryset: QuerySet) -> None:
|
|
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.refresh_draft_key()
|
|
messages.success(request, f"{len(queryset)} draft keys refreshed.")
|
|
|
|
actions = [publish, unpublish, refresh_draft_key]
|
|
|
|
class Media:
|
|
css = {"all": ("admin_articles.css",)}
|
|
|
|
def response_post_save_add(
|
|
self, request: HttpRequest, obj: Article
|
|
) -> HttpResponseRedirect:
|
|
if "_preview" in request.POST:
|
|
return cast(HttpResponseRedirect, redirect("article-detail", slug=obj.slug))
|
|
return super().response_post_save_add(request, obj)
|
|
|
|
def response_change(self, request: HttpRequest, obj: Article) -> HttpResponse:
|
|
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)
|
|
|
|
def read_time(self, instance: Article) -> str:
|
|
return f"{instance.get_read_time()} min"
|
|
|
|
@admin.display(boolean=True)
|
|
def has_custom_css(self, instance: Article) -> bool:
|
|
return bool(instance.custom_css)
|
|
|
|
|
|
@register(Tag)
|
|
class TagAdmin(admin.ModelAdmin):
|
|
list_display = ["name", "slug"]
|
|
search_fields = ["slug", "name"]
|
|
list_display_links = ["slug", "name"]
|
|
prepopulated_fields = {"slug": ("name",)}
|