hostProps[pName] = pValue
if not diracAdmin.csModifyHost(hostName, hostProps,
createIfNonExistant=True)['OK']:
errorList.append(("add host", "Cannot register host %s" % hostName))
exitCode = 255
else:
result = diracAdmin.csCommitChanges()
if not result['OK']:
errorList.append(("commit", result['Message']))
exitCode = 255
if exitCode == 0:
from DIRAC.FrameworkSystem.Client.ComponentMonitoringClient import ComponentMonitoringClient
cmc = ComponentMonitoringClient()
ret = cmc.hostExists(dict(HostName=hostName))
if not ret['OK']:
Script.gLogger.error(
'Cannot check if host is registered in ComponentMonitoring',
ret['Message'])
elif ret['Value']:
Script.gLogger.info('Host already registered in ComponentMonitoring')
else:
ret = cmc.addHost(dict(HostName=hostName, CPU='TO_COME'))
if not ret['OK']:
Script.gLogger.error('Failed to add Host to ComponentMonitoring',
ret['Message'])
for error in errorList:
Script.gLogger.error("%s: %s" % error)
def main():
global hostName
global hostDN
global hostProperties
Script.registerSwitch("H:", "HostName:", "Name of the Host (Mandatory)",
setHostName)
Script.registerSwitch("D:", "HostDN:",
"DN of the Host Certificate (Mandatory)", setHostDN)
Script.registerSwitch(
"P:", "Property:",
"Property to be added to the Host (Allow Multiple instances or None)",
addProperty)
# Registering arguments will automatically add their description to the help menu
Script.registerArgument([
"Property=<Value>: Other properties to be added to the Host like (Responsible=XXX)"
],
mandatory=False)
_, args = Script.parseCommandLine(ignoreErrors=True)
if hostName is None or hostDN is None:
Script.showHelp(exitCode=1)
from DIRAC.Interfaces.API.DiracAdmin import DiracAdmin
diracAdmin = DiracAdmin()
exitCode = 0
errorList = []
hostProps = {"DN": hostDN}
if hostProperties:
hostProps["Properties"] = ", ".join(hostProperties)
for prop in args:
pl = prop.split("=")
if len(pl) < 2:
errorList.append(
("in arguments",
"Property %s has to include a '=' to separate name from value"
% prop))
exitCode = 255
else:
pName = pl[0]
pValue = "=".join(pl[1:])
gLogger.info("Setting property %s to %s" % (pName, pValue))
hostProps[pName] = pValue
if not diracAdmin.csModifyHost(
hostName, hostProps, createIfNonExistant=True)["OK"]:
errorList.append(("add host", "Cannot register host %s" % hostName))
exitCode = 255
else:
result = diracAdmin.csCommitChanges()
if not result["OK"]:
errorList.append(("commit", result["Message"]))
exitCode = 255
if exitCode == 0:
from DIRAC.FrameworkSystem.Client.ComponentMonitoringClient import ComponentMonitoringClient
cmc = ComponentMonitoringClient()
ret = cmc.hostExists(dict(HostName=hostName))
if not ret["OK"]:
gLogger.error(
"Cannot check if host is registered in ComponentMonitoring",
ret["Message"])
elif ret["Value"]:
gLogger.info("Host already registered in ComponentMonitoring")
else:
ret = cmc.addHost(dict(HostName=hostName, CPU="TO_COME"))
if not ret["OK"]:
gLogger.error("Failed to add Host to ComponentMonitoring",
ret["Message"])
for error in errorList:
gLogger.error("%s: %s" % error)
DIRAC.exit(exitCode)
def main():
global hostName
global hostDN
global hostProperties
Script.registerSwitch('H:', 'HostName:', 'Name of the Host (Mandatory)',
setHostName)
Script.registerSwitch('D:', 'HostDN:',
'DN of the Host Certificate (Mandatory)', setHostDN)
Script.registerSwitch(
'P:', 'Property:',
'Property to be added to the Host (Allow Multiple instances or None)',
addProperty)
Script.parseCommandLine(ignoreErrors=True)
if hostName is None or hostDN is None:
Script.showHelp(exitCode=1)
args = Script.getPositionalArgs()
from DIRAC.Interfaces.API.DiracAdmin import DiracAdmin
diracAdmin = DiracAdmin()
exitCode = 0
errorList = []
hostProps = {'DN': hostDN}
if hostProperties:
hostProps['Properties'] = ', '.join(hostProperties)
for prop in args:
pl = prop.split("=")
if len(pl) < 2:
errorList.append(
("in arguments",
"Property %s has to include a '=' to separate name from value"
% prop))
exitCode = 255
else:
pName = pl[0]
pValue = "=".join(pl[1:])
Script.gLogger.info("Setting property %s to %s" % (pName, pValue))
hostProps[pName] = pValue
if not diracAdmin.csModifyHost(
hostName, hostProps, createIfNonExistant=True)['OK']:
errorList.append(("add host", "Cannot register host %s" % hostName))
exitCode = 255
else:
result = diracAdmin.csCommitChanges()
if not result['OK']:
errorList.append(("commit", result['Message']))
exitCode = 255
if exitCode == 0:
from DIRAC.FrameworkSystem.Client.ComponentMonitoringClient import ComponentMonitoringClient
cmc = ComponentMonitoringClient()
ret = cmc.hostExists(dict(HostName=hostName))
if not ret['OK']:
Script.gLogger.error(
'Cannot check if host is registered in ComponentMonitoring',
ret['Message'])
elif ret['Value']:
Script.gLogger.info(
'Host already registered in ComponentMonitoring')
else:
ret = cmc.addHost(dict(HostName=hostName, CPU='TO_COME'))
if not ret['OK']:
Script.gLogger.error(
'Failed to add Host to ComponentMonitoring',
ret['Message'])
for error in errorList:
Script.gLogger.error("%s: %s" % error)
DIRAC.exit(exitCode)
