modifying admin site

Author: CaptainAwesomeDi

polls/admin.py

@@ -1,4 +1,18 @@
 from django.contrib import admin
-from .models import Question
+from .models import Question,Choice
 # Register your models here.
-admin.site.register(Question)
+class ChoiceInline(admin.StackedInline):
+  model = Choice
+  extra = 3
+
+class QuestionAdmin(admin.ModelAdmin):
+  list_filter = ['pub_date']
+  list_display = ('question_text','pub_date','was_published_recently')
+  fieldsets = [
+    (None,               {'fields': ['question_text']}),
+    ('Date information', {'fields': ['pub_date'],'classes': ['collapse']}),
+  ]
+  inlines = [ChoiceInline]
+
+admin.site.register(Question,QuestionAdmin)
+admin.site.register(Choice)

polls/models.py

@@ -15,6 +15,10 @@ def was_published_recently(self):
         now = timezone.now()
         return now - datetime.timedelta(days=1) <=self.pub_date <= now
 
+    was_published_recently.admin_order_field = 'pub_date'
+    was_published_recently.boolean = True
+    was_published_recently.short_description = "Published recently?"
+
 
 class Choice(models.Model):
     question = models.ForeignKey(Question, on_delete=models.CASCADE)