from django.contrib import admin
from django.conf.urls import url

from .models import Email, Tag
from .views import SendEmailAdminView, UpdateTargetCountView
from .forms import EmailAdminForm


class EmailAdmin(admin.ModelAdmin):
    list_display = ['__str__', 'is_sent']
    list_filter = ['is_sent']
    form = EmailAdminForm
    readonly_fields = ('targetted_users', 'is_sent',)
    add_fieldsets = (
        (None, {
            'fields': ('subject', 'message'),
        }),
    )

    def get_fieldsets(self, request, obj=None):
        if not obj:
            return self.add_fieldsets
        return super(EmailAdmin, self).get_fieldsets(request, obj)

    def get_urls(self):
        urls = super(EmailAdmin, self).get_urls()
        my_urls = [
            url(r'^(?P<pk>\d+)/send_email/$',
                self.admin_site.admin_view(SendEmailAdminView.as_view()),
                name='send_email'),
            url(r'^(?P<pk>\d+)/update_target_count/$',
                self.admin_site.admin_view(UpdateTargetCountView.as_view()),
                name='update_target_count'),
        ]
        return my_urls + urls

admin.site.register(Email, EmailAdmin)
admin.site.register(Tag)
