Sith/core/views/page.py

# -*- coding:utf-8 -*
#
# Copyright 2016,2017
# - Skia <skia@libskia.so>
#
# Ce fichier fait partie du site de l'Association des Étudiants de l'UTBM,
# http://ae.utbm.fr.
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License a published by the Free Software
# Foundation; either version 3 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Sofware Foundation, Inc., 59 Temple
# Place - Suite 330, Boston, MA 02111-1307, USA.
#
#

# This file contains all the views that concern the page model
from django.core.urlresolvers import reverse_lazy
from django.views.generic import ListView, DetailView
from django.views.generic.edit import UpdateView, CreateView, DeleteView
from django.forms.models import modelform_factory
from django.http import Http404

from core.models import Page, PageRev, LockError
from core.views.forms import MarkdownInput, PageForm, PagePropForm
from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, CanCreateMixin


class CanEditPagePropMixin(CanEditPropMixin):

    def dispatch(self, request, *args, **kwargs):
        res = super(CanEditPagePropMixin, self).dispatch(request, *args, **kwargs)
        if self.object.is_club_page:
            raise Http404
        return res


class PageListView(CanViewMixin, ListView):
    model = Page
    template_name = 'core/page_list.jinja'


class PageView(CanViewMixin, DetailView):
    model = Page
    template_name = 'core/page_detail.jinja'

    def get_object(self):
        self.page = Page.get_page_by_full_name(self.kwargs['page_name'])
        return self.page

    def get_context_data(self, **kwargs):
        context = super(PageView, self).get_context_data(**kwargs)
        if "page" not in context.keys():
            context['new_page'] = self.kwargs['page_name']
        return context


class PageHistView(CanViewMixin, DetailView):
    model = Page
    template_name = 'core/page_hist.jinja'

    def get_object(self):
        self.page = Page.get_page_by_full_name(self.kwargs['page_name'])
        return self.page


class PageRevView(CanViewMixin, DetailView):
    model = Page
    template_name = 'core/page_detail.jinja'

    def get_object(self):
        self.page = Page.get_page_by_full_name(self.kwargs['page_name'])
        return self.page

    def get_context_data(self, **kwargs):
        context = super(PageRevView, self).get_context_data(**kwargs)
        if self.page is not None:
            context['page'] = self.page
            try:
                rev = self.page.revisions.get(id=self.kwargs['rev'])
                context['rev'] = rev
            except:
                # By passing, the template will just display the normal page without taking revision into account
                pass
        else:
            context['new_page'] = self.kwargs['page_name']
        return context


class PageCreateView(CanCreateMixin, CreateView):
    model = Page
    form_class = PageForm
    template_name = 'core/page_prop.jinja'

    def get_initial(self):
        init = {}
        if 'page' in self.request.GET.keys():
            page_name = self.request.GET['page']
            parent_name = '/'.join(page_name.split('/')[:-1])
            parent = Page.get_page_by_full_name(parent_name)
            if parent is not None:
                init['parent'] = parent.id
            init['name'] = page_name.split('/')[-1]
        return init

    def get_context_data(self, **kwargs):
        context = super(PageCreateView, self).get_context_data(**kwargs)
        context['new_page'] = True
        return context

    def form_valid(self, form):
        form.instance.set_lock(self.request.user)
        ret = super(PageCreateView, self).form_valid(form)
        return ret


class PagePropView(CanEditPagePropMixin, UpdateView):
    model = Page
    form_class = PagePropForm
    template_name = 'core/page_prop.jinja'
    slug_field = '_full_name'
    slug_url_kwarg = 'page_name'

    def get_object(self):
        o = super(PagePropView, self).get_object()
        # Create the page if it does not exists
        # if p == None:
        #    parent_name = '/'.join(page_name.split('/')[:-1])
        #    name = page_name.split('/')[-1]
        #    if parent_name == "":
        #        p = Page(name=name)
        #    else:
        #        parent = Page.get_page_by_full_name(parent_name)
        #        p = Page(name=name, parent=parent)
        self.page = o
        try:
            self.page.set_lock_recursive(self.request.user)
        except LockError as e:
            raise e
        return self.page


class PageEditView(CanEditMixin, UpdateView):
    model = PageRev
    form_class = modelform_factory(model=PageRev, fields=['title', 'content', ], widgets={'content': MarkdownInput})
    template_name = 'core/pagerev_edit.jinja'

    def get_object(self):
        self.page = Page.get_page_by_full_name(self.kwargs['page_name'])
        if self.page is not None:
            # First edit
            if self.page.revisions.all() is None:
                rev = PageRev(author=request.user)
                rev.save()
                self.page.revisions.add(rev)
            try:
                self.page.set_lock(self.request.user)
            except LockError as e:
                raise e
            return self.page.revisions.last()
        return None

    def get_context_data(self, **kwargs):
        context = super(PageEditView, self).get_context_data(**kwargs)
        if self.page is not None:
            context['page'] = self.page
        else:
            context['new_page'] = self.kwargs['page_name']
        return context

    def form_valid(self, form):
        # TODO : factor that, but first make some tests
        rev = form.instance
        new_rev = PageRev(title=rev.title,
                          content=rev.content,
                          )
        new_rev.author = self.request.user
        new_rev.page = self.page
        form.instance = new_rev
        return super(PageEditView, self).form_valid(form)


class PageDeleteView(CanEditPagePropMixin, DeleteView):
    model = Page
    template_name = 'core/delete_confirm.jinja'
    pk_url_kwarg = 'page_id'

    def get_success_url(self, **kwargs):
        return reverse_lazy('core:page_list')