admin.py
126 lines
| 4.1 KiB
| text/x-python
|
PythonLexer
/ boards / admin.py
neko259
|
r0 | from django.contrib import admin | |
neko259
|
r968 | from django.utils.translation import ugettext_lazy as _ | |
neko259
|
r1500 | from django.core.urlresolvers import reverse | |
neko259
|
r1511 | from boards.models import Post, Tag, Ban, Thread, Banner, PostImage, KeyPair, GlobalId | |
neko259
|
r0 | ||
neko259
|
r333 | ||
neko259
|
r906 | @admin.register(Post) | |
neko259
|
r305 | class PostAdmin(admin.ModelAdmin): | |
neko259
|
r1500 | list_display = ('id', 'title', 'text', 'poster_ip', 'linked_images') | |
neko259
|
r969 | list_filter = ('pub_time',) | |
neko259
|
r1130 | search_fields = ('id', 'title', 'text', 'poster_ip') | |
neko259
|
r1540 | exclude = ('referenced_posts', 'refmap', 'images') | |
neko259
|
r1500 | readonly_fields = ('poster_ip', 'threads', 'thread', 'linked_images', | |
neko259
|
r1389 | 'attachments', 'uid', 'url', 'pub_time', 'opening') | |
neko259
|
r305 | ||
neko259
|
r968 | 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)) | |||
neko259
|
r1495 | def ban_with_hiding(self, request, queryset): | |
bans = 0 | |||
hidden = 0 | |||
for post in queryset: | |||
poster_ip = post.poster_ip | |||
ban, created = Ban.objects.get_or_create(ip=poster_ip) | |||
if created: | |||
bans += 1 | |||
neko259
|
r1496 | posts = Post.objects.filter(poster_ip=poster_ip, id__gte=post.id) | |
neko259
|
r1495 | hidden += posts.count() | |
posts.update(hidden=True) | |||
self.message_user(request, _('{} posters were banned, {} messages were hidden').format(bans, hidden)) | |||
neko259
|
r1500 | def linked_images(self, obj: Post): | |
images = obj.images.all() | |||
neko259
|
r1541 | image_urls = ['<a href="{}"><img src="{}" /></a>'.format( | |
reverse('admin:%s_%s_change' % (image._meta.app_label, | |||
image._meta.model_name), | |||
args=[image.id]), image.image.url_200x150) for image in images] | |||
neko259
|
r1500 | return ', '.join(image_urls) | |
linked_images.allow_tags = True | |||
neko259
|
r1495 | ||
actions = ['ban_poster', 'ban_with_hiding'] | |||
neko259
|
r968 | ||
neko259
|
r333 | ||
neko259
|
r906 | @admin.register(Tag) | |
neko259
|
r305 | class TagAdmin(admin.ModelAdmin): | |
neko259
|
r910 | def thread_count(self, obj: Tag) -> int: | |
return obj.get_thread_count() | |||
neko259
|
r1356 | def display_children(self, obj: Tag): | |
return ', '.join([str(child) for child in obj.get_children().all()]) | |||
neko259
|
r1471 | def save_model(self, request, obj, form, change): | |
super().save_model(request, obj, form, change) | |||
for thread in obj.get_threads().all(): | |||
thread.refresh_tags() | |||
neko259
|
r1356 | list_display = ('name', 'thread_count', 'display_children') | |
neko259
|
r906 | search_fields = ('name',) | |
neko259
|
r305 | ||
neko259
|
r883 | ||
neko259
|
r906 | @admin.register(Thread) | |
neko259
|
r604 | class ThreadAdmin(admin.ModelAdmin): | |
neko259
|
r910 | def title(self, obj: Thread) -> str: | |
neko259
|
r883 | return obj.get_opening_post().get_title() | |
neko259
|
r604 | ||
neko259
|
r910 | def reply_count(self, obj: Thread) -> int: | |
neko259
|
r604 | return obj.get_reply_count() | |
neko259
|
r910 | def ip(self, obj: Thread): | |
neko259
|
r906 | return obj.get_opening_post().poster_ip | |
neko259
|
r1356 | def display_tags(self, obj: Thread): | |
return ', '.join([str(tag) for tag in obj.get_tags().all()]) | |||
def op(self, obj: Thread): | |||
return obj.get_opening_post_id() | |||
neko259
|
r1470 | # Save parent tags when editing tags | |
def save_related(self, request, form, formsets, change): | |||
super().save_related(request, form, formsets, change) | |||
neko259
|
r1471 | form.instance.refresh_tags() | |
neko259
|
r1414 | list_display = ('id', 'op', 'title', 'reply_count', 'status', 'ip', | |
neko259
|
r1356 | 'display_tags') | |
neko259
|
r1414 | list_filter = ('bump_time', 'status') | |
neko259
|
r604 | search_fields = ('id', 'title') | |
neko259
|
r921 | filter_horizontal = ('tags',) | |
neko259
|
r604 | ||
neko259
|
r883 | ||
neko259
|
r933 | @admin.register(KeyPair) | |
neko259
|
r797 | class KeyPairAdmin(admin.ModelAdmin): | |
list_display = ('public_key', 'primary') | |||
list_filter = ('primary',) | |||
search_fields = ('public_key',) | |||
neko259
|
r933 | ||
neko259
|
r906 | @admin.register(Ban) | |
neko259
|
r804 | class BanAdmin(admin.ModelAdmin): | |
list_display = ('ip', 'can_read') | |||
list_filter = ('can_read',) | |||
neko259
|
r921 | search_fields = ('ip',) | |
neko259
|
r1148 | ||
@admin.register(Banner) | |||
class BannerAdmin(admin.ModelAdmin): | |||
list_display = ('title', 'text') | |||
neko259
|
r1500 | ||
@admin.register(PostImage) | |||
class PostImageAdmin(admin.ModelAdmin): | |||
search_fields = ('alias',) | |||
neko259
|
r1511 | ||
@admin.register(GlobalId) | |||
class GlobalIdAdmin(admin.ModelAdmin): | |||
def is_linked(self, obj): | |||
return Post.objects.filter(global_id=obj).exists() | |||
list_display = ('__str__', 'is_linked',) |