diff --git a/src/python/RucioUtils.py b/src/python/RucioUtils.py index 23527f3580..7e2ea03445 100644 --- a/src/python/RucioUtils.py +++ b/src/python/RucioUtils.py @@ -24,10 +24,8 @@ def getNativeRucioClient(config=None, logger=None): cl = logging.getLogger('charset_normalizer') cl.setLevel(logging.ERROR) - # Need to use config from `config.TaskWorker` object instead to switch to - # user cert in PreDag.py - rucio_cert = getattr(config.TaskWorker, "Rucio_cert", config.TaskWorker.cmscert) - rucio_key = getattr(config.TaskWorker, "Rucio_key", config.TaskWorker.cmskey) + rucio_cert = getattr(config.Services, "Rucio_cert", config.TaskWorker.cmscert) + rucio_key = getattr(config.Services, "Rucio_key", config.TaskWorker.cmskey) logger.debug("Using cert [%s]\n and key [%s] for rucio client.", rucio_cert, rucio_key) nativeClient = Client( rucio_host=config.Services.Rucio_host, diff --git a/src/python/TaskWorker/Actions/PreDAG.py b/src/python/TaskWorker/Actions/PreDAG.py index 3f7a9b5c11..4066559ebf 100644 --- a/src/python/TaskWorker/Actions/PreDAG.py +++ b/src/python/TaskWorker/Actions/PreDAG.py @@ -206,15 +206,14 @@ def executeInternal(self, *args): config.TaskWorker.cmskey = os.environ.get('X509_USER_PROXY') config.TaskWorker.envForCMSWEB = newX509env(X509_USER_CERT=config.TaskWorker.cmscert, X509_USER_KEY=config.TaskWorker.cmskey) - # also for talking with rucio - config.TaskWorker.Rucio_cert = os.environ.get('X509_USER_PROXY') - config.TaskWorker.Rucio_key = os.environ.get('X509_USER_PROXY') - # need to get username from classAd to setup for Rucio access + # and use user cert to talking with rucio task_ad = classad.parseOne(open(os.environ['_CONDOR_JOB_AD'])) username = task_ad['CRAB_UserHN'] config.Services.Rucio_account = username + config.Services.Rucio_cert = os.environ.get('X509_USER_PROXY') + config.Services.Rucio_key = os.environ.get('X509_USER_PROXY') # need the global black list config.TaskWorker.scratchDir = './scratchdir'