You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
from django.contrib import admin, messages
|
|
from django.contrib.admin import helpers
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.shortcuts import render
|
|
from django.db import models
|
|
from django import forms
|
|
|
|
from .models import Message, Attachment, MessageForm
|
|
|
|
|
|
class AttachmentInline(admin.StackedInline):
|
|
model = Attachment
|
|
extra = 0
|
|
|
|
|
|
class MessageAdmin(admin.ModelAdmin):
|
|
"""Message creation view"""
|
|
list_display = ('subject', 'from_addr', 'get_recipients', 'sent')
|
|
change_form_template = "mailer/change_form.html"
|
|
formfield_overrides = {
|
|
models.ManyToManyField: {'widget': forms.CheckboxSelectMultiple}
|
|
}
|
|
|
|
inlines = [AttachmentInline]
|
|
actions = ['send_message']
|
|
form = MessageForm
|
|
|
|
def send_message(self, request, queryset):
|
|
print("calling send_message")
|
|
if request.POST.get('confirmed'):
|
|
for msg in queryset:
|
|
msg.submit()
|
|
self.message_user(request, _("Message sent"))
|
|
else:
|
|
context = {
|
|
'action_checkbox_name': helpers.ACTION_CHECKBOX_NAME,
|
|
'mails': queryset,
|
|
'ids': queryset.values_list("id"),
|
|
'some_sent': any(m.sent for m in queryset)}
|
|
return render(request, 'mailer/confirm_send.html', context)
|
|
send_message.short_description = _("Send message")
|
|
|
|
def response_change(self, request, obj):
|
|
if "_send" in request.POST:
|
|
if not obj.submit():
|
|
messages.error(request, _("Failed to send message"))
|
|
else:
|
|
messages.info(request, _("Successfully sent message"))
|
|
return super(MessageAdmin, self).response_change(request, obj)
|
|
|
|
def response_add(self, request, obj):
|
|
if "_send" in request.POST:
|
|
if not obj.submit():
|
|
messages.error(request, _("Failed to send message"))
|
|
else:
|
|
messages.info(request, _("Successfully sent message"))
|
|
return super(MessageAdmin, self).response_add(request, obj)
|
|
|
|
|
|
admin.site.register(Message, MessageAdmin)
|