forked from feincms/feincms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmodeladmins.py
More file actions
235 lines (199 loc) · 8.18 KB
/
Copy pathmodeladmins.py
File metadata and controls
235 lines (199 loc) · 8.18 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
# ------------------------------------------------------------------------
# ------------------------------------------------------------------------
import os
from django import forms
from django.conf import settings as django_settings
from django.contrib import admin, messages
from django.contrib.admin.helpers import ACTION_CHECKBOX_NAME
from django.contrib.auth.decorators import permission_required
from django.contrib.sites.shortcuts import get_current_site
from django.core.files.images import get_image_dimensions
from django.http import HttpResponseRedirect
from django.shortcuts import render
from django.template.defaultfilters import filesizeformat
from django.urls import reverse
from django.utils.safestring import mark_safe
from django.utils.translation import gettext_lazy as _, ngettext
from django.views.decorators.csrf import csrf_protect
from feincms.extensions import ExtensionModelAdmin
from feincms.translations import admin_translationinline, lookup_translations
from feincms.utils import shorten_string
from .forms import MediaCategoryAdminForm, MediaFileAdminForm
from .models import Category, MediaFileTranslation
from .thumbnail import admin_thumbnail
from .zip import import_zipfile
# -----------------------------------------------------------------------
class CategoryAdmin(admin.ModelAdmin):
form = MediaCategoryAdminForm
list_display = ["path"]
list_filter = ["parent"]
list_per_page = 25
search_fields = ["title"]
prepopulated_fields = {"slug": ("title",)}
# ------------------------------------------------------------------------
@admin.action(description=_("Add selected media files to category"))
def assign_category(modeladmin, request, queryset):
class AddCategoryForm(forms.Form):
_selected_action = forms.CharField(widget=forms.MultipleHiddenInput)
category = forms.ModelChoiceField(Category.objects.all())
form = None
if "apply" in request.POST:
form = AddCategoryForm(request.POST)
if form.is_valid():
category = form.cleaned_data["category"]
count = 0
for mediafile in queryset:
category.mediafile_set.add(mediafile)
count += 1
message = ngettext(
"Successfully added %(count)d media file to %(category)s.",
"Successfully added %(count)d media files to %(category)s.",
count,
) % {"count": count, "category": category}
modeladmin.message_user(request, message)
return HttpResponseRedirect(request.get_full_path())
if "cancel" in request.POST:
return HttpResponseRedirect(request.get_full_path())
if not form:
form = AddCategoryForm(
initial={"_selected_action": request.POST.getlist(ACTION_CHECKBOX_NAME)}
)
return render(
request,
"admin/medialibrary/add_to_category.html",
{"mediafiles": queryset, "category_form": form, "opts": modeladmin.model._meta},
)
# -------------------------------------------------------------------------
@admin.action(description=_("Export selected media files as zip file"))
def save_as_zipfile(modeladmin, request, queryset):
from .zip import export_zipfile
site = get_current_site(request)
try:
zip_name = export_zipfile(site, queryset)
messages.info(request, _("ZIP file exported as %s") % zip_name)
except Exception as e:
messages.error(request, _("ZIP file export failed: %s") % str(e))
return
return HttpResponseRedirect(os.path.join(django_settings.MEDIA_URL, zip_name))
# ------------------------------------------------------------------------
class MediaFileAdmin(ExtensionModelAdmin):
form = MediaFileAdminForm
save_on_top = True
date_hierarchy = "created"
inlines = [admin_translationinline(MediaFileTranslation)]
list_display = ["admin_thumbnail", "__str__", "file_info", "formatted_created"]
list_display_links = ["__str__"]
list_filter = ["type", "categories"]
list_per_page = 25
search_fields = ["copyright", "file", "translations__caption"]
filter_horizontal = ("categories",)
actions = [assign_category, save_as_zipfile]
def get_urls(self):
from django.urls import path
return [
path(
"mediafile-bulk-upload/",
self.admin_site.admin_view(MediaFileAdmin.bulk_upload),
{},
name="mediafile_bulk_upload",
)
] + super().get_urls()
def changelist_view(self, request, extra_context=None):
if extra_context is None:
extra_context = {}
extra_context["categories"] = Category.objects.order_by("title")
return super().changelist_view(request, extra_context=extra_context)
@admin.display(description=_("Preview"))
def admin_thumbnail(self, obj):
image = admin_thumbnail(obj)
if image:
return mark_safe(
"""
<a href="%(url)s" target="_blank">
<img src="%(image)s" alt="" />
</a>"""
% {"url": obj.file.url, "image": image}
)
return ""
@admin.display(
description=_("file size"),
ordering="file_size",
)
def formatted_file_size(self, obj):
return filesizeformat(obj.file_size)
@admin.display(
description=_("created"),
ordering="created",
)
def formatted_created(self, obj):
return obj.created.strftime("%Y-%m-%d")
@admin.display(
description=_("file type"),
ordering="type",
)
def file_type(self, obj):
t = obj.filetypes_dict[obj.type]
if obj.type == "image":
# get_image_dimensions is expensive / slow if the storage is not
# local filesystem (indicated by availability the path property)
try:
obj.file.path
except NotImplementedError:
return t
try:
d = get_image_dimensions(obj.file.file, close=True)
if d:
t += " %d×%d" % (d[0], d[1])
except (OSError, TypeError, ValueError) as e:
t += " (%s)" % e
return mark_safe(t)
@admin.display(
description=_("file info"),
ordering="file",
)
def file_info(self, obj):
"""
Method for showing the file name in admin.
Note: This also includes a hidden field that can be used to extract
the file name later on, this can be used to access the file name from
JS, like for example a TinyMCE connector shim.
"""
return mark_safe(
(
'<input type="hidden" class="medialibrary_file_path"'
' name="_media_path_%d" value="%s" id="_refkey_%d" />'
" %s <br />%s, %s"
)
% (
obj.id,
obj.file.name,
obj.id,
shorten_string(os.path.basename(obj.file.name), max_length=40),
self.file_type(obj),
self.formatted_file_size(obj),
)
)
@staticmethod
@csrf_protect
@permission_required("medialibrary.add_mediafile")
def bulk_upload(request):
if request.method == "POST" and "data" in request.FILES:
try:
count = import_zipfile(
request.POST.get("category"),
request.POST.get("overwrite", False),
request.FILES["data"],
)
messages.info(request, _("%d files imported") % count)
except Exception as e:
messages.error(request, _("ZIP import failed: %s") % e)
else:
messages.error(request, _("No input file given"))
return HttpResponseRedirect(reverse("admin:medialibrary_mediafile_changelist"))
def get_queryset(self, request):
return super().get_queryset(request).transform(lookup_translations())
def save_model(self, request, obj, form, change):
if obj.id:
obj.purge_translation_cache()
return super().save_model(request, obj, form, change)
# ------------------------------------------------------------------------