def get_access_requested(self):
dropbox_info = self._get_dropbox_info()
request_tokens = self._make_call(
dropboxApiEndpoint + "/1/oauth/request_token",
'OAuth oauth_version="1.0", oauth_signature_method="PLAINTEXT",'
'oauth_consumer_key="' + dropbox_info.app_key + '", oauth_signature="' + dropbox_info.app_secret + '&"',
None
)
# It returns: oauth_token_secret=b9q1n5il4lcc&oauth_token=mh7an9dkrg59
tokens = request_tokens.split('&')
newuser = CloudStorageUser(
user_dn=self.user_dn,
storage_name=dropbox_info.storage_name,
request_token=tokens[1].split('=')[1],
request_token_secret=tokens[0].split('=')[1],
vo_name=''
)
try:
Session.add(newuser)
Session.commit()
except:
Session.rollback()
raise
return request_tokens
def fill_activities():
activity = ActivityShare(vo='testvo',
activity_share=json.dumps({
"data brokering":
0.3,
"data consolidation":
0.4,
"default":
0.02,
"express":
0.4,
"functional test":
0.2,
"production":
0.5,
"production input":
0.25,
"production output":
0.25,
"recovery":
0.4,
"staging":
0.5,
"t0 export":
0.7,
"t0 tape":
0.7,
"user subscriptions":
0.1
}))
Session.add(activity)
Session.commit()
def get_access_granted(self):
dropbox_user_info = self._get_dropbox_user_info()
if not dropbox_user_info:
raise HTTPBadRequest('No registered user for the service "%s" has been found' % self.service)
dropbox_info = self._get_dropbox_info()
if not dropbox_info:
raise HTTPNotFound('Dropbox info not found in the database')
access_tokens = self._make_call(
dropboxApiEndpoint + "/1/oauth/access_token",
'OAuth oauth_version="1.0", oauth_signature_method="PLAINTEXT", oauth_consumer_key="' +
dropbox_info.app_key + '", oauth_token="' + dropbox_user_info.request_token +
'", oauth_signature="' + dropbox_info.app_secret + '&' + dropbox_user_info.request_token_secret + '"',
None
)
# It returns: oauth_token=<access-token>&oauth_token_secret=<access-token-secret>&uid=<user-id>
access_tokens = access_tokens.split('&')
dropbox_user_info.access_token = access_tokens[1].split('=')[1]
dropbox_user_info.access_token_secret = access_tokens[0].split('=')[1]
try:
Session.add(dropbox_user_info)
Session.commit()
except:
Session.rollback()
raise
return access_tokens
def get_access_granted(self):
dropbox_user_info = self._get_dropbox_user_info()
if not dropbox_user_info:
raise HTTPBadRequest('No registered user for the service "%s" has been found' % self.service)
dropbox_info = self._get_dropbox_info()
if not dropbox_info:
raise HTTPNotFound('Dropbox info not found in the databse')
access_tokens = self._make_call(
dropboxApiEndpoint + "/1/oauth/access_token",
'OAuth oauth_version="1.0", oauth_signature_method="PLAINTEXT", oauth_consumer_key="' +
dropbox_info.app_key + '", oauth_token="' + dropbox_user_info.request_token +
'", oauth_signature="' + dropbox_info.app_secret + '&' + dropbox_user_info.request_token_secret + '"',
None
)
# It returns: oauth_token=<access-token>&oauth_token_secret=<access-token-secret>&uid=<user-id>
access_tokens = access_tokens.split('&')
dropbox_user_info.access_token = access_tokens[1].split('=')[1]
dropbox_user_info.access_token_secret = access_tokens[0].split('=')[1]
try:
Session.add(dropbox_user_info)
Session.commit()
except:
Session.rollback()
raise
return access_tokens
def get_access_requested(self):
dropbox_info = self._get_dropbox_info()
request_tokens = self._make_call(
dropboxApiEndpoint + "/1/oauth/request_token",
'OAuth oauth_version="1.0", oauth_signature_method="PLAINTEXT",'
'oauth_consumer_key="' + dropbox_info.app_key + '", oauth_signature="' + dropbox_info.app_secret + '&"',
None
)
# It returns: oauth_token_secret=b9q1n5il4lcc&oauth_token=mh7an9dkrg59
tokens = request_tokens.split('&')
newuser = CloudStorageUser(
user_dn=self.user_dn,
cloudStorage_name=dropbox_info.cloudStorage_name,
request_token=tokens[1].split('=')[1],
request_token_secret=tokens[0].split('=')[1]
)
try:
Session.add(newuser)
Session.commit()
except:
Session.rollback()
raise
return request_tokens
def audit_configuration(action, config):
"""
Logs and stores in the DB a configuration action
"""
audit = ConfigAudit(datetime=datetime.utcnow(),
dn=request.environ['fts3.User.Credentials'].user_dn,
config=config,
action=action)
Session.add(audit)
log.info(action)
def _save_credential(self, dlg_id, dn, proxy, voms_attrs, termination_time):
credential = Credential(
dlg_id=dlg_id,
dn=dn,
proxy=proxy,
voms_attrs=voms_attrs,
termination_time=termination_time
)
Session.add(credential)
Session.commit()
return credential
def setUp(self):
super(TestDrain, self).setUp()
self.setup_gridsite_environment()
Session.add(
Host(hostname='host1.cern.ch',
service_name='fts3',
beat=datetime.utcnow(),
drain=False))
Session.add(
Host(hostname='host1.cern.ch',
service_name='bringonline',
beat=datetime.utcnow(),
drain=False))
Session.add(
Host(hostname='host2.cern.ch',
service_name='fts3',
beat=datetime.utcnow(),
drain=False))
Session.add(
Host(hostname='host2.cern.ch',
service_name='bringonline',
beat=datetime.utcnow(),
drain=False))
Session.commit()
def request(self, id, start_response):
user = request.environ['fts3.User.Credentials']
credentialCache = Session.query(CredentialCache).get((id, user.user_dn))
if credentialCache is None:
(proxyRequest, proxyKey) = generateProxyRequest(user.dn)
credentialCache = CredentialCache(dlg_id = user.delegation_id,
dn = user.user_dn,
cert_request = proxyRequest.as_pem(),
priv_key = proxyKey.as_pem(cipher = None),
voms_attrs = ' '.join(user.voms_cred))
Session.add(credentialCache)
Session.commit()
start_response('200 OK', [('X-Delegation-ID', credentialCache.dlg_id)])
return credentialCache.cert_request
def request(self, id, start_response):
user = request.environ['fts3.User.Credentials']
credentialCache = Session.query(CredentialCache).get(
(id, user.user_dn))
if credentialCache is None:
(proxyRequest, proxyKey) = generateProxyRequest(user.dn)
credentialCache = CredentialCache(
dlg_id=user.delegation_id,
dn=user.user_dn,
cert_request=proxyRequest.as_pem(),
priv_key=proxyKey.as_pem(cipher=None),
voms_attrs=' '.join(user.voms_cred))
Session.add(credentialCache)
Session.commit()
start_response('200 OK', [('X-Delegation-ID', credentialCache.dlg_id)])
return credentialCache.cert_request
def fill_optimizer():
evolution = OptimizerEvolution(datetime=datetime.datetime.utcnow(),
source_se='http://site01.es',
dest_se='http://dest.ch',
success=90,
active=10,
throughput=10)
Session.add(evolution)
evolution = OptimizerEvolution(datetime=datetime.datetime.utcnow(),
source_se='http://site02.ch',
dest_se='http://dest.ch',
success=95,
active=10,
throughput=15)
Session.add(evolution)
evolution = OptimizerEvolution(datetime=datetime.datetime.utcnow(),
source_se='http://site03.fr',
dest_se='http://dest.ch',
success=100,
active=10,
throughput=20)
Session.add(evolution)
Session.commit()
