def create_prepare_resource_support(self, resource_id="", prepare_resource_type=None, origin_resource_type=None):
if not isinstance(resource_id, types.StringType):
raise Inconsistent("The parameter resource_id is not a single resource id string")
if not self.service_provider or not self._rr:
raise Inconsistent("This class is not initialized properly")
if prepare_resource_type is not None and prepare_resource_type not in getextends(OT.ResourcePrepareSupport):
raise BadRequest(
"The requested resource %s is not extended from %s" % (prepare_resource_type, OT.ResourcePrepareSupport)
)
resource_data = IonObject(prepare_resource_type)
# Check to make sure the extended resource decorator raise OriginResourceType matches the type of the resource type
origin_resource_decorator = resource_data.get_class_decorator_value("OriginResourceType")
if origin_resource_decorator is None and origin_resource_type is None:
raise NotFound("OriginResourceType decorator not found in object specification %s", prepare_resource_type)
origin_resource_type = origin_resource_type if origin_resource_type is not None else origin_resource_decorator
if origin_resource_type is None:
raise NotFound("OriginResourceType decorator not found in object specification %s", prepare_resource_type)
resource_object = None
if resource_id:
resource_object = self._rr.read(resource_id)
if origin_resource_type != resource_object.type_ and not issubtype(
resource_object.type_, origin_resource_type
):
raise Inconsistent(
"The OriginResourceType decorator of the requested resource %s(%s) does not match the type of the specified resource id(%s)."
% (prepare_resource_type, origin_resource_type, resource_object.type_)
)
resource_data._id = resource_object._id
else:
resource_object = IonObject(origin_resource_type)
resource_data.resource = resource_object
resource_data.resource_schema = get_object_schema(origin_resource_type)
for field in resource_data._schema:
deco_value = resource_data.get_decorator_value(field, "AssociatedResources")
assoc_dict = {}
if deco_value is not None:
if deco_value.find(",") == -1:
associated_resources = [deco_value]
else:
associated_resources = deco_value.split(",")
for res in associated_resources:
assoc = self.get_associated_resource_info(origin_resource_type, resource_id, res)
assoc_dict[assoc.key] = assoc
setattr(resource_data, field, assoc_dict)
continue
return resource_data
def get_associated_resource_info(self, origin_resource_type="", resource_id="", assoc_resource_type=None):
if not origin_resource_type:
raise Inconsistent("The parameter origin_resource_type is not set")
if not isinstance(resource_id, types.StringType):
raise Inconsistent("The parameter resource_id is not a single resource id string")
if not self.service_provider or not self._rr:
raise Inconsistent("This class is not initialized properly")
if assoc_resource_type is not None and assoc_resource_type not in getextends(OT.AssociatedResources):
raise BadRequest(
"The requested resource %s is not extended from %s" % (assoc_resource_type, OT.AssociatedResources)
)
resource_data = IonObject(assoc_resource_type)
log.debug("get_associated_resource_info: %s", assoc_resource_type)
for field in resource_data._schema:
deco_value = resource_data.get_decorator_value(field, "ResourceType")
if deco_value is not None:
# Set the type of the associated resource to be used as the key in dict of associations
setattr(resource_data, "resource_type", deco_value)
res_list, _ = self._rr.find_resources(restype=deco_value, id_only=False)
exclude_lcs_filter_value = resource_data.get_decorator_value(field, "ExcludeLifecycleStates")
if exclude_lcs_filter_value is not None and exclude_lcs_filter_value.find(",") > -1:
exclude_filter = exclude_lcs_filter_value.split(",")
res_list = [res for res in res_list if res.lcstate not in exclude_filter]
# Look for ResourceSPOFilter filter and filter above results to exclude other resources already assigned
# This filter MUST be a comma separated value of Subject, Predicate, Object
res_filter_value = resource_data.get_decorator_value(field, "ResourceSPOFilter")
if res_filter_value is not None and res_filter_value.find(",") > -1:
assoc_filter = res_filter_value.split(",")
res_associations = self._rr.find_associations(predicate=assoc_filter[1], id_only=False)
assoc_list = [a for a in res_associations if a.st == assoc_filter[0] and a.ot == assoc_filter[2]]
def resource_available(res):
# give me a list of associations relevent to the passed in resource
rel_assocs = [
a for a in assoc_list if res._id == (a.o if a.st == origin_resource_type else a.s)
]
# of those resources, give me the ids of resources whos types match what i am currently building for
assocs = [(a.s if a.st == origin_resource_type else a.o) for a in rel_assocs]
# return true if there are no associations for the resource in question, or it has an association to
# the current resource we are building for
return len(assocs) == 0 or resource_id in assocs
# Iterate over the list and remove any object which are assigned to other resources and not the target resource
final_list = []
final_list.extend([res for res in res_list if resource_available(res)])
setattr(resource_data, field, final_list)
else:
setattr(resource_data, field, res_list)
continue
deco_value = resource_data.get_decorator_value(field, "Association")
if deco_value is not None:
# If the association is related to an existing resource and this is a create then skip
if not resource_id and (
resource_data.is_decorator(field, "ResourceSubject")
or resource_data.is_decorator(field, "ResourceObject")
):
continue
resource_sub = resource_id if resource_data.is_decorator(field, "ResourceSubject") else None
resource_obj = resource_id if resource_data.is_decorator(field, "ResourceObject") else None
assoc_list = self._rr.find_associations(
subject=resource_sub, predicate=deco_value, object=resource_obj, id_only=False
)
subject_type = resource_data.get_decorator_value(field, "SubjectType")
if subject_type is not None:
assoc_list = [assoc for assoc in assoc_list if (assoc.st == subject_type)]
object_type = resource_data.get_decorator_value(field, "ObjectType")
if object_type is not None:
assoc_list = [assoc for assoc in assoc_list if (assoc.ot == object_type)]
setattr(resource_data, field, assoc_list)
continue
# set key if we have one
key = resource_data.get_class_decorator_value("Key")
if key is None:
resource_data.key = resource_data.resource_type
else:
resource_data.key = key
return resource_data
def create_extended_resource_container(
self,
extended_resource_type,
resource_id,
computed_resource_type=None,
ext_associations=None,
ext_exclude=None,
**kwargs
):
"""
Returns an extended resource container for a given resource_id.
"""
overall_start_time = time.time()
self.ctx = None # Clear the context in case this instance gets reused
if not isinstance(resource_id, types.StringType):
raise Inconsistent("The parameter resource_id is not a single resource id string")
if not self.service_provider or not self._rr:
raise Inconsistent("This class is not initialized properly")
if extended_resource_type not in getextends(OT.ResourceContainer):
raise BadRequest(
"The requested resource %s is not extended from %s" % (extended_resource_type, OT.ResourceContainer)
)
if computed_resource_type and computed_resource_type not in getextends(OT.BaseComputedAttributes):
raise BadRequest(
"The requested resource %s is not extended from %s"
% (computed_resource_type, OT.BaseComputedAttributes)
)
resource_object = self._rr.read(resource_id)
if not resource_object:
raise NotFound("The Resource %s does not exist" % resource_id)
res_container = IonObject(extended_resource_type)
# Check to make sure the extended resource decorator raise OriginResourceType matches the type of the resource type
originResourceType = res_container.get_class_decorator_value("OriginResourceType")
if originResourceType is None:
log.error(
"The requested extended resource %s does not contain an OriginResourceType decorator.",
extended_resource_type,
)
elif originResourceType != resource_object.type_ and not issubtype(resource_object.type_, originResourceType):
raise Inconsistent(
"The OriginResourceType decorator of the requested resource %s(%s) does not match the type of the specified resource id(%s)."
% (extended_resource_type, originResourceType, resource_object.type_)
)
res_container._id = resource_object._id
res_container.resource = resource_object
# Initialize context object field and load resource associations
self._prepare_context(resource_object._id)
# Fill lcstate related resource container fields
self.set_container_lcstate_info(res_container)
# Fill resource container info; currently only type_version
self.set_res_container_info(res_container)
# Fill resource container fields
self.set_container_field_values(res_container, ext_exclude, **kwargs)
# Fill computed attributes
self.set_computed_attributes(res_container, computed_resource_type, ext_exclude, **kwargs)
# Fill additional associations
self.set_extended_associations(res_container, ext_associations, ext_exclude)
res_container.ts_created = get_ion_ts()
overall_stop_time = time.time()
log.debug(
"Time to process extended resource container %s %f secs",
extended_resource_type,
overall_stop_time - overall_start_time,
)
# log.info("ResourceContainer: %s" % res_container)
return res_container
