from dati_geo_app.admin import (AjaxAutocompleteListFilterModelAdmin,
                                RicercaOrdinataMixin)
from django.contrib import admin
from django.core.cache import cache
from django.core.paginator import Paginator
from django.db.models import Case, Count, Value, When
from django.db.models.functions import Concat
from django.db.models import F
from import_export.admin import ImportExportModelAdmin
from polymorphic.admin import (PolymorphicChildModelAdmin,
                               PolymorphicChildModelFilter,
                               PolymorphicInlineSupportMixin,
                               PolymorphicParentModelAdmin,
                               StackedPolymorphicInline)

from . import models, resources

# Modified version of a GIST I found in a SO thread
# cfr. http://masnun.rocks/2017/03/20/django-admin-expensive-count-all-queries/


class AutocompleteAdmin(admin.ModelAdmin):
    class Media:
        js = ('admin/js/autocomplete_auto_focus.js', )


class StackedInlineCollassati(AutocompleteAdmin):
    class Media:
        js = ('admin/js/stacked_inline_collassati.js',)
        css = { 'all': ('admin/css/stacked_inline_collassati.css',)}


class HiddenIfNotSuperuser(admin.ModelAdmin):
    def get_model_perms(self, request):
        if not request.user.is_superuser:
            return {}
        else:
            return super().get_model_perms(request)


class HiddenModel(admin.ModelAdmin):
    def get_model_perms(self, request):
        return {}


class CachingPaginator(Paginator):
    def _get_count(self):
        if not hasattr(self, "_count"):
            self._count = None
        if self._count is None:
            try:
                key = "adm:{0}:count".format(
                    hash(self.object_list.query.__str__()))
                self._count = cache.get(key, -1)
                if self._count == -1:
                    self._count = super().count
                    cache.set(key, self._count, 60)
            except:
                self._count = len(self.object_list)
        return self._count
    count = property(_get_count)

# # Main reusable Admin class for only viewing
# class ViewAdminMixin(admin.ModelAdmin):
#     def has_add_permission(self, request):
#         return False
#
#     def has_change_permission(self, request, obj=None):
#         return False
#
#     def has_delete_permission(self, request, obj=None):
#         return False

# --------------- FINE PREFISSO TEMPLATE ---------------


@admin.register(models.ContattoAziendale)
class ContattoAziendaleAdmin(ImportExportModelAdmin):
    # resource = resources.ContattoAziendaleResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    pass


@admin.register(models.VoceContattoRubrica)
class VoceContattoRubricaAdmin(ImportExportModelAdmin, PolymorphicParentModelAdmin):
    # resource = resources.VoceContattoRubricaResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    base_model = models.VoceContattoRubrica
    child_models = (models.Indirizzo, models.Email, models.Pec, models.Telefono, models.Fax,)
    list_filter = (PolymorphicChildModelFilter,)
    readonly_fields = ('descrizione','tipo',)
    list_display = ('soggetto','descrizione','tipo',)

    def descrizione(self,obj):
        return str(getattr(obj,obj.polymorphic_ctype.model))

    def tipo(self, obj):
        return getattr(obj,obj.polymorphic_ctype.model)._meta.verbose_name.title()
    tipo.admin_order_field = '_tipo'

    def get_queryset(self, request):
        qs=super().get_queryset(request).prefetch_related('polymorphic_ctype')
        qs=qs.annotate(_tipo=F('polymorphic_ctype__model'))
        return qs
    pass


@admin.register(models.Telefono)
class TelefonoAdmin(HiddenModel, PolymorphicChildModelAdmin):
    # resource = resources.TelefonoResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    fields = ('soggetto','numero','interno','note',)
    autocomplete_fields = ('soggetto',)
    pass


@admin.register(models.Email)
class EmailAdmin(HiddenModel, PolymorphicChildModelAdmin):
    # resource = resources.EmailResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    fields = ('soggetto','indirizzo_email','note',)
    autocomplete_fields = ('soggetto',)
    pass


class VoceContattoRubricaInline(StackedPolymorphicInline):
    class IndirizzoInline(StackedPolymorphicInline.Child):
        model = models.Indirizzo
        autocomplete_fields = ('comune','cap',)

    class EmailInline(StackedPolymorphicInline.Child):
        model = models.Email

    class PecInline(StackedPolymorphicInline.Child):
        model = models.Pec

    class TelefonoInline(StackedPolymorphicInline.Child):
        model = models.Telefono

    class FaxInline(StackedPolymorphicInline.Child):
        model = models.Fax

    model = models.VoceContattoRubrica
    child_inlines = (
        IndirizzoInline,
        EmailInline,
        PecInline,
        TelefonoInline,
        FaxInline,
    )


