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 \
                     }
        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.6
Committed:	Tue May 26 16:53:23 2009 UTC (15 years, 11 months ago) by spiga
Content type:	text/x-python
Branch:	MAIN
CVS Tags:	CRAB_2_6_0, CRAB_2_6_0_pre14, CRAB_2_6_0_pre13, CRAB_2_6_0_pre12, CRAB_2_6_0_pre11, CRAB_2_6_0_pre10, CRAB_2_6_0_pre9, CRAB_2_6_0_pre8, CRAB_2_6_0_pre7, CRAB_2_6_0_pre6, CRAB_2_6_0_pre5, CRAB_2_6_0_pre4, CRAB_2_6_0_pre3
Changes since 1.5:	+1 -1 lines
Log Message:	adapting code to EDG --> GRID migration
#	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.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 ))