27 lines
944 B
Python
27 lines
944 B
Python
from django.contrib import admin
|
|
from .models import TelegramData
|
|
from .utils import telegramUtils
|
|
from core.admin import UniversalModelAdmin
|
|
|
|
@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(UniversalModelAdmin):
|
|
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
|
|
|
|
|
|
|