Change admin display of Avatar (#821)

* Change admin display of avatar * code climate fixes * Merge branch 'master' into avatar_admin * Merge branch 'master' into avatar_admin
ocadotechnology · Sep 26, 2018 · f3c320b · f3c320b
1 parent ec27ef2
commit f3c320b
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/aimmo/admin.py b/aimmo/admin.py
@@ -1,5 +1,4 @@
 from django.contrib import admin
-
 from models import Avatar, Game
 
 
@@ -11,5 +10,19 @@ class GameDataAdmin(admin.ModelAdmin):
     def players(self, obj):
         return "\n".join([u.first_name for u in obj.can_play.all()])
 
-admin.site.register(Avatar)
+
+class AvatarDataAdmin(admin.ModelAdmin):
+    list_display = ['id', 'game_identifier', 'game_name', 'owner_name']
+
+    def owner_name(self, obj):
+        return obj.owner.first_name
+
+    def game_name(self, obj):
+        return obj.game.name
+
+    def game_identifier(self, obj):
+        return obj.game.id
+
+
 admin.site.register(Game, GameDataAdmin)
+admin.site.register(Avatar, AvatarDataAdmin)