# -*- coding: utf-8 -*-
from django.contrib import admin
from django.contrib.admin import ModelAdmin
from django.forms.models import ModelChoiceField

from mhackspace.rfid.models import Device, DeviceAuth


@admin.register(Device)
class DeviceAdmin(ModelAdmin):
    list_display = ('name', 'identifier')


# Probably need to look at this again
@admin.register(DeviceAuth)
class DeviceAuthAdmin(ModelAdmin):
    list_display = ('rfid', 'device')

    class CustomModelChoiceField(ModelChoiceField):
        def label_from_instance(self, obj):
            return obj.description + ' - ' + str(obj.user)

    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == "rfid":
            return self.CustomModelChoiceField(
                Rfid.objects.all(),
                initial=request.user)

        return super(DeviceAuthAdmin, self).formfield_for_foreignkey(
            db_field, request, **kwargs)