90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
# Copyright 2016 Tesora 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 django.core import cache
|
|
|
|
|
|
def get(cluster_id):
|
|
if not has_cluster(cluster_id):
|
|
manager = ClusterInstanceManager(cluster_id)
|
|
cache.cache.set(cluster_id, manager)
|
|
|
|
return cache.cache.get(cluster_id)
|
|
|
|
|
|
def delete(cluster_id):
|
|
manager = get(cluster_id)
|
|
manager.clear_instances()
|
|
cache.cache.delete(cluster_id)
|
|
|
|
|
|
def update(cluster_id, manager):
|
|
cache.cache.set(cluster_id, manager)
|
|
|
|
|
|
def has_cluster(cluster_id):
|
|
if cache.cache.get(cluster_id):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
class ClusterInstanceManager(object):
|
|
def __init__(self, cluster_id):
|
|
self.cluster_id = cluster_id
|
|
self.instances = []
|
|
|
|
def get_instances(self):
|
|
if not hasattr(self, 'instances'):
|
|
self.instances = []
|
|
return self.instances
|
|
|
|
def get_instance(self, id):
|
|
for instance in self.get_instances():
|
|
if instance.id == id:
|
|
return instance
|
|
return None
|
|
|
|
def add_instance(self, id, name, flavor_id,
|
|
flavor, volume, type, related_to, nics):
|
|
instance = ClusterInstance(id, name, flavor_id, flavor,
|
|
volume, type, related_to, nics)
|
|
self.instances.append(instance)
|
|
update(self.cluster_id, self)
|
|
return self.get_instances()
|
|
|
|
def delete_instance(self, id):
|
|
instance = self.get_instance(id)
|
|
if instance:
|
|
self.instances.remove(instance)
|
|
update(self.cluster_id, self)
|
|
|
|
def clear_instances(self):
|
|
self.instances = []
|
|
update(self.cluster_id, self)
|
|
|
|
|
|
class ClusterInstance(object):
|
|
def __init__(self, id, name, flavor_id, flavor, volume, type,
|
|
related_to, nics):
|
|
self.id = id
|
|
self.name = name
|
|
self.flavor_id = flavor_id
|
|
self.flavor = flavor
|
|
self.volume = volume
|
|
self.type = type
|
|
self.related_to = related_to
|
|
self.nics = nics
|