mygpo/users/admin.py

   1 from django.contrib import admin
   2 from django.contrib.auth.admin import UserAdmin
   3 from django.contrib.auth.models import User
   4
   5 from mygpo.users.models import UserProfile, Client, SyncGroup
   6
   7 # Define an inline admin descriptor for the UserProfile model
   8 # which acts a bit like a singleton
   9 class UserProfileInline(admin.StackedInline):
  10     model = UserProfile
  11     can_delete = False
  12     verbose_name_plural = 'profile'
  13
  14 # Define a new User admin
  15 class UserAdmin(UserAdmin):
  16     inlines = (UserProfileInline, )
  17
  18     list_display = ('username', 'email', 'is_active', 'is_staff',
  19                     'is_superuser', 'date_joined', 'last_login')
  20
  21 # Re-register UserAdmin
  22 admin.site.unregister(User)
  23 admin.site.register(User, UserAdmin)
  24
  25
  26 @admin.register(Client)
  27 class ClientAdmin(admin.ModelAdmin):
  28     """ Admin page for clients """
  29
  30     # configuration for the list view
  31     list_display = ('name', 'user', 'uid', 'type', )
  32
  33     # fetch the client's user for the fields in list_display
  34     list_select_related = ('user', )
  35
  36     list_filter = ('type', )
  37     search_fields = ('name', 'uid', 'user__username', )
  38
  39     raw_id_fields = ('user', )
  40
  41
  42 @admin.register(SyncGroup)
  43 class SyncGroupAdmin(admin.ModelAdmin):
  44     """ Admin page for SyncGroups """
  45
  46     list_display = ('user', 'num_clients')
  47
  48     def num_clients(self, group):
  49         """ Numer of clients that belong to this group """
  50         return Client.objects.filter(sync_group=group).count()