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