forked from rc/aircox
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
import urllib
|
|
|
|
from django.contrib import admin
|
|
from django.http import QueryDict
|
|
from django.utils.safestring import mark_safe
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from adminsortable2.admin import SortableInlineAdminMixin
|
|
|
|
from ..models import Category, NavItem, Page
|
|
|
|
|
|
__all__ = ['CategoryAdmin', 'PageAdmin', 'NavItemInline']
|
|
|
|
|
|
@admin.register(Category)
|
|
class CategoryAdmin(admin.ModelAdmin):
|
|
list_display = ['pk', 'title', 'slug']
|
|
list_editable = ['title', 'slug']
|
|
search_fields = ['title']
|
|
fields = ['title', 'slug']
|
|
prepopulated_fields = {"slug": ("title",)}
|
|
|
|
|
|
# limit category choice
|
|
class PageAdmin(admin.ModelAdmin):
|
|
list_display = ('cover_thumb', 'title', 'status', 'category', 'parent')
|
|
list_display_links = ('cover_thumb', 'title')
|
|
list_editable = ('status', 'category')
|
|
list_filter = ('status', 'category')
|
|
prepopulated_fields = {"slug": ("title",)}
|
|
|
|
# prepopulate fields using changelist's filters
|
|
prepopulated_filters = ('parent',)
|
|
|
|
search_fields = ['title', 'category__title']
|
|
fieldsets = [
|
|
('', {
|
|
'fields': ['title', 'slug', 'category', 'cover', 'content'],
|
|
}),
|
|
(_('Publication Settings'), {
|
|
'fields': ['featured', 'allow_comments', 'status', 'parent'],
|
|
'classes': ('collapse',),
|
|
}),
|
|
]
|
|
|
|
change_form_template = 'admin/aircox/page_change_form.html'
|
|
|
|
def cover_thumb(self, obj):
|
|
return mark_safe('<img src="{}"/>'.format(obj.cover.icons['64'])) \
|
|
if obj.cover else ''
|
|
|
|
def get_changeform_initial_data(self, request):
|
|
data = super().get_changeform_initial_data(request)
|
|
filters = QueryDict(request.GET.get('_changelist_filters', ''))
|
|
data['parent'] = filters.get('parent', None)
|
|
return data
|
|
|
|
def get_common_context(self, query, extra_context=None):
|
|
extra_context = extra_context or {}
|
|
parent = query.get('parent', None)
|
|
if parent is not None:
|
|
extra_context['parent'] = Page.objects.get_subclass(id=parent)
|
|
return extra_context
|
|
|
|
def add_view(self, request, form_url='', extra_context=None):
|
|
filters = QueryDict(request.GET.get('_changelist_filters', ''))
|
|
extra_context = self.get_common_context(filters, extra_context)
|
|
return super().add_view(request, form_url, extra_context)
|
|
|
|
def changelist_view(self, request, extra_context=None):
|
|
extra_context = self.get_common_context(request.GET, extra_context)
|
|
return super().changelist_view(request, extra_context)
|
|
|
|
|
|
class NavItemInline(SortableInlineAdminMixin, admin.TabularInline):
|
|
model = NavItem
|
|
|
|
|
|
|