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()