64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
"""
|
|
Django Admin for Notifications
|
|
"""
|
|
|
|
from django.contrib import admin
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from .base_notification import COURSE_NOTIFICATION_APPS, COURSE_NOTIFICATION_TYPES
|
|
from .models import Notification
|
|
|
|
|
|
class NotificationAppNameListFilter(admin.SimpleListFilter):
|
|
"""
|
|
Shows list filter in django admin of notification apps
|
|
"""
|
|
title = _("Notification App")
|
|
parameter_name = "app_name"
|
|
|
|
def lookups(self, request, model_admin):
|
|
lookup_list = [
|
|
(app_name, app_name)
|
|
for app_name in COURSE_NOTIFICATION_APPS
|
|
]
|
|
return lookup_list
|
|
|
|
def queryset(self, request, queryset):
|
|
app_name = self.value()
|
|
if app_name not in COURSE_NOTIFICATION_APPS:
|
|
return queryset
|
|
return queryset.filter(app_name=app_name)
|
|
|
|
|
|
class NotificationTypeListFilter(admin.SimpleListFilter):
|
|
"""
|
|
Shows list filter in django admin of notification types
|
|
"""
|
|
title = _("Notification Type")
|
|
parameter_name = "notification_type"
|
|
|
|
def lookups(self, request, model_admin):
|
|
lookup_list = [
|
|
(notification_type, notification_type)
|
|
for notification_type in COURSE_NOTIFICATION_TYPES
|
|
]
|
|
return lookup_list
|
|
|
|
def queryset(self, request, queryset):
|
|
notification_type = self.value()
|
|
if notification_type not in COURSE_NOTIFICATION_TYPES:
|
|
return queryset
|
|
return queryset.filter(notification_type=notification_type)
|
|
|
|
|
|
class NotificationAdmin(admin.ModelAdmin):
|
|
"""
|
|
Admin for Notifications
|
|
"""
|
|
raw_id_fields = ('user',)
|
|
search_fields = ('course_id', 'app_name', 'notification_type', 'user__username')
|
|
list_filter = (NotificationAppNameListFilter, NotificationTypeListFilter)
|
|
|
|
|
|
admin.site.register(Notification, NotificationAdmin)
|