admin.py

__copyright__ = "Copyright 2016-2018, Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


from django.contrib import admin

from django import forms

from .models import VGM, Feedback, Nomination, Motion

from scipost.models import Contributor


class VGMAdmin(admin.ModelAdmin):
    search_fields = ['start_date']


admin.site.register(VGM, VGMAdmin)


class FeedbackAdmin(admin.ModelAdmin):
    search_fields = ['feedback', 'by']


admin.site.register(Feedback, FeedbackAdmin)


class NominationAdminForm(forms.ModelForm):
    in_agreement = forms.ModelMultipleChoiceField(
        required=False,
        queryset=Contributor.objects.filter(
            user__groups__name__in=['Editorial College'],
        ).order_by('user__last_name'))
    in_notsure = forms.ModelMultipleChoiceField(
        required=False,
        queryset=Contributor.objects.filter(
            user__groups__name__in=['Editorial College'],
        ).order_by('user__last_name'))
    in_disagreement = forms.ModelMultipleChoiceField(
        required=False,
        queryset=Contributor.objects.filter(
            user__groups__name__in=['Editorial College'],
        ).order_by('user__last_name'))

    class Meta:
        model = Nomination
        fields = '__all__'

class NominationAdmin(admin.ModelAdmin):
    search_fields = ['last_name', 'first_name', 'by']
    form = NominationAdminForm

admin.site.register(Nomination, NominationAdmin)


class MotionAdminForm(forms.ModelForm):
    in_agreement = forms.ModelMultipleChoiceField(
        required=False,
        queryset=Contributor.objects.filter(
            user__groups__name__in=['Editorial College'],
        ).order_by('user__last_name'))
    in_notsure = forms.ModelMultipleChoiceField(
        required=False,
        queryset=Contributor.objects.filter(
            user__groups__name__in=['Editorial College'],
        ).order_by('user__last_name'))
    in_disagreement = forms.ModelMultipleChoiceField(
        required=False,
        queryset=Contributor.objects.filter(
            user__groups__name__in=['Editorial College'],
        ).order_by('user__last_name'))

    class Meta:
        model = Motion
        fields = '__all__'

class MotionAdmin(admin.ModelAdmin):
    search_fields = ['background', 'motion', 'put_forward_by']
    form = MotionAdminForm

admin.site.register(Motion, MotionAdmin)