def downloadVOMSProxy(
self,
userDN,
userGroup,
limited=False,
requiredTimeLeft=1200,
cacheTime=14400,
requiredVOMSAttribute=None,
proxyToConnect=None,
token=None,
):
"""Download a proxy if needed and transform it into a VOMS one
:param str userDN: user DN
:param str userGroup: user group
:param boolean limited: if need limited proxy
:param int requiredTimeLeft: required proxy live time in a seconds
:param int cacheTime: store in a cache time in a seconds
:param str requiredVOMSAttribute: VOMS attr to add to the proxy
:param X509Chain proxyToConnect: proxy as a chain
:param str token: valid token to get a proxy
:return: S_OK(X509Chain)/S_ERROR()
"""
cacheKey = (userDN, userGroup, requiredVOMSAttribute, limited)
if self.__vomsProxiesCache.exists(cacheKey, requiredTimeLeft):
return S_OK(self.__vomsProxiesCache.get(cacheKey))
req = X509Request()
req.generateProxyRequest(limited=limited)
if proxyToConnect:
rpcClient = Client(url="Framework/ProxyManager",
proxyChain=proxyToConnect,
timeout=120)
else:
rpcClient = Client(url="Framework/ProxyManager", timeout=120)
if token:
retVal = rpcClient.getVOMSProxyWithToken(
userDN,
userGroup,
req.dumpRequest()["Value"],
int(cacheTime + requiredTimeLeft),
token,
requiredVOMSAttribute,
)
else:
retVal = rpcClient.getVOMSProxy(userDN, userGroup,
req.dumpRequest()["Value"],
int(cacheTime + requiredTimeLeft),
requiredVOMSAttribute)
if not retVal["OK"]:
return retVal
chain = X509Chain(keyObj=req.getPKey())
retVal = chain.loadChainFromString(retVal["Value"])
if not retVal["OK"]:
return retVal
self.__vomsProxiesCache.add(cacheKey,
chain.getRemainingSecs()["Value"], chain)
return S_OK(chain)
