oacpl/newsletters/admin.py

38 lines
1.2 KiB
Python

from django.contrib import admin
from django.utils import timezone
from OACPL import settings
from .models import Newsletter, Subscriber
@admin.register(Subscriber)
class SubscriberAdmin(admin.ModelAdmin):
list_display = ['email', 'date']
list_filter = ['date']
search_fields = ['email']
@admin.register(Newsletter)
class NewsletterAdmin(admin.ModelAdmin):
list_display = ['subject', 'created', 'publish']
search_fields = ['subject', 'created', 'publish']
def get_form(self, request, obj=None, **kwargs):
if obj:
self.fields = ['creator', 'subject', 'body', 'sent', 'publish']
if obj.sent:
self.readonly_fields = ['creator', 'subject', 'body', 'sent', 'publish']
else:
self.readonly_fields = ['creator', 'sent']
else:
self.fields = ['subject', 'body', 'publish']
self.readonly_fields = []
return super(NewsletterAdmin, self).get_form(request, obj, **kwargs)
def save_model(self, request, instance, form, change):
user = request.user
instance = form.save(commit=False)
instance.creator = user
instance.save()
return instance