# Import your custom models
from .models import Post, Comment
from django.contrib.auth.forms import UserCreationForm, UserChangeForm
from django.contrib.auth.models import User

import extra_views

import djadmin2
from djadmin2.models import ModelAdmin2


class CommentInline(extra_views.InlineFormSet):
    model = Comment


class PostAdmin(ModelAdmin2):
    inlines = [CommentInline]


class UserAdmin2(ModelAdmin2):
    create_form_class = UserCreationForm
    update_form_class = UserChangeForm


#  Register each model with the admin
djadmin2.default.register(Post, PostAdmin)
djadmin2.default.register(Comment)
djadmin2.default.register(User, UserAdmin2)