@admin.register(models.PersonaFisica)
class PersonaFisicaAdmin(PolymorphicInlineSupportMixin, PolymorphicChildModelAdmin, StackedInlineCollassati):
    # resource = resources.PersonaFisicaResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    show_in_index = False
    get_model_perms = lambda self, req: {}
    inlines = (VoceContattoRubricaInline,)
    pass


@admin.register(models.PersonaGiuridica)
class PersonaGiuridicaAdmin(PolymorphicInlineSupportMixin, PolymorphicChildModelAdmin, StackedInlineCollassati):
    # resource = resources.PersonaGiuridicaResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    show_in_index = False
    get_model_perms = lambda self, req: {}
    inlines = (VoceContattoRubricaInline,)
    pass


@admin.register(models.Sede)
class SedeAdmin(ImportExportModelAdmin):
    # resource = resources.SedeResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    pass


@admin.register(models.Fax)
class FaxAdmin(HiddenModel, PolymorphicChildModelAdmin):
    # resource = resources.FaxResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    fields = ('soggetto','numero','note',)
    autocomplete_fields = ('soggetto',)
    pass


@admin.register(models.Pec)
class PecAdmin(ImportExportModelAdmin):
    # resource = resources.PecResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    fields = ('soggetto','indirizzo_email','note',)
    autocomplete_fields = ('soggetto',)
    pass


@admin.register(models.SoggettoContattabile)
class SoggettoContattabileAdmin(PolymorphicParentModelAdmin):
    # resource = resources.SoggettoContattabileResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    base_model = models.SoggettoContattabile
    child_models = (models.PersonaFisica, models.PersonaGiuridica,)
    list_filter = (PolymorphicChildModelFilter,)
    readonly_fields = ('descrizione','tipo',)
    list_display = ('descrizione','tipo',)
    search_fields = ('personafisica__nome','personafisica__cognome','personagiuridica__denominazione',)

    def descrizione(self,obj):
        return str(getattr(obj,obj.polymorphic_ctype.model))

    def tipo(self, obj):
        return getattr(obj,obj.polymorphic_ctype.model)._meta.verbose_name.title()
    
    def get_queryset(self, request):
        qs=super().get_queryset(request).prefetch_related('polymorphic_ctype')
        return qs
    pass


@admin.register(models.Indirizzo)
class IndirizzoAdmin(HiddenModel, RicercaOrdinataMixin, AjaxAutocompleteListFilterModelAdmin, PolymorphicChildModelAdmin):
    # resource = resources.IndirizzoResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    list_per_page = 15
    paginator = CachingPaginator
    show_full_result_count = False
    fields=('soggetto','dug','duf','civico','cap','comune','altro',)
    search_fields = ('soggetto','indirizzo','cap_id__exact',)
    autocomplete_fields = ('soggetto','dug','cap','comune',)
    list_display = ('soggetto','indirizzo', 'cap','comune','provincia','regione',)
    ordering = ('comune_id','duf','civico',)
    autocomplete_list_filter = ('comune',)
    list_filter = ('comune__provincia__regione_id',)

    # def get_queryset(self, request):
    #     queryset = super().get_queryset(request).select_related('comune__provincia__regione').annotate(indirizzo=Concat('dug_id',Value(' '),'duf',Value(' '),'civico'))
    #     return queryset

    def provincia(self, obj):
        return obj.comune.provincia
    provincia.short_description = 'Provincia'
    provincia.admin_order_field = 'comune__provincia_id'

    def regione(self, obj):
        return obj.comune.provincia.regione_id
    regione.short_description = 'Regione'
    regione.admin_order_field = 'comune__provincia__regione_id'

    def indirizzo(self, obj):
        return obj.indirizzo
    indirizzo.short_description = 'Indirizzo'
    indirizzo.admin_order_field = ('comune_id','duf','civico',)
    pass


@admin.register(models.Societa)
class SocietaAdmin(ImportExportModelAdmin):
    # resource = resources.SocietaResource
    # list_per_page = 15
    # paginator = CachingPaginator
    # show_full_result_count = False
    pass