from django.contrib import admin

from import_export import resources
from import_export.admin import ImportExportModelAdmin

from .models import InvitedGuest

class ExportInvitedGuestResource(resources.ModelResource):
    class Meta:
        model = InvitedGuest
        fields = ('name', 'attending')
        export_order = ('name', 'attending')
        exclude = 'rsvpCode'

class ImportInvitedGuestResource(resources.ModelResource):
    class Meta:
        model = InvitedGuest
        fields = ('name', 'rsvpCode')
        skip_unchanged = True
        report_skipped = False
        import_id_fields = ['name']


class InvitedGuestAdmin(ImportExportModelAdmin):
    resource_class = ImportInvitedGuestResource
    fields = ['name', 'rsvpCode', 'attending']
    list_display = ('name', 'rsvpCode', 'attending')
    search_fields = ['name']
    list_filter = ['attending']
    def get_export_resource_class(self):
        return ExportInvitedGuestResource


admin.site.register(InvitedGuest, InvitedGuestAdmin)