rucio 32.8.6__py3-none-any.whl

rucio/api/request.py

@@ -0,0 +1,220 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Interface for the requests abstraction layer
+"""
+
+from typing import TYPE_CHECKING
+
+from rucio.api import permission
+from rucio.common import exception
+from rucio.common.types import InternalAccount, InternalScope
+from rucio.common.utils import api_update_return_dict
+from rucio.core import request
+from rucio.core.rse import get_rse_id
+from rucio.db.sqla.session import read_session, stream_session, transactional_session
+
+if TYPE_CHECKING:
+    from sqlalchemy.orm import Session
+
+
+@transactional_session
+def queue_requests(requests, issuer, vo='def', *, session: "Session"):
+    """
+    Submit transfer or deletion requests on destination RSEs for data identifiers.
+
+    :param requests: List of dictionaries containing 'scope', 'name', 'dest_rse_id', 'request_type', 'attributes'
+    :param issuer: Issuing account as a string.
+    :param vo: The VO to act on.
+    :param session: The database session in use.
+    :returns: List of Request-IDs as 32 character hex strings
+    """
+
+    kwargs = {'requests': requests, 'issuer': issuer}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='queue_requests', kwargs=kwargs, session=session):
+        raise exception.AccessDenied(f'{issuer} can not queue request')
+
+    for req in requests:
+        req['scope'] = InternalScope(req['scope'], vo=vo)
+        if 'account' in req:
+            req['account'] = InternalAccount(req['account'], vo=vo)
+
+    new_requests = request.queue_requests(requests, session=session)
+    return [api_update_return_dict(r, session=session) for r in new_requests]
+
+
+@transactional_session
+def cancel_request(request_id, issuer, account, vo='def', *, session: "Session"):
+    """
+    Cancel a request.
+
+    :param request_id: Request Identifier as a 32 character hex string.
+    :param issuer: Issuing account as a string.
+    :param account: Account identifier as a string.
+    :param vo: The VO to act on.
+    :param session: The database session in use.
+    """
+
+    kwargs = {'account': account, 'issuer': issuer, 'request_id': request_id}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='cancel_request_', kwargs=kwargs, session=session):
+        raise exception.AccessDenied('%s cannot cancel request %s' % (account, request_id))
+
+    raise NotImplementedError
+
+
+@transactional_session
+def cancel_request_did(scope, name, dest_rse, request_type, issuer, account, vo='def', *, session: "Session"):
+    """
+    Cancel a request based on a DID and request type.
+
+    :param scope: Data identifier scope as a string.
+    :param name: Data identifier name as a string.
+    :param dest_rse: RSE name as a string.
+    :param request_type: Type of the request as a string.
+    :param issuer: Issuing account as a string.
+    :param account: Account identifier as a string.
+    :param vo: The VO to act on.
+    :param session: The database session in use.
+    """
+
+    dest_rse_id = get_rse_id(rse=dest_rse, vo=vo, session=session)
+
+    kwargs = {'account': account, 'issuer': issuer}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='cancel_request_did', kwargs=kwargs, session=session):
+        raise exception.AccessDenied(f'{account} cannot cancel {request_type} request for {scope}:{name}')
+
+    scope = InternalScope(scope, vo=vo)
+    return request.cancel_request_did(scope, name, dest_rse_id, request_type, session=session)
+
+
+@transactional_session
+def get_next(request_type, state, issuer, account, vo='def', *, session: "Session"):
+    """
+    Retrieve the next request matching the request type and state.
+
+    :param request_type: Type of the request as a string.
+    :param state: State of the request as a string.
+    :param issuer: Issuing account as a string.
+    :param account: Account identifier as a string.
+    :param vo: The VO to act on.
+    :param session: The database session in use.
+    :returns: Request as a dictionary.
+    """
+
+    kwargs = {'account': account, 'issuer': issuer, 'request_type': request_type, 'state': state}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='get_next', kwargs=kwargs, session=session):
+        raise exception.AccessDenied(f'{account} cannot get the next request of type {request_type} in state {state}')
+
+    reqs = request.get_and_mark_next(request_type, state, session=session)
+    return [api_update_return_dict(r, session=session) for r in reqs]
+
+
+@read_session
+def get_request_by_did(scope, name, rse, issuer, vo='def', *, session: "Session"):
+    """
+    Retrieve a request by its DID for a destination RSE.
+
+    :param scope: The scope of the data identifier as a string.
+    :param name: The name of the data identifier as a string.
+    :param rse: The destination RSE of the request as a string.
+    :param issuer: Issuing account as a string.
+    :param vo: The VO to act on.
+    :param session: The database session in use.
+    :returns: Request as a dictionary.
+    """
+    rse_id = get_rse_id(rse=rse, vo=vo, session=session)
+
+    kwargs = {'scope': scope, 'name': name, 'rse': rse, 'rse_id': rse_id, 'issuer': issuer}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='get_request_by_did', kwargs=kwargs, session=session):
+        raise exception.AccessDenied(f'{issuer} cannot retrieve the request DID {scope}:{name} to RSE {rse}')
+
+    scope = InternalScope(scope, vo=vo)
+    req = request.get_request_by_did(scope, name, rse_id, session=session)
+
+    return api_update_return_dict(req, session=session)
+
+
+@read_session
+def get_request_history_by_did(scope, name, rse, issuer, vo='def', *, session: "Session"):
+    """
+    Retrieve a historical request by its DID for a destination RSE.
+
+    :param scope: The scope of the data identifier as a string.
+    :param name: The name of the data identifier as a string.
+    :param rse: The destination RSE of the request as a string.
+    :param issuer: Issuing account as a string.
+    :param vo: The VO to act on.
+    :param session: The database session in use.
+    :returns: Request as a dictionary.
+    """
+    rse_id = get_rse_id(rse=rse, vo=vo, session=session)
+
+    kwargs = {'scope': scope, 'name': name, 'rse': rse, 'rse_id': rse_id, 'issuer': issuer}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='get_request_history_by_did', kwargs=kwargs, session=session):
+        raise exception.AccessDenied(f'{issuer} cannot retrieve the request DID {scope}:{name} to RSE {rse}')
+
+    scope = InternalScope(scope, vo=vo)
+    req = request.get_request_history_by_did(scope, name, rse_id, session=session)
+
+    return api_update_return_dict(req, session=session)
+
+
+@stream_session
+def list_requests(src_rses, dst_rses, states, issuer, vo='def', *, session: "Session"):
+    """
+    List all requests in a specific state from a source RSE to a destination RSE.
+
+    :param src_rses: source RSEs.
+    :param dst_rses: destination RSEs.
+    :param states: list of request states.
+    :param issuer: Issuing account as a string.
+    :param session: The database session in use.
+    """
+    src_rse_ids = [get_rse_id(rse=rse, vo=vo, session=session) for rse in src_rses]
+    dst_rse_ids = [get_rse_id(rse=rse, vo=vo, session=session) for rse in dst_rses]
+
+    kwargs = {'src_rse_id': src_rse_ids, 'dst_rse_id': dst_rse_ids, 'issuer': issuer}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='list_requests', kwargs=kwargs, session=session):
+        raise exception.AccessDenied(f'{issuer} cannot list requests from RSEs {src_rses} to RSEs {dst_rses}')
+
+    for req in request.list_requests(src_rse_ids, dst_rse_ids, states, session=session):
+        req = req.to_dict()
+        yield api_update_return_dict(req, session=session)
+
+
+@stream_session
+def list_requests_history(src_rses, dst_rses, states, issuer, vo='def', offset=None, limit=None, *, session: "Session"):
+    """
+    List all historical requests in a specific state from a source RSE to a destination RSE.
+
+    :param src_rses: source RSEs.
+    :param dst_rses: destination RSEs.
+    :param states: list of request states.
+    :param issuer: Issuing account as a string.
+    :param offset: offset (for paging).
+    :param limit: limit number of results.
+    :param session: The database session in use.
+    """
+    src_rse_ids = [get_rse_id(rse=rse, vo=vo, session=session) for rse in src_rses]
+    dst_rse_ids = [get_rse_id(rse=rse, vo=vo, session=session) for rse in dst_rses]
+
+    kwargs = {'src_rse_id': src_rse_ids, 'dst_rse_id': dst_rse_ids, 'issuer': issuer}
+    if not permission.has_permission(issuer=issuer, vo=vo, action='list_requests_history', kwargs=kwargs, session=session):
+        raise exception.AccessDenied(f'{issuer} cannot list requests from RSEs {src_rses} to RSEs {dst_rses}')
+
+    for req in request.list_requests_history(src_rse_ids, dst_rse_ids, states, offset, limit, session=session):
+        req = req.to_dict()
+        yield api_update_return_dict(req, session=session)