from django.contrib import admin
from import_export.admin import ImportExportModelAdmin
from .models import TelegramData
from .utils import telegramUtils 

@admin.action(description="Clear all expired messages")
def messages_cleaning(modeladmin, request, queryset):
    telegram = telegramUtils()
    telegram.clear_expired_messages()

# Register your models here.
@admin.register(TelegramData)
class JobAdmin(ImportExportModelAdmin):
    list_display = ('message_id','chat_id','description','creation_date','expiration_date')
    actions = [messages_cleaning]

    def delete_queryset(self, request, queryset):
        """Given a queryset, delete it from the database."""
        print("here")
        telegram = telegramUtils()
        list_of_message_id = list(queryset.all().values_list('message_id',flat=True))
        telegram.delete_message(list_of_message_id)
        #no queryset.delete() because it is specified in delete_message function already