Limit moderation by category: subcategoryuserlimit model - modify admin for config

1 files changed, 18 insertions, 4 deletions

diff --git a/chimere/admin.py b/chimere/admin.py
index 60d1141..5a95ece 100644
--- a/chimere/admin.py
+++ b/chimere/admin.py
@@ -20,12 +20,11 @@
 """
 Settings for administration pages
 """
-import datetime
-
-from django import forms
 from django.conf import settings
 from django.contrib import admin, messages
 from django.contrib.admin import SimpleListFilter
+from django.contrib.auth.admin import UserAdmin as VanillaUserAdmin
+from django.contrib.auth.models import User
 from django.core.exceptions import ObjectDoesNotExist
 from django.db.models import Q
 from django.http import HttpResponse, HttpResponseRedirect
@@ -45,7 +44,8 @@ from chimere.models import Category, Icon, SubCategory, Marker, \
      PropertyModel, News, Route, Area, ColorTheme, Color, \
      MultimediaFile, PictureFile, Importer, Layer, AreaLayers,\
      PropertyModelChoice, MultimediaExtension, Page,\
-     get_areas_for_user, get_users_by_area, ImporterKeyCategories
+     get_areas_for_user, get_users_by_area, ImporterKeyCategories,\
+     SubCategoryUserLimit
 from chimere.utils import unicode_normalize, ShapefileManager, KMLManager,\
                           CSVManager
 
@@ -164,6 +164,20 @@ def managed_modified(modeladmin, request, queryset):
                               context_instance=RequestContext(request))
 managed_modified.short_description = _(u"Managed modified items")
 
+
+class CatLimitInline(admin.TabularInline):
+    model = SubCategoryUserLimit
+    extra = 5
+
+
+class UserAdmin(VanillaUserAdmin):
+    list_display = ('username', 'email', 'first_name', 'last_name', 'is_staff')
+    inlines = (CatLimitInline,)
+
+admin.site.unregister(User)
+admin.site.register(User, UserAdmin)
+
+
 class PictureInline(admin.TabularInline):
     model = PictureFile
     extra = 1