Completed
Push — master ( c427d1...c91022 )
by Batiste
7s
created

PageAdmin.get_fieldsets()   D

Complexity

Conditions 8

Size

Total Lines 41

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
dl 0
loc 41
rs 4
c 0
b 0
f 0
cc 8
1
# -*- coding: utf-8 -*-
2
"""Page Admin module."""
3
from pages import settings
4
from pages.models import Page, Content, PageAlias
5
from pages.phttp import get_language_from_request, get_template_from_request
6
from pages.utils import get_placeholders
7
from pages.templatetags.pages_tags import PlaceholderNode
8
from pages.admin.forms import make_form
9
from pages.admin.views import traduction, get_content, sub_menu
10
from pages.admin.views import change_status, modify_content, delete_content
11
from pages.admin.views import move_page
12
13
from collections import defaultdict
14
from django.contrib import admin
15
from django.utils.translation import ugettext_lazy as _
16
from django.utils.encoding import force_text
17
from django.conf import settings as global_settings
18
from django.http import HttpResponseRedirect, Http404
19
from django.contrib.admin.sites import AlreadyRegistered
20
if global_settings.USE_I18N:
21
    from django.views.i18n import javascript_catalog
22
else:
23
    from django.views.i18n import null_javascript_catalog as javascript_catalog
24
25
from os.path import join
26
27
28
class PageAdmin(admin.ModelAdmin):
29
    """Page Admin class."""
30
31
    # these mandatory fields are not versioned
32
    mandatory_placeholders = ('title', 'slug')
33
    general_fields = ['title', 'slug', 'status', 'target',
34
        'position', 'freeze_date', 'template', 'language',
35
        'redirect_to', 'redirect_to_url']
36
37
    if settings.PAGE_USE_SITE_ID and not settings.PAGE_HIDE_SITES:
38
        general_fields.append('sites')
39
    insert_point = general_fields.index('status') + 1
40
41
    # Strange django behavior. If not provided, django will try to find
42
    # 'page' foreign key in all registered models
43
    inlines = []
44
45
    if settings.PAGE_TAGGING:
46
        general_fields.insert(insert_point, 'tags')
47
48
    # Add support for future dating and expiration based on settings.
49
    if settings.PAGE_SHOW_END_DATE:
50
        general_fields.insert(insert_point, 'publication_end_date')
51
    if settings.PAGE_SHOW_START_DATE:
52
        general_fields.insert(insert_point, 'publication_date')
53
54
    from pages.urlconf_registry import registry
55
    if(len(registry)):
56
        general_fields.append('delegate_to')
57
        insert_point = general_fields.index('status') + 1
58
59
    page_templates = settings.get_page_templates()
60
61
    fieldsets = (
62
        [_('General'), {
63
            'fields': general_fields,
64
            'classes': ('module-general',),
65
        }],
66
    )
67
68
    class Media:
69
        css = {
70
            'all': [join(settings.PAGES_MEDIA_URL, path) for path in (
71
                'css/rte.css',
72
                'css/pages.css'
73
            )]
74
        }
75
        js = [join(settings.PAGES_MEDIA_URL, path) for path in (
76
            'javascript/jquery.js',
77
            'javascript/jquery.rte.js',
78
            'javascript/pages.js',
79
            'javascript/pages_list.js',
80
            'javascript/pages_form.js',
81
            'javascript/jquery.query-2.1.7.js',
82
        )]
83
84
    @classmethod
85
    def add_action(cls, method):
86
        if method not in cls.actions:
87
            cls.actions.append(method)
88
89
    def get_urls(self):
90
        urls = super(PageAdmin, self).get_urls()
91
        from django.conf.urls import url
92
93
        pages_urls = [
94
            url(r'^$', self.list_pages, name='page-changelist'),
95
            url(r'^(?P<page_id>[0-9]+)/traduction/(?P<language_id>[-\w]+)/$',
96
                traduction, name='page-traduction'),
97
            url(r'^(?P<page_id>[0-9]+)/get-content/(?P<content_id>[0-9]+)/$',
98
                get_content, name='page-get-content'),
99
            url(r'^(?P<page_id>[0-9]+)/modify-content/(?P<content_type>[-\w]+)/(?P<language_id>[-\w]+)/$',
100
                modify_content, name='page-modify-content'),
101
            url(r'^(?P<page_id>[0-9]+)/delete-content/(?P<language_id>[-\w]+)/$',
102
                delete_content, name='page-delete-content'),
103
            url(r'^(?P<page_id>[0-9]+)/sub-menu/$',
104
                sub_menu, name='page-sub-menu'),
105
            url(r'^(?P<page_id>[0-9]+)/move-page/$',
106
                move_page, name='page-move-page'),
107
            url(r'^(?P<page_id>[0-9]+)/change-status/$',
108
                change_status, name='page-change-status'),
109
        ]
110
111
        return pages_urls + urls
