|
|
|
|
@ -1,4 +1,5 @@
|
|
|
|
|
from django.contrib import admin
|
|
|
|
|
from django.contrib.admin import AdminSite
|
|
|
|
|
from django.contrib.admin.sites import NotRegistered
|
|
|
|
|
from django.contrib.auth.admin import UserAdmin
|
|
|
|
|
from django.contrib.auth.models import User
|
|
|
|
|
@ -9,17 +10,28 @@ 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 MyAdminSite(AdminSite):
|
|
|
|
|
def index(self, request, extra_context=None):
|
|
|
|
|
if extra_context is None:
|
|
|
|
|
extra_context = {}
|
|
|
|
|
extra_context['is_superadmin'] = request.user.is_superuser
|
|
|
|
|
return super(MyAdminSite, self).index(request, extra_context)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
my_admin_site = MyAdminSite(name='myadmin')
|
|
|
|
|
|
|
|
|
|
my_admin_site.register(AO)
|
|
|
|
|
my_admin_site.register(Rayon)
|
|
|
|
|
my_admin_site.register(Rivals)
|
|
|
|
|
my_admin_site.register(PointDist)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class PlacementPointAdmin(admin.ModelAdmin):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
admin.site.register(PlacementPoint, PlacementPointAdmin)
|
|
|
|
|
my_admin_site.register(PlacementPoint, PlacementPointAdmin)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UserAdmin(UserAdmin):
|
|
|
|
|
@ -34,8 +46,8 @@ class UserAdmin(UserAdmin):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
admin.site.unregister(User)
|
|
|
|
|
my_admin_site.unregister(User)
|
|
|
|
|
except NotRegistered:
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
admin.site.register(User, UserAdmin)
|
|
|
|
|
my_admin_site.register(User, UserAdmin)
|
|
|
|
|
|