from django.contrib import admin

from . import models


@admin.register(models.NewsItem)
class NewsItemModelAdmin(admin.ModelAdmin):
    list_display = ['title', 'published_at', 'archived']
    actions = ['archive_news_items', 'unarchive_news_items']

    def archive_news_items(self, request, queryset):
        queryset.filter(archived=False).update(archived=True)
    archive_news_items.description = 'Mark newsitem(s) as archived'

    def unarchive_news_items(self, request, queryset):
        queryset.filter(archived=True).update(archived=False)
    unarchive_news_items.description = 'Mark newsitem(s) as not archived'