public-health-ch/feedler/admin.py

# -*- coding: utf-8 -*-

from django.db import models
from django.contrib.auth.decorators import login_required
from django.utils.decorators import method_decorator
from django.conf.urls import url
from django.core.urlresolvers import reverse
from django.utils.functional import cached_property
from django.utils.translation import ugettext as _
from django.shortcuts import redirect

from wagtail.contrib.modeladmin.helpers import AdminURLHelper, ButtonHelper
from wagtail.contrib.modeladmin.options import ModelAdmin
from wagtail.contrib.modeladmin.views import IndexView
from wagtail.wagtailadmin import messages

from feedler.models import Entry
from feedler.refresh import refresh_streams
from feedler.models.admin import FeedlySettings

class RefreshButtonHelper(ButtonHelper):
    """
    This helper constructs a refresh button
    """
    button_classnames = ['icon', 'icon-download']
    def refresh_button(self, classnames_add=None, classnames_exclude=None):
        if classnames_add is None: classnames_add = []
        if classnames_exclude is None: classnames_exclude = []
        classnames = self.button_classnames + classnames_add
        cn = self.finalise_classname(classnames, classnames_exclude)
        text = _('Sync {}'.format(self.verbose_name_plural.title()))
        return {
            'url': self.url_helper.get_action_url('refresh', query_params=self.request.GET),
            'label': text, 'classname': cn, 'title': text,
        }

class RefreshAdminURLHelper(AdminURLHelper):
    """
    This helper constructs the different urls, to overwrite the default behaviour
    and append the filters to the action.
    """
    non_object_specific_actions = ('create', 'choose_parent', 'index', 'refresh')
    def get_action_url(self, action, *args, **kwargs):
        query_params = kwargs.pop('query_params', None)
        url_name = self.get_action_url_name(action)
        if action in self.non_object_specific_actions:
            url = reverse(url_name)
        else:
            url = reverse(url_name, args=args, kwargs=kwargs)
        if query_params:
            url += '?{params}'.format(params=query_params.urlencode())
        return url
    def get_action_url_pattern(self, action):
        if action in self.non_object_specific_actions:
            return self._get_action_url_pattern(action)
        return self._get_object_specific_action_url_pattern(action)

class RefreshView(IndexView):
    """
    A Class Based View which will handle the button click
    """
    # def export_csv(self):
    #     data = self.queryset.all()
    #     response = ...
    #     return response
    @method_decorator(login_required)
    def dispatch(self, request, *args, **kwargs):
        super().dispatch(request, *args, **kwargs)
        if not refresh_streams(FeedlySettings.for_site(request.site)):
            messages.error(
                request, _('Sorry, could not refresh streams. Please contact your administrator.'))
        return redirect('/admin/feedler/entry/')


class EntryModelAdminMixin(object):
    """
    A mixin to add to your model admin which hooks the different helpers, the view
    and register the new urls.
    """
    button_helper_class = RefreshButtonHelper
    url_helper_class = RefreshAdminURLHelper
    view_class = RefreshView

    def get_admin_urls_for_registration(self):
        urls = super().get_admin_urls_for_registration()
        urls += (
            url(
                self.url_helper.get_action_url_pattern('refresh'),
                self.refresh_view,
                name=self.url_helper.get_action_url_name('refresh')
            ),
        )
        return urls

    def refresh_view(self, request):
        kwargs = {'model_admin': self}
        view_class = self.view_class
        return view_class.as_view(**kwargs)(request)