commit b2fde610167913a48d54d3bb8985541f566ee4ba
author Tony Mack <tmack@tux.cs.princeton.edu> Mon Dec 15 11:45:02 2014 -0500
committer Sapan Bhatia <sapanb@cs.princeton.edu> Fri Dec 19 14:01:43 2014 -0500
tree 6bbff38c42535a2ac5021f5273d78fd7e4607658
parent 2d73512ebc3f99283ca8bff13d9d3099ce072518

fix bug preventing deployment admin from updating deployment flavors on ssave. fix nameError bug

planetstack/core/admin.py

diff --git a/planetstack/core/admin.py b/planetstack/core/admin.py
index bc84554..7fa537a 100644
--- a/planetstack/core/admin.py
+++ b/planetstack/core/admin.py
@@ -604,7 +604,15 @@
         #    a better way...
 
         self.manipulate_m2m_objs(deployment, self.cleaned_data['sites'], deployment.sitedeployments.all(), SiteDeployments, "deployment", "site")
-        self.manipulate_m2m_objs(deployment, self.cleaned_data['images'], deployment.imagedeployments.all(), DeploymentImages, "deployment", "image")
+        self.manipulate_m2m_objs(deployment, self.cleaned_data['images'], deployment.imagedeployments.all(), ImageDeployments, "deployment", "image")
+        # manipulate_m2m_objs doesn't work for Flavor/Deployment relationship
+        # so well handle that manually here
+        for flavor in deployment.flavors.all():
+            if getattr(flavor, 'name') not in self.cleaned_data['flavors']:
+                flavor.delete()
+        for flavor in self.cleaned_data['flavors']:
+            if flavor not in deployment.flavors.all():
+                flavor.deployments.add(deployment)
 
       self.save_m2m()