cms_plugins.py

from __future__ import unicode_literals

from cms.plugin_base import CMSPluginBase
from cms.plugin_pool import plugin_pool
from django.utils.translation import ugettext as _
from django.conf import settings
from django.template.loader import select_template

from .models import FilerLinkPlugin


class FilerLinkPlugin(CMSPluginBase):
    module = 'Filer'
    model = FilerLinkPlugin
    name = _("Link")
    text_enabled = True
    raw_id_fields = ('page_link', )
    TEMPLATE_NAME = 'cmsplugin_filer_link/plugins/link/%s.html'
    render_template = 'cmsplugin_filer_link/link.html'

    def render(self, context, instance, placeholder):
        self.render_template = select_template((
            self.TEMPLATE_NAME % instance.link_style,
            'cmsplugin_filer_link/link.html',)
        )

        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 settings.STATIC_URL + "cms/images/plugins/link.png"


plugin_pool.register_plugin(FilerLinkPlugin)