pandaserver/test/boostUser.py

0001 import sys
0002
0003 from pandacommon.pandalogger.PandaLogger import PandaLogger
0004 from pandacommon.pandautils.thread_utils import GenericThread
0005
0006 from pandaserver.config import panda_config
0007 from pandaserver.taskbuffer.Initializer import initializer
0008 from pandaserver.taskbuffer.TaskBuffer import taskBuffer
0009
0010 # initialize DB using dummy connection
0011 initializer.init()
0012
0013 # logger
0014 _logger = PandaLogger().getLogger("boostUser")
0015 _logger.debug("================= start ==================")
0016
0017 # instantiate TB
0018 requester_id = GenericThread().get_full_id(__name__, sys.modules[__name__].__file__)
0019 taskBuffer.init(panda_config.dbhost, panda_config.dbpasswd, nDBConnection=1, requester=requester_id)
0020
0021 user = sys.stdin.read()
0022 user = user[:-1]
0023
0024 sql = "UPDATE atlas_panda.%s set currentPriority=:prio where prodUserName=:uname and prodSourceLabel IN (:label1,:label2) and currentPriority<:prio"
0025 varMap = {}
0026 varMap[":prio"] = 4000
0027 varMap[":uname"] = user
0028 varMap[":label1"] = "user"
0029 varMap[":label2"] = "panda"
0030 for table in ("jobsactive4", "jobsdefined4"):
0031     _logger.debug((sql % table) + str(varMap))
0032     ret = taskBuffer.querySQLS(sql % table, varMap)
0033     _logger.debug(f"ret -> {str(ret)}")
0034
0035 _logger.debug("================= end ==================")