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