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("CredentialRenew::run() called")
        ## TEMPORARY FIXME  
        if self.credentialType == 'Proxy':
            subServer = SubmitterServer(self.cfg_params, None, "all")
            subServer.moveProxy()
        else: 
            self.renewer()    
        common.logger.info("Credential successfully delegated to the server.\n")
        return

    def renewer(self):
        """
        """
        myproxyserver = self.cfg_params.get('GRID.proxy_server', 'myproxy.cern.ch')
        configAPI = {'credential' : self.credentialType, \
                     'myProxySvr' : myproxyserver,\
                     'serverDN'   : self.server_dn,\
                     'shareDir'   : common.work_space.shareDir() ,\
                     'userName'   : getUserName(),\
                     'serverName' : self.server_name, \
                     'logger' : common.logger() \
                     }
        try:
            CredAPI =  CredentialAPI( configAPI )            
        except Exception, err : 
            common.logger.debug( "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.info("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( "Registering Credentials : " +str(traceback.format_exc()))
            raise CrabException("ERROR: Unable to register %s delegating server: %s\n"%(self.credentialType,self.server_name ))
Revision:	1.7
Committed:	Wed Jul 1 16:07:07 2009 UTC (15 years, 9 months ago) by spiga
Content type:	text/x-python
Branch:	MAIN
CVS Tags:	CRAB_2_6_6_pre6, CRAB_2_6_6_pre5, CRAB_2_6_6_pre4, CRAB_2_6_6_pre3, CRAB_2_6_6_pre2, CRAB_2_6_6_check, CRAB_2_6_6, CRAB_2_6_6_pre1, CRAB_2_6_5, CRAB_2_6_5_pre1, CRAB_2_6_4, CRAB_2_6_4_pre1, CRAB_2_6_3_patch_2, CRAB_2_6_3_patch_2_pre2, CRAB_2_6_3_patch_2_pre1, CRAB_2_6_3_patch_1, CRAB_2_6_3, CRAB_2_6_3_pre5, CRAB_2_6_3_pre4, CRAB_2_6_3_pre3, CRAB_2_6_3_pre2, CRAB_2_7_0_pre2, CRAB_2_6_3_pre1, test_1, CRAB_2_7_0_pre1, CRAB_2_6_2, CRAB_2_6_2_pre2, CRAB_2_6_2_pre1, CRAB_2_6_1_pre4, CRAB_2_6_1_pre3, CRAB_2_6_1_pre2, CRAB_2_6_1_pre1, CRAB_2_6_1
Branch point for:	CRAB_2_6_X_br
Changes since 1.6:	+2 -1 lines
Log Message:	pass logger to credential API
#	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	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	spiga	1.6	myproxyserver = self.cfg_params.get('GRID.proxy_server', 'myproxy.cern.ch')
37	spiga	1.1	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.7	'serverName' : self.server_name, \
43			'logger' : common.logger() \
44	spiga	1.1	}
45			try:
46			CredAPI = CredentialAPI( configAPI )
47			except Exception, err :
48	spiga	1.5	common.logger.debug( "Configuring Credential API: " +str(traceback.format_exc()))
49	spiga	1.1	raise CrabException("ERROR: Unable to configure Credential Client API %s\n"%str(err))
50			if not CredAPI.checkCredential(Time=100) :
51	spiga	1.5	common.logger.info("Please renew your %s :\n"%self.credentialType)
52	spiga	1.1	try:
53			CredAPI.ManualRenewCredential()
54			except Exception, ex:
55			raise CrabException(str(ex))
56			try:
57	spiga	1.2	dict = CredAPI.registerCredential()
58	spiga	1.1	except Exception, err:
59	spiga	1.5	common.logger.debug( "Registering Credentials : " +str(traceback.format_exc()))
60	spiga	1.1	raise CrabException("ERROR: Unable to register %s delegating server: %s\n"%(self.credentialType,self.server_name ))