112
113
114
    def i18n_javascript(self, request):
115
        """Displays the i18n JavaScript that the Django admin
116
        requires.
117
118
        This takes into account the ``USE_I18N`` setting. If it's set to False, the
119
        generated JavaScript will be leaner and faster.
120
        """
121
        return javascript_catalog(request, packages='pages')
122
123
    def save_model(self, request, page, form, change):
124
        """Move the page in the tree if necessary and save every
125
        placeholder :class:`Content <pages.models.Content>`.
126
        """
127
        language = form.cleaned_data['language']
128
        target = form.data.get('target', None)
129
        position = form.data.get('position', None)
130
        page.save()
131
132
        # if True, we need to move the page
133
        if target and position:
134
            try:
135
                target = self.model.objects.get(pk=target)
136
            except self.model.DoesNotExist:
137
                pass
138
            else:
139
                target.invalidate()
140
                page.move_to(target, position)
141
142
        for name in self.mandatory_placeholders:
143
            data = form.cleaned_data[name]
144
            placeholder = PlaceholderNode(name)
145
            extra_data = placeholder.get_extra_data(form.data)
146
            placeholder.save(page, language, data, change,
147
                extra_data=extra_data)
148
149
        for placeholder in get_placeholders(page.get_template()):
150
            if(placeholder.ctype in form.cleaned_data and placeholder.ctype
151
                    not in self.mandatory_placeholders):
152
                data = form.cleaned_data[placeholder.ctype]
153
                extra_data = placeholder.get_extra_data(form.data)
154
                placeholder.save(page, language, data, change,
155
                    extra_data=extra_data)
156
157
        page.invalidate()
158
159
    def get_fieldsets(self, request, obj=None):
160
        """
161
        Add fieldsets of placeholders to the list of already
162
        existing fieldsets.
163
        """
164
165
        # some ugly business to remove freeze_date
166
        # from the field list
167
        general_module = {
168
            'fields': list(self.general_fields),
169
            'classes': ('module-general',),
170
        }
171
172
        default_fieldsets = list(self.fieldsets)
173
        if not request.user.has_perm('pages.can_freeze'):
174
            general_module['fields'].remove('freeze_date')
175
        if not request.user.has_perm('pages.can_publish'):
176
            general_module['fields'].remove('status')
177
178
        default_fieldsets[0][1] = general_module
179
180
        placeholder_fieldsets = []
181
        section_placeholder_fieldsets = defaultdict(list)
182
        template = get_template_from_request(request, obj)
183
        for placeholder in get_placeholders(template):
184
            if placeholder.ctype not in self.mandatory_placeholders and not placeholder.section:
185
                placeholder_fieldsets.append(placeholder.ctype)
186
            elif placeholder.section:
187
                section_placeholder_fieldsets[placeholder.section].append(placeholder.ctype)
188
189
        additional_fieldsets = []
190
        for title, fieldset in section_placeholder_fieldsets.items():
191
            additional_fieldsets.append((title, {
192
                'fields': fieldset,
193
                'classes': ('module-content collapse grp-collapse grp-closed',),
194
            }))
195
        additional_fieldsets.append((_('Content'), {
196
            'fields': placeholder_fieldsets,
197
            'classes': ('module-content',),
198
        }))
199
        return default_fieldsets + additional_fieldsets
200
201
    def save_form(self, request, form, change):
202
        """Given a ModelForm return an unsaved instance. ``change`` is True if
203
        the object is being changed, and False if it's being added."""
204
        instance = super(PageAdmin, self).save_form(request, form, change)
205
        instance.template = form.cleaned_data['template']
206
        if not change:
207
            instance.author = request.user
208
        return instance
209
210
    def get_form(self, request, obj=None, **kwargs):
211
        """Get a :class:`Page <pages.admin.forms.PageForm>` for the
212
        :class:`Page <pages.models.Page>` and modify its fields depending on
213
        the request."""
214
215
        template = get_template_from_request(request, obj)
216
217
        #model = create_page_model(get_placeholders(template))
218
219
        form = make_form(self.model, get_placeholders(template))
220
221
        # bound the form
222
        language = get_language_from_request(request)
223
        form.base_fields['language'].initial = language
224
        if obj:
225
            initial_slug = obj.slug(language=language, fallback=False)
226
            initial_title = obj.title(language=language, fallback=False)
227
            form.base_fields['slug'].initial = initial_slug
228
            form.base_fields['title'].initial = initial_title
229
230
        template = get_template_from_request(request, obj)
231
        page_templates = settings.get_page_templates()
232
        if len(page_templates) > 0:
233
            template_choices = list(page_templates)
234
            template_choices.insert(0, (settings.PAGE_DEFAULT_TEMPLATE,
235
                    _('Default template')))
236
            form.base_fields['template'].choices = template_choices
237
            form.base_fields['template'].initial = force_text(template)
238
239
        for placeholder in get_placeholders(template):
