res = sw.addTagAtSite(mytag, mysite)
if not res['OK']:
gLogger.error(res['Message'])
dexit(1)
else:
gLogger.notice("Added %s to %s" % (mytag, mysite))
#This is the most common Call from Clients
res = sw.getSitesForTag(mytag)
if not res['OK']:
gLogger.error(res['Message'])
else:
gLogger.notice("Sites: ", res['Value'])
#Get the tags with valid status
res = sw.getTagsWithStatus("New")
if not res['OK']:
gLogger.error(res['Message'])
else:
gLogger.notice("Found the tags:",res['Value'])
#Get tags with f****d up status
res = sw.getTagsWithStatus("NewSomething")
if not res['OK']:
gLogger.error(res['Message'])
else:
gLogger.notice("Found the tags:", res['Value'])
res = sw.updateStatus(mytag, mysite, "Probing")
if not res['OK']:
gLogger.error(res['Message'])
class SoftwareMonitorAgent(AgentModule):
""" This agent picks up "New" tags and submits jobs and those that
are OK will report back to the service directly. For now it enforces the
transition from New to Probing to Valid.
Also resets the tags that have been Probing for too long to New
"""
def initialize(self):
""" Initialize the agent.
"""
self.am_setOption("PollingTime", 86400) #Once a day is enough
self.swtc = SoftwareTagClient()
self.submitjobs = self.am_getOption('SubmitJobs', False)
if self.submitjobs:
self.log.info(
"Will submit probe jobs to validate the software tags")
else:
self.log.info("Will mark as Valid all 'New' tags directly.")
self.delay = self.am_getOption("Delay", 86400)
self.log.info(
"Will reset to 'New' the tasks that have been 'Probing' for %s seconds"
% self.delay)
self.script = self.am_getOption("SoftwareManagementScript",
"ProbeSoftwareArea.py")
self.am_setOption('shifterProxy', 'SoftwareManager')
#Needs to be able to submit job for that VO
return S_OK()
def execute(self):
""" Get all New tags, mark them as Installing. Old Installing tags are reset to New
"""
#### get site mask ###
diracAdmin = DiracAdmin()
res = diracAdmin.getSiteMask(printOutput=False)
if not res["OK"]:
self.log.error("error retrieving site mask: %s" %
str(res["Message"]))
site_mask = res["Value"]
res = self.swtc.getTagsWithStatus("New")
if not res['OK']:
return res
if not res['Value']:
self.log.info("No 'New' tags to consider")
for tag, ces in res['Value'].items():
for ce in ces:
res = getSiteForCEs([ce])
if not res["OK"]:
self.log.error("could not retrieve Site name for CE %s" %
ce)
sites = res["Value"].keys()
for site in sites:
if site not in site_mask:
self.log.info("CE/Site disabled %s" % site)
continue
# ignore this CE
res = self.swtc.updateCEStatus(tag, ce, 'Installing')
if not res['OK']:
self.log.error(res['Message'])
continue
res = None
if self.submitjobs:
res = self.submitProbeJobs(ce)
else:
res = self.swtc.updateCEStatus(tag, ce, 'Valid')
if not res['OK']:
self.log.error(res['Message'])
else:
self.log.info("Done with %s at %s" % (tag, ce))
##Also, reset to New tags that were in Probing for too long.
res = self.swtc.getTagsWithStatus("Installing", self.delay)
if not res['OK']:
self.log.error("Failed to get old 'Installing' tags")
else:
if not res['Value']:
self.log.info("No 'Installing' tags to reset")
for tag, ces in res['Value'].items():
for ce in ces:
res = self.swtc.updateCEStatus(tag, ce, 'New')
if not res['OK']:
self.log.error(res['Message'])
continue
return S_OK()
def submitProbeJobs(self, ce):
""" Submit some jobs to the CEs
"""
#need credentials, should be there since the initialize
from DIRAC.Interfaces.API.Dirac import Dirac
d = Dirac()
from DIRAC.Interfaces.API.Job import Job
from DIRAC.ConfigurationSystem.Client.Helpers.Operations import Operations
import os
ops = Operations("glast.org")
scriptname = ops.getValue("ResourceStatus/SofwareManagementScript",
self.script)
j = Job()
j.setDestinationCE(ce)
j.setCPUTime(1000)
j.setName("Probe %s" % ce)
j.setJobGroup("SoftwareProbe")
j.setExecutable("%s/GlastDIRAC/ResourceStatusSystem/Client/%s" %
(os.environ['DIRAC'], scriptname),
logFile='SoftwareProbe.log')
j.setOutputSandbox('*.log')
res = d.submit(j)
if not res['OK']:
return res
return S_OK()
class SoftwareMonitorAgent(AgentModule):
""" This agent picks up "New" tags and submits jobs and those that
are OK will report back to the service directly. For now it enforces the
transition from New to Probing to Valid.
Also resets the tags that have been Probing for too long to New
"""
def initialize(self):
""" Initialize the agent.
"""
self.am_setOption( "PollingTime", 86400 ) #Once a day is enough
self.swtc = SoftwareTagClient()
self.submitjobs = self.am_getOption( 'SubmitJobs', False )
if self.submitjobs:
self.log.info("Will submit probe jobs to validate the software tags")
else:
self.log.info("Will mark as Valid all 'New' tags directly.")
self.delay = self.am_getOption("Delay", 86400)
self.log.info("Will reset to 'New' the tasks that have been 'Probing' for %s seconds" % self.delay)
self.script = self.am_getOption("SoftwareManagementScript", "ProbeSoftwareArea.py")
self.am_setOption( 'shifterProxy', 'SoftwareManager' )
#Needs to be able to submit job for that VO
return S_OK()
def execute(self):
""" Get all New tags, mark them as Installing. Old Installing tags are reset to New
"""
res = self.swtc.getTagsWithStatus("New")
if not res['OK']:
return res
if not res['Value']:
self.log.info("No 'New' tags to consider")
for tag, ces in res['Value'].items():
for ce in ces:
res = self.swtc.updateCEStatus(tag, ce, 'Installing')
if not res['OK']:
self.log.error(res['Message'])
continue
res = None
if self.submitjobs:
res = self.submitProbeJobs(ce)
else:
res = self.swtc.updateCEStatus(tag, ce, 'Valid')
if not res['OK']:
self.log.error(res['Message'])
else:
self.log.info("Took care of %s at %s" %(tag, ce))
##Also, reset to New tags that were in Probing for too long.
res = self.swtc.getTagsWithStatus("Installing",olderthan=self.delay)
if not res['OK']:
self.log.error("Failed to get old 'Installing' tags")
else:
if not res['Value']:
self.log.info("No 'Installing' tags to reset")
for tag, ces in res['Value'].items():
for ce in ces:
res = self.swtc.updateCEStatus(tag, ce, 'New')
if not res['OK']:
self.log.error(res['Message'])
continue
return S_OK()
def submitProbeJobs(self, ce):
""" Submit some jobs to the CEs
"""
#need credentials, should be there since the initialize
from DIRAC.Interfaces.API.Dirac import Dirac
d = Dirac()
from DIRAC.Interfaces.API.Job import Job
from DIRAC.ConfigurationSystem.Client.Helpers.Operations import Operations
import DIRAC
ops = Operations()
scriptname = ops.getValue("ResourceStatus/SofwareManagementScript", self.script)
j = Job()
j.setDestinationCE(ce)
j.setCPUTime(1000)
j.setName("Probe %s" % ce)
j.setJobGroup("SoftwareProbe")
j.setExecutable("%s/GlastDIRAC/ResourceStatusSystem/Client/%s" % (DIRAC.rootPath, scriptname),
logFile='SoftwareProbe.log')
j.setOutputSandbox('*.log')
res = d.submit(j)
if not res['OK']:
return res
return S_OK()
res = sw.addTagAtSite(mytag, mysite)
if not res['OK']:
gLogger.error(res['Message'])
dexit(1)
else:
gLogger.notice("Added %s to %s" % (mytag, mysite))
#This is the most common Call from Clients
res = sw.getSitesForTag(mytag)
if not res['OK']:
gLogger.error(res['Message'])
else:
gLogger.notice("Sites: ", res['Value'])
#Get the tags with valid status
res = sw.getTagsWithStatus("New")
if not res['OK']:
gLogger.error(res['Message'])
else:
gLogger.notice("Found the tags:", res['Value'])
#Get tags with f****d up status
res = sw.getTagsWithStatus("NewSomething")
if not res['OK']:
gLogger.error(res['Message'])
else:
gLogger.notice("Found the tags:", res['Value'])
res = sw.updateStatus(mytag, mysite, "Probing")
if not res['OK']:
gLogger.error(res['Message'])