from django.contrib import admin

from .models import Choice, Question, Tag


class ChoiceInline(admin.TabularInline):
    model = Choice
    extra = 3


class QuestionAdmin(admin.ModelAdmin):
    fieldsets = [
        (None, {"fields": ["question_text"]}),
        ("Published date", {"fields": ["pub_date"], "classes": ["collapse"]}),
        ("End date", {"fields": ["end_date"], "classes": ["collapse"]}),
        ("Short Description", {"fields": ["short_description"], "classes": ["collapse"]}),
        ("Long Description", {"fields": ["long_description"], "classes": ["collapse"]}),
        ("Add Tag", {"fields": ["tags"], "classes": ["collapse"]})
    ]
    list_display = ["question_text", "pub_date", "end_date", "was_published_recently", "can_vote",
                    "trending_score", "get_tags"]
    inlines = [ChoiceInline]
    list_filter = ["pub_date", "end_date"]
    search_fields = ["question_text"]


# https://stackoverflow.com/questions/10904848/adding-inline-many-to-many-objects-in-django-admin
admin.site.register(Question, QuestionAdmin)
admin.site.register(Tag)  # Add Field to modify tags objects in Question