240
            ctype = placeholder.ctype
241
            if obj:
242
                initial = placeholder.get_content(obj, language, lang_fallback=False)
243
            else:
244
                initial = None
245
            form.base_fields[ctype] = placeholder.get_field(obj,
246
                language, initial=initial)
247
248
        return form
249
250
    def change_view(self, request, object_id, form_url='', extra_context=None):
251
        """The ``change`` admin view for the
252
        :class:`Page <pages.models.Page>`."""
253
        language = get_language_from_request(request)
254
        extra_context = {
255
            'language': language,
256
            'page_languages': settings.PAGE_LANGUAGES,
257
        }
258
        try:
259
            int(object_id)
260
        except ValueError:
261
            raise Http404('The "%s" part of the location is invalid.'
262
                % str(object_id))
263
        try:
264
            obj = self.model.objects.get(pk=object_id)
265
        except self.model.DoesNotExist:
266
            # Don't raise Http404 just yet, because we haven't checked
267
            # permissions yet. We don't want an unauthenticated user to be able
268
            # to determine whether a given object exists.
269
            obj = None
270
        else:
271
            template = get_template_from_request(request, obj)
272
            extra_context['placeholders'] = get_placeholders(template)
273
            extra_context['traduction_languages'] = [l for l in
274
                settings.PAGE_LANGUAGES if Content.objects.get_content(obj,
275
                                    l[0], "title") and l[0] != language]
276
        extra_context['page'] = obj
277
278
        response = super(PageAdmin, self).change_view(request, object_id,
279
            form_url=form_url, extra_context=extra_context)
280
        if request.method == 'POST' and isinstance(response, HttpResponseRedirect):
281
            if '_continue' in request.POST or '_saveasnew' in request.POST or '_addanother' in request.POST:
282
                addlanguage = True
283
            else:
284
                addlanguage = False
285
            if addlanguage:
286
                from six.moves import urllib
287
                splitted = list(urllib.parse.urlparse(response.url))
288
                query = urllib.parse.parse_qs(splitted[4])
289
                query['language'] = language
290
                splitted[4] = urllib.parse.urlencode(query)
291
                response = HttpResponseRedirect(urllib.parse.urlunparse(splitted))
292
        return response
293
294
    def add_view(self, request, form_url='', extra_context=None):
295
        """The ``add`` admin view for the :class:`Page <pages.models.Page>`."""
296
        extra_context = {
297
            'language': get_language_from_request(request),
298
            'page_languages': settings.PAGE_LANGUAGES,
299
        }
300
        return super(PageAdmin, self).add_view(request, form_url,
301
                                               extra_context)
302
303
    def has_add_permission(self, request):
304
        """Return ``True`` if the current user has permission to add a new
305
        page."""
306
        return request.user.has_perm('pages.add_page')
307
308
    def has_change_permission(self, request, obj=None):
309
        """Return ``True`` if the current user has permission
310
        to change the page."""
311
        return request.user.has_perm('pages.change_page')
312
313
    def has_delete_permission(self, request, obj=None):
314
        """Return ``True`` if the current user has permission on the page."""
315
        return request.user.has_perm('pages.delete_page')
316
317
    def list_pages(self, request, template_name=None, extra_context=None):
318
        """List root pages"""
319
        if not self.admin_site.has_permission(request):
320
            return self.admin_site.login(request)
321
        language = get_language_from_request(request)
322
323
        query = request.POST.get('q', '').strip()
324
325
        if query:
326
            page_ids = list(set([c.page.pk for c in
327
                Content.objects.filter(body__icontains=query)]))
328
            pages = Page.objects.filter(pk__in=page_ids)
329
        else:
330
            pages = Page.objects.root()
331
        if settings.PAGE_HIDE_SITES:
332
            pages = pages.filter(sites=global_settings.SITE_ID)
333
334
        context = {
335
            'can_publish': request.user.has_perm('pages.can_publish'),
336
            'can_import': settings.PAGE_IMPORT_ENABLED,
337
            'lang': language,  # lang is the key used by show_content
338
            'pages': pages,
339
            'opts': self.model._meta,
340
            'q': query
341
        }
342
343
        context.update(extra_context or {})
344
        change_list = self.changelist_view(request, context)
345
346
        return change_list
347
348
349
try:
350
    admin.site.register(Page, PageAdmin)
351
except AlreadyRegistered:
352
    pass
353
354
355
class ContentAdmin(admin.ModelAdmin):
356
    list_display = ('__unicode__', 'type', 'language', 'page')
357
    list_filter = ('page',)
358
    search_fields = ('body',)
359
360
#admin.site.register(Content, ContentAdmin)
361
362
363
class AliasAdmin(admin.ModelAdmin):
364
    list_display = ('page', 'url',)
365
    list_editable = ('url',)
366
367
try:
368
    admin.site.register(PageAlias, AliasAdmin)
369
except AlreadyRegistered:
370
    pass
371