Add generic header plugin & model

datacenterlight/cms_plugins.py

@@ -1,3 +1,5 @@
+from django.core.urlresolvers import reverse
+
 from cms.plugin_base import CMSPluginBase
 from cms.plugin_pool import plugin_pool
 
@@ -6,7 +8,8 @@ from .cms_models import (
     DCLFooterPluginModel, DCLLinkPluginModel, DCLNavbarDropdownPluginModel,
     DCLSectionIconPluginModel, DCLSectionImagePluginModel,
     DCLSectionPluginModel, DCLNavbarPluginModel,
-    DCLSectionPromoPluginModel, DCLCalculatorPluginModel
+    DCLSectionPromoPluginModel, DCLCalculatorPluginModel,
+    GenericProductHeaderPluginModel
 )
 from .models import VMTemplate
 from datacenterlight.utils import clear_all_session_vars
@@ -178,3 +181,23 @@ class DCLFooterPlugin(CMSPluginBase):
     cache = False
     allow_children = True
     child_classes = ['DCLLinkPlugin']
+
+
+@plugin_pool.register_plugin
+class GenericProductHeaderPlugin(CMSPluginBase):
+    module = "Datacenterlight"
+    name = "Generic Product Header Plugin"
+    model = GenericProductHeaderPluginModel
+    render_template = "datacenterlight/cms/generic_product_header.html"
+    cache = False
+
+    def render(self, context, instance, placeholder):
+        product_url = None
+        if instance.generic_product:
+            product_url = reverse(
+                'show_product',
+                kwargs={'product_slug': instance.generic_product.product_slug}
+            )
+        context["product_url"] = product_url
+        context["instance"] = instance
+        return context