Completed
Push — master ( beb414...ca28b2 )
by Batiste
10s
created

PageAdmin.get_urls()   B

Complexity

Conditions 1

Size

Total Lines 27

Duplication

Lines 0
Ratio 0 %

Importance

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