118 lines
4.1 KiB
Python
118 lines
4.1 KiB
Python
# Copyright (c) 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.
|
|
|
|
import types
|
|
|
|
import murano.dsl.murano_object as murano_object
|
|
|
|
|
|
class ObjRef(object):
|
|
def __init__(self, obj):
|
|
self.ref_obj = obj
|
|
|
|
|
|
def _serialize_tree(root_object, designer_attributes):
|
|
serialized_objects = set()
|
|
tree = _pass1_serialize(
|
|
root_object, None, serialized_objects, designer_attributes)
|
|
_pass2_serialize(tree, serialized_objects)
|
|
return tree, serialized_objects
|
|
|
|
|
|
def serialize(root_object, executor):
|
|
if root_object is None:
|
|
tree = None
|
|
tree_copy = None
|
|
attributes = []
|
|
else:
|
|
tree, serialized_objects = _serialize_tree(
|
|
root_object, executor.object_store.designer_attributes)
|
|
tree_copy, _ = _serialize_tree(root_object, None)
|
|
attributes = executor.attribute_store.serialize(serialized_objects)
|
|
|
|
return {
|
|
'Objects': tree,
|
|
'ObjectsCopy': tree_copy,
|
|
'Attributes': attributes
|
|
}
|
|
|
|
|
|
def _cmp_objects(obj1, obj2):
|
|
if obj1 is None and obj2 is None:
|
|
return True
|
|
if obj1 is None or obj2 is None:
|
|
return False
|
|
return obj1.object_id == obj2.object_id
|
|
|
|
|
|
def _pass1_serialize(value, parent, serialized_objects,
|
|
designer_attributes_getter):
|
|
if isinstance(value, (types.StringTypes, types.IntType, types.FloatType,
|
|
types.BooleanType, types.NoneType)):
|
|
return value
|
|
elif isinstance(value, murano_object.MuranoObject):
|
|
if not _cmp_objects(value.parent, parent) \
|
|
or value.object_id in serialized_objects:
|
|
return ObjRef(value)
|
|
else:
|
|
result = value.to_dictionary()
|
|
if designer_attributes_getter is not None:
|
|
result['?'].update(designer_attributes_getter(value.object_id))
|
|
serialized_objects.add(value.object_id)
|
|
return _pass1_serialize(
|
|
result, value, serialized_objects, designer_attributes_getter)
|
|
|
|
elif isinstance(value, types.DictionaryType):
|
|
result = {}
|
|
for d_key, d_value in value.iteritems():
|
|
result_key = str(d_key)
|
|
result[result_key] = _pass1_serialize(
|
|
d_value, parent, serialized_objects,
|
|
designer_attributes_getter)
|
|
return result
|
|
elif isinstance(value, types.ListType):
|
|
return [_pass1_serialize(t, parent, serialized_objects,
|
|
designer_attributes_getter) for t in value]
|
|
elif isinstance(value, types.TupleType):
|
|
return _pass1_serialize(
|
|
list(value), parent, serialized_objects,
|
|
designer_attributes_getter)
|
|
else:
|
|
raise ValueError()
|
|
|
|
|
|
def _pass2_serialize(value, serialized_objects):
|
|
if isinstance(value, types.DictionaryType):
|
|
for d_key, d_value in value.iteritems():
|
|
if isinstance(d_value, ObjRef):
|
|
if d_value.ref_obj.object_id in serialized_objects:
|
|
value[d_key] = d_value.ref_obj.object_id
|
|
else:
|
|
value[d_key] = None
|
|
else:
|
|
_pass2_serialize(d_value, serialized_objects)
|
|
elif isinstance(value, types.ListType):
|
|
index = 0
|
|
while index < len(value):
|
|
item = value[index]
|
|
if isinstance(item, ObjRef):
|
|
if item.ref_obj.object_id in serialized_objects:
|
|
value[index] = item.ref_obj.object_id
|
|
else:
|
|
value.pop(index)
|
|
index -= 1
|
|
else:
|
|
_pass2_serialize(item, serialized_objects)
|
|
index += 1
|