from django.contrib import admin
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib.auth.models import User
from .models import UserProfile, CandidateProfile, Post, Voter, CandidatePost, Vote


class UserProfileInline(admin.StackedInline):
    model = UserProfile
    can_delete = False
    verbose_name_plural = "user_profile"


class UserAdmin(BaseUserAdmin):
    inlines = [UserProfileInline]


admin.site.unregister(User)
admin.site.register(User, UserAdmin)
admin.site.register(CandidateProfile)
admin.site.register(Post)
admin.site.register(Voter)
admin.site.register(CandidatePost)
admin.site.register(Vote)