models in deployment.py appear to have been moved to site.py

[plstackapi.git] / planetstack / core / models / serviceclass.py

diff --git a/planetstack/core/models/serviceclass.py b/planetstack/core/models/serviceclass.py
index 1f9fc58..3b6ee82 100644 (file)
--- a/planetstack/core/models/serviceclass.py
+++ b/planetstack/core/models/serviceclass.py
@@ -2,7 +2,7 @@ import os
 from django.db import models
 from core.models import PlCoreBase
 from core.models import Site
-from core.models import DeploymentNetwork
+from core.models import Deployment
 
 # Create your models here.
 
@@ -20,3 +20,13 @@ class ServiceClass(PlCoreBase):
 
     def __unicode__(self):  return u'%s' % (self.name)
 
+    @staticmethod
+    def get_default():
+        try:
+            return ServiceClass.objects.get(name="Best Effort")
+        except ServiceClass.DoesNotExist:
+            return None
+
+    def save_by_user(self, user, *args, **kwds):
+        if self.can_update(user):
+            super(ServiceClass, self).save(*args, **kwds)