X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fcore%2Fmodels%2Fnode.py;h=94f93a53341ffe913e9a3343ac2db3fa4d4b0f84;hb=451c4fc73c5fbb698af6e67e9488c1e7d167d209;hp=9271268e33b5d1a98a37b7e6724bf549b0f96d54;hpb=4df786ffd7b10707a21667dfb4494b1d05ff4eae;p=plstackapi.git

diff --git a/planetstack/core/models/node.py b/planetstack/core/models/node.py
index 9271268..94f93a5 100644
--- a/planetstack/core/models/node.py
+++ b/planetstack/core/models/node.py
@@ -1,7 +1,7 @@
 import os
 from django.db import models
 from core.models import PlCoreBase
-from core.models import Site,Deployment
+from core.models import Site, SiteDeployment
 from core.models import Tag
 from django.contrib.contenttypes import generic
 
@@ -9,9 +9,14 @@ from django.contrib.contenttypes import generic
 
 class Node(PlCoreBase):
     name = models.CharField(max_length=200, unique=True, help_text="Name of the Node")
-    site  = models.ForeignKey(Site, related_name='nodes')
-    deployment = models.ForeignKey(Deployment, related_name='nodes')
+    site_deployment = models.ForeignKey(SiteDeployment, related_name='nodes')
+    site = models.ForeignKey(Site, null=True, blank=True, related_name='nodes')
     tags = generic.GenericRelation(Tag)
 
     def __unicode__(self):  return u'%s' % (self.name)
 
+    def save(self, *args, **kwds):
+        if self.site is None and self.site_deployment is not None:
+            self.site = self.site_deployment.site
+
+        super(Node, self).save(*args, **kwds)