add Vendor model to select flavor/image at Tenant creation
Change-Id: I48343e55d24d5d298174f96886674cd75553fc41
diff --git a/xos/models.py b/xos/models.py
index d4d9485..0414d9f 100644
--- a/xos/models.py
+++ b/xos/models.py
@@ -1,5 +1,6 @@
from core.models.plcorebase import *
from models_decl import VMMEService_decl
+from models_decl import VMMEVendor_decl
from models_decl import VMMETenant_decl
from django.db import models
@@ -25,6 +26,10 @@
t.save()
return t
+class VMMEVendor(VMMEVendor_decl):
+ class Meta:
+ proxy = True
+
class VMMETenant(VMMETenant_decl):
class Meta:
proxy = True
@@ -35,6 +40,17 @@
self._meta.get_field("provider_service").default = vmmeservices[0].id
super(VMMETenant, self).__init__(*args, **kwargs)
+ @property
+ def image(self):
+ if not self.vmme_vendor:
+ return super(VMMETenant, self).image
+ return self.vmme_vendor.image
+
+ def save_instance(self, instance):
+ if self.vmme_vendor:
+ instance.flavor = self.vmme_vendor.flavor
+ super(VMMETenant, self).save_instance(instance)
+
def save(self, *args, **kwargs):
if not self.creator:
if not getattr(self, "caller", None):