from django.contrib import admin from django.utils.translation import ugettext as _ from games.forms import GameForm from games.models import Console, Game, Timeline class StatusFilter(admin.SimpleListFilter): """ Remove CREATED status. """ title = _('state') parameter_name = 'status' def lookups(self, request, model_admin): """ Return only games choices (not CREATED from Item abstract class). """ return Game.STATUS_CHOICES def queryset(self, request, queryset): """ Filter on 'status' field. """ if self.value(): return queryset.filter(status=self.value()) else: return queryset class GameAdmin(admin.ModelAdmin): list_display = ( 'name', 'playing', 'status', 'wish') list_filter = [ StatusFilter, 'playing', 'wish'] search_fields = ('name',) fieldsets = [ (_('Game Information'), { 'fields': [ ('name', 'collection')]}), (_('Progress'), {'fields': [ ('status'), ('note')]}), ('', {'fields': [ ('playing'), ('unplayed'), ('wish')]}) ] form = GameForm class TimelineAdmin(admin.ModelAdmin): list_display = ( 'date', 'status', 'item') admin.site.register(Console) admin.site.register(Game, GameAdmin) admin.site.register(Timeline, TimelineAdmin)