from django.contrib import admin

from models import EveCharacter
from models import EveApiKeyPair
from models import EveAllianceInfo
from models import EveCorporationInfo
from authentication.managers import AuthServicesInfoManager

admin.site.register(EveAllianceInfo)
admin.site.register(EveCorporationInfo)

class EveApiKeyPairAdmin(admin.ModelAdmin):
    search_fields = ['api_id', 'user__username']
    list_display = ['api_id', 'user']

class EveCharacterAdmin(admin.ModelAdmin):
    search_fields = ['character_name', 'corporation_name', 'alliance_name', 'user__username', 'api_id']
    list_display = ('character_name', 'corporation_name', 'alliance_name', 'user', 'main_character')

    def main_character(self, obj):
        auth = AuthServicesInfoManager.get_auth_for_user(obj.user)
        if auth and auth.main_char_id:
            try:
                return EveCharacter.objects.get(character_id=auth.main_char_id)
            except EveCharacter.DoesNotExist:
                pass
        return None

admin.site.register(EveCharacter, EveCharacterAdmin)
admin.site.register(EveApiKeyPair, EveApiKeyPairAdmin)