| 1 |
from Actor import *
|
| 2 |
from crab_util import *
|
| 3 |
import common
|
| 4 |
import traceback
|
| 5 |
from ProdCommon.Credential.CredentialAPI import CredentialAPI
|
| 6 |
|
| 7 |
|
| 8 |
class CredentialRenew(Actor):
|
| 9 |
|
| 10 |
def __init__(self, cfg_params):
|
| 11 |
self.cfg_params=cfg_params
|
| 12 |
self.credentialType = 'Proxy'
|
| 13 |
if common.scheduler.name().upper() in ['LSF', 'CAF']:
|
| 14 |
self.credentialType = 'Token'
|
| 15 |
|
| 16 |
# init client server params...
|
| 17 |
CliServerParams(self)
|
| 18 |
|
| 19 |
def run(self):
|
| 20 |
"""
|
| 21 |
"""
|
| 22 |
common.logger.debug(5, "CredentialRenew::run() called")
|
| 23 |
self.renewer()
|
| 24 |
common.logger.message("Credential successfully delegated to the server.\n")
|
| 25 |
return
|
| 26 |
|
| 27 |
def renewer(self):
|
| 28 |
"""
|
| 29 |
"""
|
| 30 |
myproxyserver = self.cfg_params.get('EDG.proxy_server', 'myproxy.cern.ch')
|
| 31 |
configAPI = {'credential' : self.credentialType, \
|
| 32 |
'myProxySvr' : myproxyserver,\
|
| 33 |
'serverDN' : self.server_dn,\
|
| 34 |
'shareDir' : common.work_space.shareDir() ,\
|
| 35 |
'userName' : UnixUserName(),\
|
| 36 |
'serverName' : self.server_name \
|
| 37 |
}
|
| 38 |
try:
|
| 39 |
CredAPI = CredentialAPI( configAPI )
|
| 40 |
except Exception, err :
|
| 41 |
common.logger.debug(3, "Configuring Credential API: " +str(traceback.format_exc()))
|
| 42 |
raise CrabException("ERROR: Unable to configure Credential Client API %s\n"%str(err))
|
| 43 |
if not CredAPI.checkCredential(Time=100) :
|
| 44 |
common.logger.message("Please renew your %s :\n"%self.credentialType)
|
| 45 |
try:
|
| 46 |
CredAPI.ManualRenewCredential()
|
| 47 |
except Exception, ex:
|
| 48 |
raise CrabException(str(ex))
|
| 49 |
try:
|
| 50 |
dict = CredAPI.registerCredential()
|
| 51 |
except Exception, err:
|
| 52 |
common.logger.debug(3, "Registering Credentials : " +str(traceback.format_exc()))
|
| 53 |
raise CrabException("ERROR: Unable to register %s delegating server: %s\n"%(self.credentialType,self.server_name ))
|
