distcloud/dcorch/objects/resource.py

92 lines
3.2 KiB
Python

# Copyright (c) 2015 Ericsson AB.
# All Rights Reserved.
#
# 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.
"""Resource object."""
from dcorch.common import exceptions
from dcorch.db import api as db_api
from dcorch.objects import base
from oslo_versionedobjects import base as ovo_base
from oslo_versionedobjects import fields
@base.OrchestratorObjectRegistry.register
class Resource(base.OrchestratorObject, base.VersionedObjectDictCompat):
"""DC Orchestrator subcloud object."""
fields = {
'id': fields.IntegerField(),
'uuid': fields.UUIDField(),
'resource_type': fields.StringField(),
'master_id': fields.StringField(),
}
def create(self):
if self.obj_attr_is_set('id'):
raise exceptions.ObjectActionError(action='create',
reason='already created')
updates = self.obj_get_changes()
try:
resource_type = updates.pop('resource_type')
except KeyError:
raise exceptions.ObjectActionError(
action="create",
reason="cannot create a Resource object without a "
"resource_type set")
db_resource = db_api.resource_create(
self._context, resource_type, updates)
return self._from_db_object(self._context, self, db_resource)
@classmethod
def get_by_type_and_master_id(cls, context, resource_type, master_id):
db_resource = db_api.resource_get_by_type_and_master_id(
context, resource_type, master_id)
return cls._from_db_object(context, cls(), db_resource)
@classmethod
def get_by_id(cls, context, id):
db_resource = db_api.resource_get_by_id(context, id)
return cls._from_db_object(context, cls(), db_resource)
def delete(self):
db_api.resource_delete(
self._context, self.resource_type, self.master_id)
def save(self):
updates = self.obj_get_changes()
updates.pop('id', None)
updates.pop('uuid', None)
db_resource = db_api.resource_update(self._context, self.id, updates)
self._from_db_object(self._context, self, db_resource)
self.obj_reset_changes()
@base.OrchestratorObjectRegistry.register
class ResourceList(ovo_base.ObjectListBase, base.OrchestratorObject):
"""DC Orchestrator resource list object."""
VERSION = '1.1'
fields = {
'objects': fields.ListOfObjectsField('Resource'),
}
@classmethod
def get_all(cls, context, resource_type=None):
resources = db_api.resource_get_all(
context, resource_type)
return ovo_base.obj_make_list(
context, cls(context), Resource, resources)