Mercurial > public > sg101

annotate gpp/bio/admin.py @ 348:d1b11096595b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix #168; when nailing a spammer, clear their profile text fields. Guard against topics and forums that don't exist when deleting posts in the signal handler. Make the forum stats template tag only display the latest active users.
author Brian Neal <bgneal@gmail.com>
date Wed, 02 Mar 2011 02:18:28 +0000
parents 69d0306a6fe7
children 9d470c7a2b93
rev   line source
gremmie@1 1 """
gremmie@1 2 This file contains the admin definitions for the bio application.
gremmie@1 3 """
bgneal@147 4 import datetime
gremmie@1 5
gremmie@1 6 from django.contrib import admin
bgneal@204 7
bgneal@347 8 import django.contrib.auth.models
bgneal@347 9 import django.contrib.auth.admin
bgneal@347 10
bgneal@147 11 import bio.models
bgneal@207 12 import bio.badges
bgneal@147 13 from comments.models import Comment
bgneal@147 14 from forums.tools import delete_user_posts
bgneal@147 15
gremmie@1 16
bgneal@204 17 class BadgeOwnerInline(admin.TabularInline):
bgneal@204 18 model = bio.models.BadgeOwnership
bgneal@204 19 extra = 1
bgneal@204 20
bgneal@204 21
gremmie@1 22 class UserProfileAdmin(admin.ModelAdmin):
bgneal@138 23 search_fields = ('user__username', 'user__first_name', 'user__last_name',
bgneal@138 24 'user__email')
gremmie@1 25 exclude = ('profile_html', 'signature_html')
bgneal@215 26 list_display = ('__unicode__', 'user_is_active', 'get_status_display', 'status_date')
bgneal@277 27 readonly_fields = ('status', 'status_date', 'update_date')
bgneal@147 28 list_filter = ('status', )
bgneal@147 29 date_hierarchy = 'status_date'
bgneal@204 30 inlines = (BadgeOwnerInline, )
bgneal@147 31 actions = (
bgneal@347 32 'mark_active',
bgneal@147 33 'mark_resigned',
bgneal@147 34 'mark_removed',
bgneal@147 35 'mark_suspended',
bgneal@147 36 'mark_spammer',
bgneal@215 37 'mark_stranger',
bgneal@147 38 )
bgneal@147 39
bgneal@147 40 def get_status_display(self, obj):
bgneal@147 41 return obj.get_status_display()
bgneal@147 42 get_status_display.short_description = 'Status'
bgneal@147 43
bgneal@147 44 def mark_user_status(self, request, qs, status):
bgneal@147 45 """
bgneal@347 46 Common code for the admin actions. Updates the status field in the
bgneal@347 47 profiles to 'status'. Updates the status_date. Sets the is_active
bgneal@147 48 field to True if the status is STA_ACTIVE and False otherwise.
bgneal@147 49 """
bgneal@147 50 now = datetime.datetime.now()
bgneal@147 51 for profile in qs:
bgneal@215 52 profile.user.is_active = (status == bio.models.STA_ACTIVE or
bgneal@215 53 status == bio.models.STA_STRANGER)
bgneal@147 54 profile.user.save()
bgneal@147 55 profile.status = status
bgneal@147 56 profile.status_date = now
bgneal@147 57 profile.save()
bgneal@147 58
bgneal@147 59 count = qs.count()
bgneal@147 60 msg = "1 user" if count == 1 else "%d users" % count
bgneal@147 61 self.message_user(request, "%s successfully marked as %s." % (msg,
bgneal@147 62 bio.models.USER_STATUS_CHOICES[status][1]))
bgneal@147 63
bgneal@147 64 def mark_active(self, request, qs):
bgneal@147 65 """
bgneal@147 66 Marks users as active. Updates their profile status to STA_ACTIVE.
bgneal@147 67 """
bgneal@147 68 self.mark_user_status(request, qs, bio.models.STA_ACTIVE)
bgneal@147 69 mark_active.short_description = "Mark selected users as active"
bgneal@147 70
bgneal@147 71 def mark_resigned(self, request, qs):
bgneal@147 72 """
bgneal@147 73 Marks users as inactive. Updates their profile status to STA_RESIGNED.
bgneal@147 74 """
bgneal@147 75 self.mark_user_status(request, qs, bio.models.STA_RESIGNED)
bgneal@147 76 mark_resigned.short_description = "Mark selected users as resigned"
bgneal@147 77
bgneal@147 78 def mark_removed(self, request, qs):
bgneal@147 79 """
bgneal@147 80 Marks users as inactive. Updates their profile status to STA_REMOVED.
bgneal@147 81 """
bgneal@147 82 self.mark_user_status(request, qs, bio.models.STA_REMOVED)
bgneal@147 83 mark_removed.short_description = "Mark selected users as removed"
bgneal@147 84
bgneal@147 85 def mark_suspended(self, request, qs):
bgneal@147 86 """
bgneal@147 87 Marks users as inactive. Updates their profile status to STA_SUSPENDED.
bgneal@147 88 """
bgneal@147 89 self.mark_user_status(request, qs, bio.models.STA_SUSPENDED)
bgneal@147 90 mark_suspended.short_description = "Mark selected users as suspended"
bgneal@147 91
bgneal@147 92 def mark_spammer(self, request, qs):
bgneal@147 93 """
bgneal@147 94 Marks users as inactive. Updates their profile status to STA_SPAMMER.
bgneal@147 95 Deletes all their comments and forum posts.
bgneal@147 96 """
bgneal@147 97 self.mark_user_status(request, qs, bio.models.STA_SPAMMER)
bgneal@147 98 for profile in qs:
bgneal@147 99 Comment.objects.filter(user=profile.user).delete()
bgneal@147 100 delete_user_posts(profile.user)
bgneal@348 101 profile.location = ''
bgneal@348 102 profile.occupation = ''
bgneal@348 103 profile.interests = ''
bgneal@348 104 profile.profile_text = ''
bgneal@348 105 profile.signature = ''
bgneal@348 106 profile.save()
bgneal@147 107 mark_spammer.short_description = "Mark selected users as spammers"
gremmie@1 108
bgneal@215 109 def mark_stranger(self, request, qs):
bgneal@215 110 """
bgneal@215 111 Marks users as strangers. Updates their profile status to STA_STRANGER.
bgneal@215 112 """
bgneal@215 113 self.mark_user_status(request, qs, bio.models.STA_STRANGER)
bgneal@215 114 mark_stranger.short_description = "Mark selected users as strangers"
bgneal@215 115
bgneal@138 116
bgneal@138 117 class UserProfileFlagAdmin(admin.ModelAdmin):
bgneal@138 118 list_display = ('__unicode__', 'flag_date', 'get_profile_url')
bgneal@207 119 actions = ('accept_flags', )
bgneal@207 120
bgneal@207 121 def accept_flags(self, request, qs):
bgneal@207 122 """This action awards a security pin to the user that reported the
bgneal@207 123 profile and then deletes the flag.
bgneal@207 124 """
bgneal@207 125 for flag in qs:
bgneal@207 126 bio.badges.award_badge(bio.badges.SECURITY_PIN, flag.user)
bgneal@207 127 flag.delete()
bgneal@207 128
bgneal@207 129 accept_flags.short_description = "Accept selected flagged profiles"
bgneal@207 130
bgneal@207 131
bgneal@204 132 class BadgeAdmin(admin.ModelAdmin):
bgneal@204 133 list_display = ('name', 'html', 'order', 'numeric_id', 'description')
bgneal@204 134 list_editable = ('order', 'numeric_id')
bgneal@204 135
bgneal@204 136
bgneal@347 137 # We like the User admin but would like a date hierarcy on date_joined.
bgneal@347 138 class UserAdmin(django.contrib.auth.admin.UserAdmin):
bgneal@347 139 date_hierarchy = 'date_joined'
bgneal@347 140
bgneal@347 141
bgneal@204 142 admin.site.register(bio.models.UserProfile, UserProfileAdmin)
bgneal@204 143 admin.site.register(bio.models.UserProfileFlag, UserProfileFlagAdmin)
bgneal@204 144 admin.site.register(bio.models.Badge, BadgeAdmin)
bgneal@347 145
bgneal@347 146 # Unregister existing ModelAdmin for User, then register ours
bgneal@347 147 admin.site.unregister(django.contrib.auth.models.User)
bgneal@347 148 admin.site.register(django.contrib.auth.models.User, UserAdmin)