diff --git a/LTA/LTAIngest/LTAIngestClient/lib/rpc.py b/LTA/LTAIngest/LTAIngestClient/lib/rpc.py index b37154e49910a8088b9fd6bcc54a7fd05e712787..80e498701eab86e8fc9c0e9c5404c79731782eee 100644 --- a/LTA/LTAIngest/LTAIngestClient/lib/rpc.py +++ b/LTA/LTAIngest/LTAIngestClient/lib/rpc.py @@ -20,15 +20,7 @@ class IngestRPC(RPCWrapper): return self.rpc('SetExportJobPriority', export_id=export_group_id, priority=priority) def getStatusReport(self): - report = self.rpc('GetStatusReport') - - for export_id, status_dict in list(report.items()): - if 'series' in status_dict: - for series_dict in list(status_dict['series'].values()): - if 'timestamps' in series_dict: - #convert qpid timestamp to python datetime - series_dict['timestamps'] = [t.datetime() for t in series_dict['timestamps']] - return report + return self.rpc('GetStatusReport') def getJobStatus(self, job_id): return self.rpc('GetJobStatus', job_id=job_id)[job_id] diff --git a/SAS/MoM/MoMQueryService/MoMQueryServiceClient/momqueryrpc.py b/SAS/MoM/MoMQueryService/MoMQueryServiceClient/momqueryrpc.py index 5fbb6eabc6e8ad7a55abac3a93a2756f51f69db5..bf674cf080ecddfa56ff3e62b248cc927165f583 100644 --- a/SAS/MoM/MoMQueryService/MoMQueryServiceClient/momqueryrpc.py +++ b/SAS/MoM/MoMQueryService/MoMQueryServiceClient/momqueryrpc.py @@ -237,9 +237,6 @@ class MoMQueryRPC(RPCWrapper): :rtype dict with all projects''' logger.info("Requesting all projects") projects = self.rpc('GetProjects') - for project in projects: - project['statustime'] = project['statustime'].datetime() - logger.info("Received %s projects", (len(projects))) return projects diff --git a/SAS/OTDB_Services/otdbrpc.py b/SAS/OTDB_Services/otdbrpc.py index 247cb8d52be6fa40aee5770bf08e8bde5ec91473..7cc1100d7fb4914b80b154c01dd0ef2b85d4e84a 100644 --- a/SAS/OTDB_Services/otdbrpc.py +++ b/SAS/OTDB_Services/otdbrpc.py @@ -60,8 +60,6 @@ class OTDBRPC(RPCWrapper): def taskGetTreeInfo(self, otdb_id): info = self.rpc('TaskGetTreeInfo', otdb_id=otdb_id) - for key in ['creationtime', 'modificationtime', 'starttime', 'stoptime']: - info[key] = info[key].datetime() return info def taskGetStatus(self, otdb_id): diff --git a/SAS/ResourceAssignment/ResourceAssignmentEditor/lib/changeshandler.py b/SAS/ResourceAssignment/ResourceAssignmentEditor/lib/changeshandler.py index 73b56809dae274699d2812a2a4e7f170a32c9982..a2f917bcb21088b7a2d3b81a895d63927eccf9a1 100644 --- a/SAS/ResourceAssignment/ResourceAssignmentEditor/lib/changeshandler.py +++ b/SAS/ResourceAssignment/ResourceAssignmentEditor/lib/changeshandler.py @@ -138,16 +138,12 @@ class ChangesHandler: def onTaskUpdated(self, updated_task): '''onTaskUpdated is called upon receiving a TaskUpdated message.''' - updated_task['starttime'] = updated_task['starttime'].datetime() - updated_task['endtime'] = updated_task['endtime'].datetime() task_change = {'changeType':CHANGE_UPDATE_TYPE, 'objectType':'task', 'value':updated_task} self._handleChange(task_change) def onTaskInserted(self, new_task): '''onTaskInserted is called upon receiving a TaskInserted message. :param new_task: dictionary with the inserted task''' - new_task['starttime'] = new_task['starttime'].datetime() - new_task['endtime'] = new_task['endtime'].datetime() updateTaskMomDetails(new_task, self._momqueryrpc) task_change = {'changeType':CHANGE_INSERT_TYPE, 'objectType':'task', 'value':new_task} self._handleChange(task_change) @@ -161,16 +157,12 @@ class ChangesHandler: def onResourceClaimUpdated(self, updated_claim): '''onResourceClaimUpdated is called upon receiving a ResourceClaimUpdated message. :param updated_claim: dictionary with the updated claim''' - updated_claim['starttime'] = updated_claim['starttime'].datetime() - updated_claim['endtime'] = updated_claim['endtime'].datetime() claim_change = {'changeType':CHANGE_UPDATE_TYPE, 'objectType':'resourceClaim', 'value':updated_claim} self._handleChange(claim_change) def onResourceClaimInserted(self, new_claim): '''onResourceClaimInserted is called upon receiving a ResourceClaimInserted message. :param new_claim: dictionary with the inserted claim''' - new_claim['starttime'] = new_claim['starttime'].datetime() - new_claim['endtime'] = new_claim['endtime'].datetime() claim_change = {'changeType':CHANGE_INSERT_TYPE, 'objectType':'resourceClaim', 'value':new_claim} self._handleChange(claim_change) diff --git a/SAS/ResourceAssignment/ResourceAssignmentService/service.py b/SAS/ResourceAssignment/ResourceAssignmentService/service.py index beaefef7cebf12336f783d46eaab34df1b380e57..600697f1f1fa4afbecdd50ee5c72472a02cc8c32 100644 --- a/SAS/ResourceAssignment/ResourceAssignmentService/service.py +++ b/SAS/ResourceAssignment/ResourceAssignmentService/service.py @@ -126,8 +126,8 @@ class RADBHandler(MessageHandlerInterface): def _getResourceClaims(self, **kwargs): return self.radb.getResourceClaims(claim_ids=kwargs.get('claim_ids'), - lower_bound=kwargs.get('lower_bound').datetime() if kwargs.get('lower_bound') else None, - upper_bound=kwargs.get('upper_bound').datetime() if kwargs.get('upper_bound') else None, + lower_bound=kwargs.get('lower_bound'), + upper_bound=kwargs.get('upper_bound'), resource_ids=kwargs.get('resource_ids'), task_ids=kwargs.get('task_ids'), status=kwargs.get('status'), @@ -142,9 +142,6 @@ class RADBHandler(MessageHandlerInterface): def _insertResourceClaims(self, **kwargs): logger.info('InsertResourceClaims: %s' % dict({k:v for k,v in list(kwargs.items()) if v != None})) claims = kwargs['claims'] - for claim in claims: - claim['starttime'] = claim['starttime'].datetime() - claim['endtime'] = claim['endtime'].datetime() ids = self.radb.insertResourceClaims(kwargs['task_id'], claims, @@ -156,8 +153,8 @@ class RADBHandler(MessageHandlerInterface): logger.info('InsertResourceClaim: %s' % dict({k:v for k,v in list(kwargs.items()) if v != None})) id = self.radb.insertResourceClaim(kwargs['resource_id'], kwargs['task_id'], - kwargs['starttime'].datetime(), - kwargs['endtime'].datetime(), + kwargs['starttime'], + kwargs['endtime'], kwargs.get('status_id', kwargs.get('status')), kwargs['claim_size'], kwargs['username'], @@ -178,8 +175,8 @@ class RADBHandler(MessageHandlerInterface): updated = self.radb.updateResourceClaim(id, resource_id=kwargs.get('resource_id'), task_id=kwargs.get('task_id'), - starttime=kwargs.get('starttime').datetime() if kwargs.get('starttime') else None, - endtime=kwargs.get('endtime').datetime() if kwargs.get('endtime') else None, + starttime=kwargs.get('starttime'), + endtime=kwargs.get('endtime'), status=kwargs.get('status_id', kwargs.get('status')), claim_size=kwargs.get('claim_size'), username=kwargs.get('username'), @@ -195,8 +192,8 @@ class RADBHandler(MessageHandlerInterface): where_resource_types=kwargs.get('where_resource_types'), resource_id=kwargs.get('resource_id'), task_id=kwargs.get('task_id'), - starttime=kwargs.get('starttime').datetime() if kwargs.get('starttime') else None, - endtime=kwargs.get('endtime').datetime() if kwargs.get('endtime') else None, + starttime=kwargs.get('starttime'), + endtime=kwargs.get('endtime'), status=kwargs.get('status_id', kwargs.get('status')), claim_size=kwargs.get('status'), username=kwargs.get('username'), @@ -213,8 +210,8 @@ class RADBHandler(MessageHandlerInterface): task_id = kwargs['task_id'] updated = self.radb.updateTaskAndResourceClaims(task_id, - starttime=kwargs.get('starttime').datetime() if kwargs.get('starttime') else None, - endtime=kwargs.get('endtime').datetime() if kwargs.get('endtime') else None, + starttime=kwargs.get('starttime'), + endtime=kwargs.get('endtime'), task_status=kwargs.get('task_status_id', kwargs.get('task_status')), claim_status=kwargs.get('claim_status_id', kwargs.get('claim_status')), username=kwargs.get('username'), @@ -227,8 +224,8 @@ class RADBHandler(MessageHandlerInterface): 'updated': updated} def _getResourceUsages(self, **kwargs): - usages = self.radb.getResourceUsages(lower_bound=kwargs.get('lower_bound').datetime() if kwargs.get('lower_bound') else None, - upper_bound=kwargs.get('upper_bound').datetime() if kwargs.get('upper_bound') else None, + usages = self.radb.getResourceUsages(lower_bound=kwargs.get('lower_bound'), + upper_bound=kwargs.get('upper_bound'), resource_ids=kwargs.get('resource_ids'), claim_statuses=kwargs.get('status')) return convertIntKeysToString(usages) @@ -270,8 +267,8 @@ class RADBHandler(MessageHandlerInterface): def _getTasks(self, **kwargs): logger.info('GetTasks: %s' % dict({k:v for k,v in list(kwargs.items()) if v != None})) - return self.radb.getTasks(lower_bound=kwargs.get('lower_bound').datetime() if kwargs.get('lower_bound') else None, - upper_bound=kwargs.get('upper_bound').datetime() if kwargs.get('upper_bound') else None, + return self.radb.getTasks(lower_bound=kwargs.get('lower_bound'), + upper_bound=kwargs.get('upper_bound'), task_ids=kwargs.get('task_ids'), task_status=kwargs.get('task_status'), task_type=kwargs.get('task_type'), @@ -349,15 +346,15 @@ class RADBHandler(MessageHandlerInterface): kwargs['otdb_id'], kwargs['task_status'], kwargs['task_type'], - kwargs.get('starttime').datetime() if kwargs.get('starttime') else None, - kwargs.get('endtime').datetime() if kwargs.get('endtime') else None, + kwargs.get('starttime'), + kwargs.get('endtime'), kwargs['content'], kwargs['cluster']) def _insertSpecification(self, **kwargs): logger.info('InsertSpecification: %s' % dict({k:v for k,v in list(kwargs.items()) if v != None})) - specification_id = self.radb.insertSpecification(kwargs.get('starttime').datetime() if kwargs.get('starttime') else None, - kwargs.get('endtime').datetime() if kwargs.get('endtime') else None, + specification_id = self.radb.insertSpecification(kwargs.get('starttime'), + kwargs.get('endtime'), kwargs['content'], kwargs['cluster']) return {'id':specification_id} @@ -372,8 +369,8 @@ class RADBHandler(MessageHandlerInterface): logger.info('UpdateSpecification: %s' % dict({k:v for k,v in list(kwargs.items()) if v != None})) id = kwargs['id'] updated = self.radb.updateSpecification(id, - starttime=kwargs['starttime'].datetime() if 'starttime' in kwargs else None, - endtime=kwargs['endtime'].datetime() if 'endtime' in kwargs else None, + starttime=kwargs['starttime'], + endtime=kwargs['endtime'], content=kwargs.get('content'), cluster=kwargs.get('cluster')) return {'id': id, 'updated': updated} @@ -393,16 +390,16 @@ class RADBHandler(MessageHandlerInterface): def _get_max_resource_usage_between(self, **kwargs): logger.info('get_max_resource_usage_between: %s' % kwargs) return self.radb.get_max_resource_usage_between(resource_id=kwargs.get('resource_id'), - lower_bound=kwargs['lower_bound'].datetime() if kwargs.get('lower_bound') else None, - upper_bound=kwargs['upper_bound'].datetime() if kwargs.get('upper_bound') else None, + lower_bound=kwargs['lower_bound'], + upper_bound=kwargs['upper_bound'], claim_status=kwargs.get('claim_status', 'claimed')) def _get_resource_claimable_capacity(self, **kwargs): logger.info('get_resource_claimable_capacity: %s' % kwargs) resource_claimable_capacity = self.radb.get_resource_claimable_capacity( resource_id=kwargs.get('resource_id'), - lower_bound=kwargs['lower_bound'].datetime() if kwargs.get('lower_bound') else None, - upper_bound=kwargs['upper_bound'].datetime() if kwargs.get('upper_bound') else None) + lower_bound=kwargs['lower_bound'], + upper_bound=kwargs['upper_bound']) return { 'resource_claimable_capacity': resource_claimable_capacity}