pandaserver/test/setPriority.py

0001 import optparse
0002 import sys
0003
0004 # password
0005 from pandaserver.config import panda_config
0006 from pandaserver.taskbuffer.OraDBProxy import DBProxy
0007
0008 usage = """%prog <taskID> <priority>
0009
0010   Set a priority to jobs in a task"""
0011
0012 option_parser = optparse.OptionParser(usage=usage, conflict_handler="resolve")
0013 options, args = option_parser.parse_args()
0014
0015
0016 proxyS = DBProxy()
0017 proxyS.connect(panda_config.dbhost, panda_config.dbpasswd, panda_config.dbuser, panda_config.dbname)
0018
0019 varMap = {}
0020 varMap[":prodSourceLabel"] = "managed"
0021 varMap[":taskID"] = sys.argv[1]
0022 varMap[":prio"] = sys.argv[2]
0023 sql = "UPDATE %s SET currentPriority=:prio WHERE prodSourceLabel=:prodSourceLabel AND taskID=:taskID"
0024 for table in [
0025     "ATLAS_PANDA.jobsActive4",
0026     "ATLAS_PANDA.jobsDefined4",
0027 ]:
0028     status, res = proxyS.querySQLS(sql % table, varMap)