##// END OF EJS Templates
When editing tags in admin, add their parents as in a thread creation form
neko259 -
r1470:c40a0537 default
parent child Browse files
Show More
@@ -1,75 +1,84 b''
1 1 from django.contrib import admin
2 2 from boards.models import Post, Tag, Ban, Thread, Banner
3 3 from django.utils.translation import ugettext_lazy as _
4 4
5 5
6 6 @admin.register(Post)
7 7 class PostAdmin(admin.ModelAdmin):
8 8
9 9 list_display = ('id', 'title', 'text', 'poster_ip')
10 10 list_filter = ('pub_time',)
11 11 search_fields = ('id', 'title', 'text', 'poster_ip')
12 12 exclude = ('referenced_posts', 'refmap')
13 13 readonly_fields = ('poster_ip', 'threads', 'thread', 'images',
14 14 'attachments', 'uid', 'url', 'pub_time', 'opening')
15 15
16 16 def ban_poster(self, request, queryset):
17 17 bans = 0
18 18 for post in queryset:
19 19 poster_ip = post.poster_ip
20 20 ban, created = Ban.objects.get_or_create(ip=poster_ip)
21 21 if created:
22 22 bans += 1
23 23 self.message_user(request, _('{} posters were banned').format(bans))
24 24
25 25 actions = ['ban_poster']
26 26
27 27
28 28 @admin.register(Tag)
29 29 class TagAdmin(admin.ModelAdmin):
30 30
31 31 def thread_count(self, obj: Tag) -> int:
32 32 return obj.get_thread_count()
33 33
34 34 def display_children(self, obj: Tag):
35 35 return ', '.join([str(child) for child in obj.get_children().all()])
36 36
37 37 list_display = ('name', 'thread_count', 'display_children')
38 38 search_fields = ('name',)
39 39
40 40
41 41 @admin.register(Thread)
42 42 class ThreadAdmin(admin.ModelAdmin):
43 43
44 44 def title(self, obj: Thread) -> str:
45 45 return obj.get_opening_post().get_title()
46 46
47 47 def reply_count(self, obj: Thread) -> int:
48 48 return obj.get_reply_count()
49 49
50 50 def ip(self, obj: Thread):
51 51 return obj.get_opening_post().poster_ip
52 52
53 53 def display_tags(self, obj: Thread):
54 54 return ', '.join([str(tag) for tag in obj.get_tags().all()])
55 55
56 56 def op(self, obj: Thread):
57 57 return obj.get_opening_post_id()
58 58
59 # Save parent tags when editing tags
60 def save_related(self, request, form, formsets, change):
61 super().save_related(request, form, formsets, change)
62 obj = form.instance
63 for tag in obj.get_tags().all():
64 parents = tag.get_all_parents()
65 if len(parents) > 0:
66 obj.tags.add(*parents)
67
59 68 list_display = ('id', 'op', 'title', 'reply_count', 'status', 'ip',
60 69 'display_tags')
61 70 list_filter = ('bump_time', 'status')
62 71 search_fields = ('id', 'title')
63 72 filter_horizontal = ('tags',)
64 73
65 74
66 75 @admin.register(Ban)
67 76 class BanAdmin(admin.ModelAdmin):
68 77 list_display = ('ip', 'can_read')
69 78 list_filter = ('can_read',)
70 79 search_fields = ('ip',)
71 80
72 81
73 82 @admin.register(Banner)
74 83 class BannerAdmin(admin.ModelAdmin):
75 84 list_display = ('title', 'text')
General Comments 0
You need to be logged in to leave comments. Login now