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 =queryset.values('id')
    list_id = list(queryset.values_list('id', flat=True))
    send_email_with_news.delay(list_id)


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]