##// END OF EJS Templates
Fixed post admin
neko259 -
r1606:424b5c56 default
parent child Browse files
Show More
@@ -1,156 +1,156 b''
1 from boards.models.attachment import FILE_TYPES_IMAGE
1 from boards.models.attachment import FILE_TYPES_IMAGE
2 from django.contrib import admin
2 from django.contrib import admin
3 from django.utils.translation import ugettext_lazy as _
3 from django.utils.translation import ugettext_lazy as _
4 from django.core.urlresolvers import reverse
4 from django.core.urlresolvers import reverse
5 from boards.models import Post, Tag, Ban, Thread, Banner, Attachment, KeyPair, GlobalId
5 from boards.models import Post, Tag, Ban, Thread, Banner, Attachment, KeyPair, GlobalId
6
6
7
7
8 @admin.register(Post)
8 @admin.register(Post)
9 class PostAdmin(admin.ModelAdmin):
9 class PostAdmin(admin.ModelAdmin):
10
10
11 list_display = ('id', 'title', 'text', 'poster_ip', 'linked_images',
11 list_display = ('id', 'title', 'text', 'poster_ip', 'linked_images',
12 'foreign')
12 'foreign')
13 list_filter = ('pub_time',)
13 list_filter = ('pub_time',)
14 search_fields = ('id', 'title', 'text', 'poster_ip')
14 search_fields = ('id', 'title', 'text', 'poster_ip')
15 exclude = ('referenced_posts', 'refmap', 'images', 'global_id')
15 exclude = ('referenced_posts', 'refmap', 'images', 'global_id')
16 readonly_fields = ('poster_ip', 'threads', 'thread', 'linked_images',
16 readonly_fields = ('poster_ip', 'threads', 'thread', 'linked_images',
17 'attachments', 'uid', 'url', 'pub_time', 'opening', 'linked_global_id',
17 'attachments', 'uid', 'url', 'pub_time', 'opening', 'linked_global_id',
18 'version', 'foreign')
18 'version', 'foreign')
19
19
20 def ban_poster(self, request, queryset):
20 def ban_poster(self, request, queryset):
21 bans = 0
21 bans = 0
22 for post in queryset:
22 for post in queryset:
23 poster_ip = post.poster_ip
23 poster_ip = post.poster_ip
24 ban, created = Ban.objects.get_or_create(ip=poster_ip)
24 ban, created = Ban.objects.get_or_create(ip=poster_ip)
25 if created:
25 if created:
26 bans += 1
26 bans += 1
27 self.message_user(request, _('{} posters were banned').format(bans))
27 self.message_user(request, _('{} posters were banned').format(bans))
28
28
29 def ban_with_hiding(self, request, queryset):
29 def ban_with_hiding(self, request, queryset):
30 bans = 0
30 bans = 0
31 hidden = 0
31 hidden = 0
32 for post in queryset:
32 for post in queryset:
33 poster_ip = post.poster_ip
33 poster_ip = post.poster_ip
34 ban, created = Ban.objects.get_or_create(ip=poster_ip)
34 ban, created = Ban.objects.get_or_create(ip=poster_ip)
35 if created:
35 if created:
36 bans += 1
36 bans += 1
37 posts = Post.objects.filter(poster_ip=poster_ip, id__gte=post.id)
37 posts = Post.objects.filter(poster_ip=poster_ip, id__gte=post.id)
38 hidden += posts.count()
38 hidden += posts.count()
39 posts.update(hidden=True)
39 posts.update(hidden=True)
40 self.message_user(request, _('{} posters were banned, {} messages were hidden').format(bans, hidden))
40 self.message_user(request, _('{} posters were banned, {} messages were hidden').format(bans, hidden))
41
41
42 def linked_images(self, obj: Post):
42 def linked_images(self, obj: Post):
43 images = obj.attachments.filter(mimetype__in=FILE_TYPES_IMAGE)
43 images = obj.attachments.filter(mimetype__in=FILE_TYPES_IMAGE)
44 image_urls = ['<a href="{}"><img src="{}" /></a>'.format(
44 image_urls = ['<a href="{}"><img src="{}" /></a>'.format(
45 reverse('admin:%s_%s_change' % (image._meta.app_label,
45 reverse('admin:%s_%s_change' % (image._meta.app_label,
46 image._meta.model_name),
46 image._meta.model_name),
47 args=[image.id]), image.file.url_200x150) for image in images]
47 args=[image.id]), image.get_thumb_url()) for image in images]
48 return ', '.join(image_urls)
48 return ', '.join(image_urls)
49 linked_images.allow_tags = True
49 linked_images.allow_tags = True
50
50
51 def linked_global_id(self, obj: Post):
51 def linked_global_id(self, obj: Post):
52 global_id = obj.global_id
52 global_id = obj.global_id
53 if global_id is not None:
53 if global_id is not None:
54 return '<a href="{}">{}</a>'.format(
54 return '<a href="{}">{}</a>'.format(
55 reverse('admin:%s_%s_change' % (global_id._meta.app_label,
55 reverse('admin:%s_%s_change' % (global_id._meta.app_label,
56 global_id._meta.model_name),
56 global_id._meta.model_name),
57 args=[global_id.id]), str(global_id))
57 args=[global_id.id]), str(global_id))
58 linked_global_id.allow_tags = True
58 linked_global_id.allow_tags = True
59
59
60 def save_model(self, request, obj, form, change):
60 def save_model(self, request, obj, form, change):
61 obj.increment_version()
61 obj.increment_version()
62 obj.save()
62 obj.save()
63 obj.clear_cache()
63 obj.clear_cache()
64
64
65 def foreign(self, obj: Post):
65 def foreign(self, obj: Post):
66 return obj is not None and obj.global_id is not None and\
66 return obj is not None and obj.global_id is not None and\
67 not obj.global_id.is_local()
67 not obj.global_id.is_local()
68
68
69 actions = ['ban_poster', 'ban_with_hiding']
69 actions = ['ban_poster', 'ban_with_hiding']
70
70
71
71
72 @admin.register(Tag)
72 @admin.register(Tag)
73 class TagAdmin(admin.ModelAdmin):
73 class TagAdmin(admin.ModelAdmin):
74
74
75 def thread_count(self, obj: Tag) -> int:
75 def thread_count(self, obj: Tag) -> int:
76 return obj.get_thread_count()
76 return obj.get_thread_count()
77
77
78 def display_children(self, obj: Tag):
78 def display_children(self, obj: Tag):
79 return ', '.join([str(child) for child in obj.get_children().all()])
79 return ', '.join([str(child) for child in obj.get_children().all()])
80
80
81 def save_model(self, request, obj, form, change):
81 def save_model(self, request, obj, form, change):
82 super().save_model(request, obj, form, change)
82 super().save_model(request, obj, form, change)
83 for thread in obj.get_threads().all():
83 for thread in obj.get_threads().all():
84 thread.refresh_tags()
84 thread.refresh_tags()
85 list_display = ('name', 'thread_count', 'display_children')
85 list_display = ('name', 'thread_count', 'display_children')
86 search_fields = ('name',)
86 search_fields = ('name',)
87
87
88
88
89 @admin.register(Thread)
89 @admin.register(Thread)
90 class ThreadAdmin(admin.ModelAdmin):
90 class ThreadAdmin(admin.ModelAdmin):
91
91
92 def title(self, obj: Thread) -> str:
92 def title(self, obj: Thread) -> str:
93 return obj.get_opening_post().get_title()
93 return obj.get_opening_post().get_title()
94
94
95 def reply_count(self, obj: Thread) -> int:
95 def reply_count(self, obj: Thread) -> int:
96 return obj.get_reply_count()
96 return obj.get_reply_count()
97
97
98 def ip(self, obj: Thread):
98 def ip(self, obj: Thread):
99 return obj.get_opening_post().poster_ip
99 return obj.get_opening_post().poster_ip
100
100
101 def display_tags(self, obj: Thread):
101 def display_tags(self, obj: Thread):
102 return ', '.join([str(tag) for tag in obj.get_tags().all()])
102 return ', '.join([str(tag) for tag in obj.get_tags().all()])
103
103
104 def op(self, obj: Thread):
104 def op(self, obj: Thread):
105 return obj.get_opening_post_id()
105 return obj.get_opening_post_id()
106
106
107 # Save parent tags when editing tags
107 # Save parent tags when editing tags
108 def save_related(self, request, form, formsets, change):
108 def save_related(self, request, form, formsets, change):
109 super().save_related(request, form, formsets, change)
109 super().save_related(request, form, formsets, change)
110 form.instance.refresh_tags()
110 form.instance.refresh_tags()
111
111
112 def save_model(self, request, obj, form, change):
112 def save_model(self, request, obj, form, change):
113 op = obj.get_opening_post()
113 op = obj.get_opening_post()
114 op.increment_version()
114 op.increment_version()
115 op.save(update_fields=['version'])
115 op.save(update_fields=['version'])
116 obj.save()
116 obj.save()
117 op.clear_cache()
117 op.clear_cache()
118
118
119 list_display = ('id', 'op', 'title', 'reply_count', 'status', 'ip',
119 list_display = ('id', 'op', 'title', 'reply_count', 'status', 'ip',
120 'display_tags')
120 'display_tags')
121 list_filter = ('bump_time', 'status')
121 list_filter = ('bump_time', 'status')
122 search_fields = ('id', 'title')
122 search_fields = ('id', 'title')
123 filter_horizontal = ('tags',)
123 filter_horizontal = ('tags',)
124
124
125
125
126 @admin.register(KeyPair)
126 @admin.register(KeyPair)
127 class KeyPairAdmin(admin.ModelAdmin):
127 class KeyPairAdmin(admin.ModelAdmin):
128 list_display = ('public_key', 'primary')
128 list_display = ('public_key', 'primary')
129 list_filter = ('primary',)
129 list_filter = ('primary',)
130 search_fields = ('public_key',)
130 search_fields = ('public_key',)
131
131
132
132
133 @admin.register(Ban)
133 @admin.register(Ban)
134 class BanAdmin(admin.ModelAdmin):
134 class BanAdmin(admin.ModelAdmin):
135 list_display = ('ip', 'can_read')
135 list_display = ('ip', 'can_read')
136 list_filter = ('can_read',)
136 list_filter = ('can_read',)
137 search_fields = ('ip',)
137 search_fields = ('ip',)
138
138
139
139
140 @admin.register(Banner)
140 @admin.register(Banner)
141 class BannerAdmin(admin.ModelAdmin):
141 class BannerAdmin(admin.ModelAdmin):
142 list_display = ('title', 'text')
142 list_display = ('title', 'text')
143
143
144
144
145 @admin.register(Attachment)
145 @admin.register(Attachment)
146 class AttachmentAdmin(admin.ModelAdmin):
146 class AttachmentAdmin(admin.ModelAdmin):
147 search_fields = ('alias',)
147 search_fields = ('alias',)
148
148
149
149
150 @admin.register(GlobalId)
150 @admin.register(GlobalId)
151 class GlobalIdAdmin(admin.ModelAdmin):
151 class GlobalIdAdmin(admin.ModelAdmin):
152 def is_linked(self, obj):
152 def is_linked(self, obj):
153 return Post.objects.filter(global_id=obj).exists()
153 return Post.objects.filter(global_id=obj).exists()
154
154
155 list_display = ('__str__', 'is_linked',)
155 list_display = ('__str__', 'is_linked',)
156 readonly_fields = ('content',)
156 readonly_fields = ('content',)
General Comments 0
You need to be logged in to leave comments. Login now