Added admin action to ban user that posted given messages

boards/admin.py

@@ -1,47 +1,59 b''
 from django.contrib import admin
 from boards.models import Post, Tag, Ban, Thread
+from django.utils.translation import ugettext_lazy as _
 
 
 @admin.register(Post)
 class PostAdmin(admin.ModelAdmin):
 
     list_display = ('id', 'title', 'text')
     list_filter = ('pub_time', 'thread_new')
     search_fields = ('id', 'title', 'text')
     exclude = ('referenced_posts', 'refmap')
     readonly_fields = ('poster_ip', 'thread_new')
 
+    def ban_poster(self, request, queryset):
+        bans = 0
+        for post in queryset:
+            poster_ip = post.poster_ip
+            ban, created = Ban.objects.get_or_create(ip=poster_ip)
+            if created:
+                bans += 1
+        self.message_user(request, _('{} posters were banned').format(bans))
+
+    actions = ['ban_poster']
+
 
 @admin.register(Tag)
 class TagAdmin(admin.ModelAdmin):
 
     def thread_count(self, obj: Tag) -> int:
         return obj.get_thread_count()
 
     list_display = ('name', 'thread_count')
     search_fields = ('name',)
 
 
 @admin.register(Thread)
 class ThreadAdmin(admin.ModelAdmin):
 
     def title(self, obj: Thread) -> str:
         return obj.get_opening_post().get_title()
 
     def reply_count(self, obj: Thread) -> int:
         return obj.get_reply_count()
 
     def ip(self, obj: Thread):
         return obj.get_opening_post().poster_ip
 
     list_display = ('id', 'title', 'reply_count', 'archived', 'ip')
     list_filter = ('bump_time', 'archived', 'bumpable')
     search_fields = ('id', 'title')
     filter_horizontal = ('tags',)
 
 
 @admin.register(Ban)
 class BanAdmin(admin.ModelAdmin):
     list_display = ('ip', 'can_read')
     list_filter = ('can_read',)
     search_fields = ('ip',)

boards/views/base.py

@@ -1,31 +1,30 b''
 from django.db import transaction
-from django.template import RequestContext
 from django.views.generic import View
 
 from boards import utils
 from boards.models.user import Ban
 
 
 BAN_REASON_SPAM = 'Autoban: spam bot'
 
 CONTEXT_FORM = 'form'
 
 
 class BaseBoardView(View):
 
     def get_context_data(self, **kwargs):
         return dict()
 
     @transaction.atomic
     def _ban_current_user(self, request):
         """
         Add current user to the IP ban list
         """
 
         ip = utils.get_client_ip(request)
         ban, created = Ban.objects.get_or_create(ip=ip)
         if created:
             ban.can_read = False
             ban.reason = BAN_REASON_SPAM
         ban.save()