From 60260ccb0811299616c08e37c50ab922767cabdb Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Wed, 27 Jun 2018 09:06:28 +0200
Subject: [PATCH] Attempt to fix flake8 error

---
 datacenterlight/cms_models.py | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/datacenterlight/cms_models.py b/datacenterlight/cms_models.py
index 8c31696f..62a7b312 100644
--- a/datacenterlight/cms_models.py
+++ b/datacenterlight/cms_models.py
@@ -3,6 +3,7 @@ from cms.extensions.extension_pool import extension_pool
 from cms.models.fields import PlaceholderField
 from cms.models.pluginmodel import CMSPlugin
 from django import forms
+from django.conf import settings
 from django.contrib.postgres.fields import ArrayField
 from django.contrib.sites.models import Site
 from django.db import models
@@ -300,15 +301,17 @@ class MultipleChoiceArrayField(ArrayField):
     Uses Django's Postgres ArrayField
     and a MultipleChoiceField for its formfield.
     """
-    VMTemplateChoices = list(
-        (
-            str(obj.opennebula_vm_template_id),
-            (obj.name + ' - ' + VMTemplate.IPV6.title()
-                if obj.vm_type == VMTemplate.IPV6 else obj.name
+    VMTemplateChoices = []
+    if settings.OPENNEBULA_DOMAIN != 'test_domain':
+        VMTemplateChoices = list(
+            (
+                str(obj.opennebula_vm_template_id),
+                (obj.name + ' - ' + VMTemplate.IPV6.title()
+                    if obj.vm_type == VMTemplate.IPV6 else obj.name
+                 )
              )
-         )
-        for obj in VMTemplate.objects.all()
-    )
+            for obj in VMTemplate.objects.all()
+        )
 
     def formfield(self, **kwargs):
         defaults = {