from django.contrib import admin
from django.contrib.admin.sites import NotRegistered
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import User

from service.models import AO
from service.models import PlacementPoint
from service.models import PointDist
from service.models import Rayon
from service.models import Rivals

admin.site.register(AO)
admin.site.register(Rayon)
admin.site.register(Rivals)
admin.site.register(PointDist)


class PlacementPointAdmin(admin.ModelAdmin):
    pass


admin.site.register(PlacementPoint, PlacementPointAdmin)


class UserAdmin(UserAdmin):
    add_fieldsets = (
        (
            None, {
                'classes': ('wide',),
                'fields': ('username', 'password1', 'password2', 'email'),
            },
        ),
    )


try:
    admin.site.unregister(User)
except NotRegistered:
    pass

admin.site.register(User, UserAdmin)