Mathis/allianceauth
1
0
Fork 0
mirror of https://gitlab.com/allianceauth/allianceauth.git synced 2025-07-12 22:10:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add main character field to character admin

Browse Source
This commit is contained in:
Adarnof 2016-07-28 12:02:39 -04:00
parent c14e0aecef
commit 507634ffdc
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
eveonline/admin.py
View File

@ -4,14 +4,27 @@ from models import EveCharacter
from models import EveApiKeyPair from models import EveApiKeyPair
from models import EveAllianceInfo from models import EveAllianceInfo
from models import EveCorporationInfo from models import EveCorporationInfo
from authentication.managers import AuthServicesInfoManager
admin.site.register(EveApiKeyPair)
admin.site.register(EveAllianceInfo) admin.site.register(EveAllianceInfo)
admin.site.register(EveCorporationInfo) admin.site.register(EveCorporationInfo)
class EveApiKeyPairAdmin(admin.ModelAdmin):
search_fields = ['api_id', 'user__username']
list_display = ['api_id', 'user']
class EveCharacterAdmin(admin.ModelAdmin): class EveCharacterAdmin(admin.ModelAdmin):
search_fields = ['character_name', 'corporation_name', 'alliance_name', 'user__username'] search_fields = ['character_name', 'corporation_name', 'alliance_name', 'user__username', 'api_id']
list_display = ('character_name', 'corporation_name', 'alliance_name', 'user') 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(EveCharacter, EveCharacterAdmin)
admin.site.register(EveApiKeyPair, EveApiKeyPairAdmin)