admin.py

from django import forms
from django.contrib import admin
from django.contrib.postgres.fields import ArrayField
from django.utils.translation import gettext_lazy as _
from ordered_model.admin import OrderedModelAdmin

from osis_role.contrib.admin import EntityRoleModelAdmin, RoleModelAdmin
from partnership.auth.roles.partnership_manager import PartnershipEntityManager
from partnership.auth.roles.partnership_viewer import PartnershipViewer
from partnership.models import *


class TypeField(forms.CheckboxSelectMultiple):
    def format_value(self, value):
        if isinstance(value, str):
            value = value.split(',')
        return super().format_value(value)


@admin.register(PartnershipEntityManager)
class PartnershipEntityManagerAdmin(EntityRoleModelAdmin):
    list_display = ('person', 'entity', 'scopes')
    search_fields = ['person__first_name', 'person__last_name', 'entity__entityversion__acronym']
    raw_id_fields = ('person', 'entity',)
    formfield_overrides = {
        ArrayField: {'widget': TypeField(choices=PartnershipType.choices())}
    }


@admin.register(PartnershipViewer)
class PartnershipViewerAdmin(RoleModelAdmin):
    raw_id_fields = ('person',)


@admin.register(PartnerEntity)
class PartnerEntityAdmin(admin.ModelAdmin):
    model = PartnerEntity
    raw_id_fields = (
        'entity',
        'contact_in',
        'contact_out',
    )


@admin.register(Partner)
class PartnerAdmin(admin.ModelAdmin):
    search_fields = (
        'organization__name',
        'organization__code',
        'erasmus_code',
        'pic_code',
    )
    list_display = ('__str__', 'organization_code', 'erasmus_code', 'pic_code')
    raw_id_fields = (
        'now_known_as',
        'medias',
        'author',
    )

    @staticmethod
    def organization_code(obj):
        return obj.organization.code

    def save_form(self, request, form, change):
        """
        Set the author if the partner is being created
        """
        partner = form.save(commit=False)
        if not change:
            partner.author = request.user.person
        return partner

    def save_formset(self, request, form, formset, change):
        """
        Given an inline formset save it to the database.
        """
        if formset.model is PartnerEntity:
            entities = formset.save(commit=False)
            for entity in entities:
                if entity.pk is None:
                    entity.author = request.user.person
                entity.save()
        else:
            formset.save()


class PartnershipYearInline(admin.TabularInline):
    model = PartnershipYear
    raw_id_fields = (
        'academic_year',
        'education_fields',
        'entities',
        'offers',
    )
    extra = 0

    def get_queryset(self, request):
        return PartnershipYear.objects.select_related('academic_year')


class PartnershipAgreementInline(admin.TabularInline):
    model = PartnershipAgreement
    raw_id_fields = (
        'media',
    )
    extra = 0


class PartnershipPartnerRelationInline(admin.TabularInline):
    model = PartnershipPartnerRelation
    raw_id_fields = ('entity', )
    extra = 0


@admin.register(Partnership)
class PartnershipAdmin(admin.ModelAdmin):
    raw_id_fields = (
        'ucl_entity',
        'supervisor',
        'contacts',
        'medias',
    )
    inlines = (
        PartnershipPartnerRelationInline,
        PartnershipYearInline,
        PartnershipAgreementInline,
    )

    def save_form(self, request, form, change):
        """
        Set the author if the partnership is being created
        """
        partnership = form.save(commit=False)
        if not change:
            partnership.author = request.user.person
        return partnership


class ValueAdmin(admin.ModelAdmin):

    def get_model_perms(self, request):
        """
        Return empty perms dict thus hiding the model from admin index.
        """
        return {}


@admin.register(Financing)
class FinancingAdmin(admin.ModelAdmin):
    list_display = ('type', 'academic_year')
    search_fields = ('type__name', 'countries__name',)
    list_filter = ('type__name', 'academic_year')


@admin.register(FundingType)
class FundingTypeAdmin(admin.ModelAdmin):
    list_display = ('name', 'url', 'program')
    search_fields = ('name',)
    list_filter = ('program__name', 'program__source__name')


@admin.register(FundingProgram)
class FundingProgramAdmin(admin.ModelAdmin):
    list_display = ('name', 'source')
    search_fields = ('name',)
    list_filter = ('source__name',)


@admin.register(FundingSource)
class FundingSourceAdmin(admin.ModelAdmin):
    search_fields = ('name',)


@admin.register(Media)
class MediaAdmin(admin.ModelAdmin):
    search_fields = ('name', 'description')
    list_filter = ('visibility', 'is_visible_in_portal')


@admin.register(Contact)
class ContactAdmin(admin.ModelAdmin):
    search_fields = ('last_name', 'first_name', 'email')


class AddressAdmin(admin.ModelAdmin):
    raw_id_fields = ('country',)


@admin.register(PartnershipYear)
class PartnershipYearAdmin(admin.ModelAdmin):
    raw_id_fields = (
        'partnership',
        'academic_year',
        'entities',
        'offers',
    )


class PartnershipTypeListFilter(admin.SimpleListFilter):
    title = _('partnership_type')
    parameter_name = 'type'

    def lookups(self, request, model_admin):
        return PartnershipType.choices()

    def queryset(self, request, queryset):
        if self.value():
            return queryset.filter(types__contains=[self.value()])


@admin.register(PartnershipMission)
class PartnershipMissionAdmin(admin.ModelAdmin):
    list_display = ('code', 'label', 'type_values')
    formfield_overrides = {
        ArrayField: {'widget': TypeField(choices=PartnershipType.choices())}
    }
    list_filter = (PartnershipTypeListFilter, )

    @admin.display(
        description=_('partnership_type')
    )
    def type_values(self, obj):
        return ", ".join(str(PartnershipType.get_value(k)) for k in obj.types)


@admin.register(PartnershipSubtype)
class PartnershipSubtypeAdmin(OrderedModelAdmin, PartnershipMissionAdmin):
    list_display = PartnershipMissionAdmin.list_display + (
        'is_active',
        'order',
        'move_up_down_links',
    )
    readonly_fields = ['order']
    list_filter = (PartnershipTypeListFilter, 'is_active')
    ordering = ('order', )


@admin.register(UCLManagementEntity)
class UCLManagementEntityAdmin(admin.ModelAdmin):
    raw_id_fields = (
        'entity',
        'academic_responsible',
        'administrative_responsible',
        'contact_in_person',
        'contact_out_person',
    )


admin.site.register(PartnerTag)
admin.site.register(PartnershipTag)
admin.site.register(PartnershipYearEducationLevel)
admin.site.register(ContactType)
admin.site.register(MediaType)