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