def add_query_execution_access_request(execution_id):
uid = current_user.id
access_request = AccessRequest.get(query_execution_id=execution_id, uid=uid)
if not access_request:
access_request = AccessRequest.create(
{"query_execution_id": execution_id, "uid": uid}
)
send_query_execution_access_request_notification(execution_id=execution_id, uid=uid)
return access_request
def create_data_doc_access_request(doc_id, uid, commit=True, session=None):
request = AccessRequest(uid=uid, data_doc_id=doc_id)
session.add(request)
if commit:
session.commit()
else:
session.flush()
session.refresh(request)
return request
def create_board_access_request(board_id, uid, commit=True, session=None):
request = AccessRequest(uid=uid, board_id=board_id)
session.add(request)
if commit:
session.commit()
else:
session.flush()
session.refresh(request)
return request
def add_query_execution_viewer(execution_id, uid):
verify_query_execution_owner(execution_id)
with DBSession() as session:
viewer = QueryExecutionViewer.create(
{
"query_execution_id": execution_id,
"uid": uid,
"created_by": current_user.id,
},
commit=False,
session=session,
)
access_request = AccessRequest.get(
session=session, query_execution_id=execution_id, uid=uid
)
if access_request:
AccessRequest.delete(id=access_request.id, session=session, commit=False)
send_query_execution_invitation_notification(
execution_id=execution_id, uid=uid, session=session
)
session.commit()
return viewer.to_dict()
def delete_query_execution_access_request(execution_id, uid):
verify_query_execution_owner(execution_id)
access_request = AccessRequest.get(query_execution_id=execution_id,
uid=uid)
if access_request:
AccessRequest.delete(id=access_request.id)
def get_query_execution_access_requests(execution_id):
verify_query_execution_owner(execution_id)
return AccessRequest.get_all(query_execution_id=execution_id)
