from django.contrib import admin

from news import models
from .tasks import send_email_with_news


@admin.register(models.NewsType)
class NewsTypeAdmin(admin.ModelAdmin):
    """News type admin."""
    list_display = ['id', 'name']
    list_display_links = ['id', 'name']


def send_email_action(modeladmin, request, queryset):
    news_ids = [n.id for n in queryset]

    send_email_with_news.delay(news_ids)

    return


send_email_action.short_description = "Send the selected news by email"


@admin.register(models.News)
class NewsAdmin(admin.ModelAdmin):
    """News admin."""
    actions = [send_email_action]