CRAB/python/CredentialRenew.py

from Actor import *
from crab_util import *
import common
import traceback
from ProdCommon.Credential.CredentialAPI import CredentialAPI
from SubmitterServer import SubmitterServer


class CredentialRenew(Actor):

    def __init__(self, cfg_params):
        self.cfg_params=cfg_params
        self.credentialType = 'Proxy'
        if common.scheduler.name().upper() in ['LSF', 'CAF']:
            self.credentialType = 'Token'
         
        # init client server params...
        CliServerParams(self)       

    def run(self):
        """
        """
        common.logger.debug(5, "CredentialRenew::run() called")
        ## TEMPORARY FIXME  
        if self.credentialType == 'Proxy':
            subServer = SubmitterServer(self.cfg_params, None, "all")
            subServer.moveProxy()
        else: 
            self.renewer()    
        common.logger.message("Credential successfully delegated to the server.\n")
        return

    def renewer(self):
        """
        """
        myproxyserver = self.cfg_params.get('EDG.proxy_server', 'myproxy.cern.ch')
        configAPI = {'credential' : self.credentialType, \
                     'myProxySvr' : myproxyserver,\
                     'serverDN'   : self.server_dn,\
                     'shareDir'   : common.work_space.shareDir() ,\
                     'userName'   : UnixUserName(),\
                     'serverName' : self.server_name \
                     }
        try:
            CredAPI =  CredentialAPI( configAPI )            
        except Exception, err : 
            common.logger.debug(3, "Configuring Credential API: " +str(traceback.format_exc()))
            raise CrabException("ERROR: Unable to configure Credential Client API  %s\n"%str(err))
        if not CredAPI.checkCredential(Time=100) :
           common.logger.message("Please renew your %s :\n"%self.credentialType)
           try:
               CredAPI.ManualRenewCredential()
           except Exception, ex:
               raise CrabException(str(ex))
        try:
            dict = CredAPI.registerCredential() 
        except Exception, err:
            common.logger.debug(3, "Registering Credentials : " +str(traceback.format_exc()))
            raise CrabException("ERROR: Unable to register %s delegating server: %s\n"%(self.credentialType,self.server_name ))
Revision:	1.3
Committed:	Mon Dec 8 21:47:47 2008 UTC (16 years, 4 months ago) by spiga
Content type:	text/x-python
Branch:	MAIN
CVS Tags:	CRAB_2_5_0_pre5, CRAB_2_5_0_pre4, CRAB_2_5_0_pre3, CRAB_2_5_0_pre2, CRAB_2_5_0_pre1, CRAB_2_4_4, CRAB_2_4_4_pre6, CRAB_2_4_4_pre5, CRAB_2_4_4_pre4, CRAB_2_4_4_pre3, CRAB_2_4_4_pre2, CRAB_2_4_4_pre1, CRAB_2_4_3, CRAB_2_4_3_pre8
Changes since 1.2:	+7 -1 lines
Log Message:	temporary patch for proxy renewal
#	User	Rev	Content
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			common.logger.debug(5, "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.1	common.logger.message("Credential successfully delegated to the server.\n")
31			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			'userName' : UnixUserName(),\
42			'serverName' : self.server_name \
43			}
44			try:
45			CredAPI = CredentialAPI( configAPI )
46			except Exception, err :
47			common.logger.debug(3, "Configuring Credential API: " +str(traceback.format_exc()))
48			raise CrabException("ERROR: Unable to configure Credential Client API %s\n"%str(err))
49			if not CredAPI.checkCredential(Time=100) :
50			common.logger.message("Please renew your %s :\n"%self.credentialType)
51			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			common.logger.debug(3, "Registering Credentials : " +str(traceback.format_exc()))
59			raise CrabException("ERROR: Unable to register %s delegating server: %s\n"%(self.credentialType,self.server_name ))