from django.contrib import admin
from django.utils.safestring import mark_safe

from .models import Account, Badge, BadgeGrant, EmailConfirmation

# Register your models here.
class AccountAdmin(admin.ModelAdmin):
    list_display = ('user', 'acctname', 'email', 'is_email_confirmed', 'has_completed_setup')
    list_filter = ('is_email_confirmed', 'has_completed_setup')
    def email(self, obj):
        return obj.user.email
    email.short_description = 'Email'
admin.site.register(Account, AccountAdmin)

class BadgeAdmin(admin.ModelAdmin):
    list_display = ('name', 'icon', 'img_url')
    def icon(self, obj):
        return (mark_safe('<img src="%s" title="%s" height="32px" width="32px"/>' % (obj.img_url, obj.name)))
    icon.short_description = 'Icon'
admin.site.register(Badge, BadgeAdmin)

class GrantAdmin(admin.ModelAdmin):
    list_display = ('badge', 'account', 'expires', 'granted_by')
admin.site.register(BadgeGrant, GrantAdmin)

class ConfirmationAdmin(admin.ModelAdmin):
    list_display = ('email', 'user', 'key', 'expires')
admin.site.register(EmailConfirmation, ConfirmationAdmin)