def delete(switchDict): ''' Given the switches, request a query 'delete' on the ResourceManagementDB that deletes from DowntimeCache all rows that match the parameters given. ''' rmsClient = ResourceManagementClient() result = { 'output': None, 'successful': None, 'message': None, 'match': None } output = rmsClient.deleteDowntimeCache( downtimeID=switchDict['downtimeID'], element=switchDict['element'], name=switchDict['name'], startDate=switchDict['startDate'], endDate=switchDict['endDate'], severity=switchDict['severity'], description=switchDict['description'], link=switchDict['link'] #dateEffective = switchDict[ 'dateEffective' ] ) result['match'] = int(output['Value']) result['successful'] = output['OK'] result['message'] = output['Message'] if 'Message' in output else None return result
def add(switchDict): ''' Given the switches, request a query 'addOrModify' on the ResourceManagementDB that inserts or updates-if-duplicated from DowntimeCache. ''' rmsClient = ResourceManagementClient() result = { 'output': None, 'successful': None, 'message': None, 'match': None } output = rmsClient.addOrModifyDowntimeCache( downtimeID=switchDict['downtimeID'], element=switchDict['element'], name=switchDict['name'], startDate=switchDict['startDate'], endDate=switchDict['endDate'], severity=switchDict['severity'], description=switchDict['description'], link=switchDict['link'] #dateEffective = switchDict[ 'dateEffective' ] ) result['match'] = int(output['Value']) result['successful'] = output['OK'] result['message'] = output['Message'] if 'Message' in output else None return result
def add( switchDict ): ''' Given the switches, request a query 'addOrModify' on the ResourceManagementDB that inserts or updates-if-duplicated from DowntimeCache. ''' rmsClient = ResourceManagementClient() result = { 'output': None, 'successful': None, 'message': None, 'match': None } output = rmsClient.addOrModifyDowntimeCache( downtimeID = switchDict[ 'downtimeID' ], element = switchDict[ 'element' ], name = switchDict[ 'name' ], startDate = switchDict[ 'startDate' ], endDate = switchDict[ 'endDate' ], severity = switchDict[ 'severity' ], description = switchDict[ 'description' ], link = switchDict[ 'link' ] #dateEffective = switchDict[ 'dateEffective' ] ) result['match'] = int( output['Value'] ) result['successful'] = output['OK'] result['message'] = output['Message'] if 'Message' in output else None return result
def select( switchDict ): ''' Given the switches, request a query 'select' on the ResourceManagementDB that gets from DowntimeCache all rows that match the parameters given. ''' rmsClient = ResourceManagementClient() meta = { 'columns' : [ 'downtimeID', 'element', 'name', 'startDate', 'endDate', 'severity', 'description', 'link', 'dateEffective' ] } result = { 'output': None, 'successful': None, 'message': None, 'match': None } output = rmsClient.selectDowntimeCache( downtimeID = switchDict[ 'downtimeID' ], element = switchDict[ 'element' ], name = switchDict[ 'name' ], #startDate = switchDict[ 'startDate' ], #endDate = switchDict[ 'endDate' ], severity = switchDict[ 'severity' ], #description = switchDict[ 'description' ], #link = switchDict[ 'link' ], #dateEffective = switchDict[ 'dateEffective' ], meta = meta ) result['output'] = [ dict( zip( output[ 'Columns' ], dt ) ) for dt in output[ 'Value' ] ] if 'ongoing' in switchDict: result['output'] = filterOngoing( result['output'] ) else: result['output'] = filterDate( result['output'], switchDict[ 'startDate' ], switchDict[ 'endDate' ] ) result['output'] = filterDescription( result['output'], switchDict[ 'description' ] ) result['match'] = len( result['output'] ) result['successful'] = output['OK'] result['message'] = output['Message'] if 'Message' in output else None return result
def select(switchDict): ''' Given the switches, request a query 'select' on the ResourceManagementDB that gets from DowntimeCache all rows that match the parameters given. ''' rmsClient = ResourceManagementClient() meta = { 'columns': [ 'downtimeID', 'element', 'name', 'startDate', 'endDate', 'severity', 'description', 'link', 'dateEffective' ] } result = { 'output': None, 'successful': None, 'message': None, 'match': None } output = rmsClient.selectDowntimeCache( downtimeID=switchDict['downtimeID'], element=switchDict['element'], name=switchDict['name'], #startDate = switchDict[ 'startDate' ], #endDate = switchDict[ 'endDate' ], severity=switchDict['severity'], #description = switchDict[ 'description' ], #link = switchDict[ 'link' ], #dateEffective = switchDict[ 'dateEffective' ], meta=meta) result['output'] = [ dict(zip(output['Columns'], dt)) for dt in output['Value'] ] if 'ongoing' in switchDict: result['output'] = filterOngoing(result['output']) else: result['output'] = filterDate(result['output'], switchDict['startDate'], switchDict['endDate']) result['output'] = filterDescription(result['output'], switchDict['description']) result['match'] = len(result['output']) result['successful'] = output['OK'] result['message'] = output['Message'] if 'Message' in output else None return result
def setUp(self): import DIRAC.ResourceStatusSystem.Client.ResourceManagementClient as mockedModule _serviceIn = ResourceManagementDB.ResourceManagementDB() self.client = mockedModule.ResourceManagementClient( serviceIn=_serviceIn)
def __init__(self, rStatus=None, rManagement=None): # Warm up local CS CSHelpers.warmUp() if rStatus is None: self.rStatus = ResourceStatusClient.ResourceStatusClient() if rManagement is None: self.rManagement = ResourceManagementClient.ResourceManagementClient( ) self.rssConfig = RssConfiguration()
def delete( switchDict ): ''' Given the switches, request a query 'delete' on the ResourceManagementDB that deletes from DowntimeCache all rows that match the parameters given. ''' rmsClient = ResourceManagementClient() result = { 'output': None, 'successful': None, 'message': None, 'match': None } output = rmsClient.deleteDowntimeCache( downtimeID = switchDict[ 'downtimeID' ], element = switchDict[ 'element' ], name = switchDict[ 'name' ], startDate = switchDict[ 'startDate' ], endDate = switchDict[ 'endDate' ], severity = switchDict[ 'severity' ], description = switchDict[ 'description' ], link = switchDict[ 'link' ] #dateEffective = switchDict[ 'dateEffective' ] ) result['match'] = int( output['Value'] ) result['successful'] = output['OK'] result['message'] = output['Message'] if 'Message' in output else None return result