diff --git a/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/propagator.py b/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/propagator.py index 66afebe4a20d3cfc6d3b4735dbf554dd994e453f..8d6f8e6b6facf13cd13f0e12c77040bb53e5378c 100755 --- a/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/propagator.py +++ b/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/propagator.py @@ -143,7 +143,7 @@ class RAtoOTDBPropagator(): logger.error(e) self.doTaskConflict(otdb_id) - def parseStorageProperties(self, storage_claim): + def ParseStorageProperties(self, storage_claim): """input something like: {u'username':u'anonymous', u'status': u'allocated', u'resource_name': u'cep4storage', u'user_id': -1, u'resource_type_id': 5, u'task_id': 6349, @@ -201,7 +201,7 @@ class RAtoOTDBPropagator(): for claim in claims: logger.debug("Processing claim: %s" % claim) if claim['resource_type_name'] == 'storage': ## TODO we will need to check for different storage names/types in the future - info['storage'] = parseStorageProperties(claim) + info['storage'] = self.ParseStorageProperties(claim) info["starttime"] = task["starttime"] info["endtime"] = task["endtime"] info["status"] = task["status"] diff --git a/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/translator.py b/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/translator.py index 989c85241ea790cc0b90979457003d3419e12a6e..be12371ee9b2492eba63907019e4b1d91c13cad2 100755 --- a/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/translator.py +++ b/SAS/ResourceAssignment/RAtoOTDBTaskSpecificationPropagator/lib/translator.py @@ -182,6 +182,7 @@ class RAtoOTDBTranslator(): return result def CreateStorageKeys(self, otdb_id, storage_properties, project_name, io_type): + result = {} if 'nr_of_uv_files' in storage_properties: result.update(self.CreateCorrelated(otdb_id, storage_properties, project_name, io_type)) if 'nr_of_cs_files' in storage_properties: diff --git a/SAS/ResourceAssignment/ResourceAssignmentEstimator/resource_estimators/base_resource_estimator.py b/SAS/ResourceAssignment/ResourceAssignmentEstimator/resource_estimators/base_resource_estimator.py index 6701cae23448c2312011fc1821ce02f461f104be..adc887b7751550612dcd29fbeef1388afec08f10 100644 --- a/SAS/ResourceAssignment/ResourceAssignmentEstimator/resource_estimators/base_resource_estimator.py +++ b/SAS/ResourceAssignment/ResourceAssignmentEstimator/resource_estimators/base_resource_estimator.py @@ -97,7 +97,7 @@ class BaseResourceEstimator(object): for sap_data_type, sap_data_value in sap['properties'].items(): if sap_data_type in data_properties: # We found this SAP's nr_of_<data_type>_files output_files[data_type][sap_data_type] = sap_data_value # We only count the nr_of_files from this SAP - output_files['saps'].append({'sap_nr': sap_nr, 'properties': {sap_data_type:sap_data_value}}) + output_files['saps'].append({'sap_nr': sap_nr, 'properties': sap['properties']}) if sap_data_type == 'start_sb_nr': output_files[data_type]['start_sb_nr'] = sap_data_value return output_files