84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
# Copyright 2014 Mirantis, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from operator import attrgetter
|
|
|
|
from fuelclient.objects.base import BaseObject
|
|
from fuelclient.objects import NodeCollection
|
|
|
|
|
|
class NodeGroup(BaseObject):
|
|
|
|
class_api_path = "nodegroups/"
|
|
instance_api_path = "nodegroups/{0}/"
|
|
|
|
@property
|
|
def env_id(self):
|
|
return self.get_fresh_data()["cluster_id"]
|
|
|
|
@property
|
|
def name(self):
|
|
return self.get_fresh_data()["name"]
|
|
|
|
@classmethod
|
|
def create(cls, name, cluster_id):
|
|
return cls.connection.post_request(
|
|
cls.class_api_path,
|
|
{'cluster_id': cluster_id, 'name': name},
|
|
)
|
|
|
|
@classmethod
|
|
def delete(cls, group_id):
|
|
return cls.connection.delete_request(
|
|
cls.instance_api_path.format(group_id)
|
|
)
|
|
|
|
def assign(self, nodes):
|
|
data = [{"id": n, "group_id": int(self.id)} for n in nodes]
|
|
NodeCollection.update(data)
|
|
|
|
|
|
class NodeGroupCollection(object):
|
|
|
|
def __init__(self, groups):
|
|
self.collection = groups
|
|
|
|
@classmethod
|
|
def init_with_ids(cls, ids):
|
|
return cls(map(NodeGroup, ids))
|
|
|
|
@classmethod
|
|
def init_with_data(cls, data):
|
|
return cls(map(NodeGroup.init_with_data, data))
|
|
|
|
def __str__(self):
|
|
return "node groups [{0}]".format(
|
|
", ".join(map(lambda n: str(n.id), self.collection))
|
|
)
|
|
|
|
def __iter__(self):
|
|
return iter(self.collection)
|
|
|
|
@property
|
|
def data(self):
|
|
return map(attrgetter("data"), self.collection)
|
|
|
|
@classmethod
|
|
def get_all(cls):
|
|
return cls(NodeGroup.get_all())
|
|
|
|
def filter_by_env_id(self, env_id):
|
|
predicate = lambda group: group.data['cluster_id'] == env_id
|
|
self.collection = filter(predicate, self.collection)
|