add hostname option to node selector
diff --git a/xos/tosca/nodeselect.py b/xos/tosca/nodeselect.py
index 8badb14..11c3c78 100644
--- a/xos/tosca/nodeselect.py
+++ b/xos/tosca/nodeselect.py
@@ -4,12 +4,17 @@
from core.models import Slice,Sliver,User,Flavor,Node,Image
class XOSNodeSelector(object):
- def __init__(self, user, mem_size=None, num_cpus=None, disk_size=None):
+ def __init__(self, user, mem_size=None, num_cpus=None, disk_size=None, hostname = None):
self.user = user
+ self.hostname = None
def get_allowed_nodes(self):
# TODO: logic to get nodes that the user can use
nodes = Node.objects.all()
+
+ if self.hostname:
+ nodes = nodes.filter(name = self.hostname)
+
return nodes
def get_nodes(self, quantity):