98 lines
2.7 KiB
Python
98 lines
2.7 KiB
Python
from django.contrib import admin
|
|
from django.db.models import Prefetch
|
|
from django.template.defaultfilters import truncatechars
|
|
|
|
from extensions.models import Version
|
|
|
|
from .models import Rating
|
|
|
|
|
|
class RatingTypeFilter(admin.SimpleListFilter):
|
|
# Human-readable title which will be displayed in the
|
|
# right admin sidebar just above the filter options.
|
|
title = 'Type'
|
|
|
|
# Parameter for the filter that will be used in the URL query.
|
|
parameter_name = 'type'
|
|
|
|
def lookups(self, request, model_admin):
|
|
"""Return a list of lookup option tuples.
|
|
|
|
The first element in each tuple is the coded value
|
|
for the option that will appear in the URL query.
|
|
The second element is the human-readable name for
|
|
the option that will appear in the right sidebar.
|
|
"""
|
|
return (
|
|
('rating', 'User Rating'),
|
|
('reply', 'Developer/Admin Reply'),
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
"""Return the filtered queryset.
|
|
|
|
Filter based on the value provided in the query string
|
|
and retrievable via `self.value()`.
|
|
"""
|
|
if self.value() == 'rating':
|
|
return queryset.filter(reply_to__isnull=True)
|
|
elif self.value() == 'reply':
|
|
return queryset.filter(reply_to__isnull=False)
|
|
return queryset
|
|
|
|
|
|
class RatingAdmin(admin.ModelAdmin):
|
|
date_hierarchy = 'date_created'
|
|
search_fields = ('extension__slug', 'extension__name', 'user__email', 'user__full_name')
|
|
raw_id_fields = (
|
|
'extension',
|
|
'version',
|
|
'user',
|
|
'reply_to',
|
|
)
|
|
readonly_fields = (
|
|
'date_created',
|
|
'date_modified',
|
|
'extension',
|
|
'version',
|
|
'text',
|
|
'score',
|
|
'user',
|
|
)
|
|
fields = ('status',) + readonly_fields
|
|
list_display = (
|
|
'date_created',
|
|
'extension',
|
|
'user',
|
|
'ip_address',
|
|
'score',
|
|
'is_reply',
|
|
'status',
|
|
'truncated_text',
|
|
)
|
|
list_filter = ('status', RatingTypeFilter, 'score')
|
|
actions = ('delete_selected',)
|
|
list_select_related = ('user',) # For extension/reply_to see get_queryset()
|
|
|
|
def get_queryset(self, request):
|
|
base_qs = Rating.objects.all()
|
|
return base_qs.prefetch_related(
|
|
Prefetch('version', queryset=Version.objects.all()),
|
|
Prefetch('reply_to', queryset=base_qs),
|
|
)
|
|
|
|
def has_add_permission(self, request):
|
|
return False
|
|
|
|
def truncated_text(self, obj):
|
|
return truncatechars(obj.text, 140) if obj.text else ''
|
|
|
|
def is_reply(self, obj):
|
|
return bool(obj.reply_to)
|
|
|
|
is_reply.boolean = True
|
|
is_reply.admin_order_field = 'reply_to'
|
|
|
|
|
|
admin.site.register(Rating, RatingAdmin)
|