add Vendor model to select flavor/image at Tenant creation
Change-Id: I237cd7cf29ced54d0e211bf8ff6d26c9ab3fb95c
diff --git a/xos/models.py b/xos/models.py
index 07c3404..ca470d2 100644
--- a/xos/models.py
+++ b/xos/models.py
@@ -1,5 +1,6 @@
from core.models.plcorebase import *
from models_decl import VPGWCService_decl
+from models_decl import VPGWCVendor_decl
from models_decl import VPGWCTenant_decl
from django.db import models
@@ -26,6 +27,10 @@
t.save()
return t
+class VPGWCVendor(VPGWCVendor_decl):
+ class Meta:
+ proxy = True
+
class VPGWCTenant(VPGWCTenant_decl):
class Meta:
proxy = True
@@ -39,6 +44,17 @@
"provider_service").default = vpgwcservice[0].id
super(VPGWCTenant, self).__init__(*args, **kwargs)
+ @property
+ def image(self):
+ if not self.vpgwc_vendor:
+ return super(VPGWCTenant, self).image
+ return self.vpgwc_vendor.image
+
+ def save_instance(self, instance):
+ if self.vpgwc_vendor:
+ instance.flavor = self.vpgwc_vendor.flavor
+ super(VPGWCTenant, self).save_instance(instance)
+
def save(self, *args, **kwargs):
if not self.creator:
if not getattr(self, "caller", None):