from django.contrib import admin, messages

from .models import Edicto, Precio

# Register your models here.


class PrecioAdmin(admin.ModelAdmin):
    def save_model(self, request, obj, form, change):

        if change:
            super().save_model(request, obj, form, change)
        else:

            precios_con_vigencia = Precio.objects.filter(vigencia_hasta__isnull=True)

            if precios_con_vigencia:
                messages.error(request, 'Error al agregar: Ya existe un objeto Precio sin vigencia_hasta. '
                            'Debe proporcionar una fecha para vigencia_hasta antes de crear un nuevo objeto Precio.')
            else:
                messages.success(request, 'El nuevo precio se agrego correctamente')
                obj.usuario = request.user
                super().save_model(request, obj, form, change)


admin.site.register(Precio, PrecioAdmin)

admin.site.register(Edicto)