def main():
global ceName
global Queue
global Site
from DIRAC import gLogger, exit as DIRACExit
from DIRAC.ConfigurationSystem.Client.Helpers import Resources
def setCEName(args):
global ceName
ceName = args
def setSite(args):
global Site
Site = args
def setQueue(args):
global Queue
Queue = args
Script.registerSwitch("N:", "Name=", "Computing Element Name (Mandatory)",
setCEName)
Script.registerSwitch("S:", "Site=", "Site Name (Mandatory)", setSite)
Script.registerSwitch("Q:", "Queue=", "Queue Name (Mandatory)", setQueue)
Script.parseCommandLine(ignoreErrors=True)
result = Resources.getQueue(Site, ceName, Queue)
if not result["OK"]:
# Normal DIRAC queue search failed, check for matching VM images
vmresult = Resources.getVMTypeConfig(Site, ceName, Queue)
if vmresult["OK"]:
gLogger.notice(json.dumps(vmresult["Value"]))
return
# Queue & VM not found, return original queue failure message
gLogger.error("Could not retrieve resource parameters",
": " + vmresult["Message"])
DIRACExit(1)
gLogger.notice(json.dumps(result["Value"]))
