dynamicweb/cmsplugin_filer_link/cms_plugins.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from django.templatetags.static import static
from django.utils.translation import gettext as _

from cms.plugin_base import CMSPluginBase
from cms.plugin_pool import plugin_pool

from .forms import FilerLinkForm
from .models import FilerLinkPlugin as FilerLinkPluginModel


class FilerLinkPlugin(CMSPluginBase):
    form = FilerLinkForm
    model = FilerLinkPluginModel
    module = 'Filer'
    name = _("Link")
    raw_id_fields = ('page_link', )
    render_template = "cmsplugin_filer_link/link.html"
    text_enabled = True

    fieldsets = (
        (None, {
            'fields': [
                'name',
                'url',
                'page_link',
                'mailto',
                'file',
                'link_style',
                'new_window',
            ]
        }),
        (_('Advanced'), {
            'classes': ['collapse', ],
            'fields': [
                'link_attributes',
            ]
        })
    )

    def render(self, context, instance, placeholder):
        context = super(FilerLinkPlugin, self).render(context, instance, placeholder)
        if instance.file:
            link = instance.file.url
        elif instance.mailto:
            link = "mailto:%s" % _(instance.mailto)
        elif instance.url:
            link = _(instance.url)
        elif instance.page_link:
            link = instance.page_link.get_absolute_url()
        else:
            link = ""
        context.update({
            'link': link,
            'style': instance.link_style,
            'name': instance.name,
            'new_window': instance.new_window,
        })
        return context

    def icon_src(self, instance):
        return static("cms/img/icons/plugins/link.png")


plugin_pool.register_plugin(FilerLinkPlugin)