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