From b872777bda964c36dd65073f03650db25e9a977c Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Wed, 13 Jun 2018 11:53:43 +0200
Subject: [PATCH] Filter context templates also by the ids that have been set
 for the calculator

---
 datacenterlight/cms_plugins.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/datacenterlight/cms_plugins.py b/datacenterlight/cms_plugins.py
index 769824e0..9c6279bc 100644
--- a/datacenterlight/cms_plugins.py
+++ b/datacenterlight/cms_plugins.py
@@ -88,9 +88,15 @@ class DCLCalculatorPlugin(CMSPluginBase):
         context = super(DCLCalculatorPlugin, self).render(
             context, instance, placeholder
         )
-        context['templates'] = VMTemplate.objects.filter(
-            vm_type=instance.vm_type
-        )
+        ids = instance.pricing.vm_templates_to_show
+        if ids:
+            context['templates'] = VMTemplate.objects.filter(
+                vm_type=instance.vm_type
+            ).filter(opennebula_vm_template_id__in=ids)
+        else:
+            context['templates'] = VMTemplate.objects.filter(
+                vm_type=instance.vm_type
+            )
         return context