cr-connect-workflow/crc/scripts/get_study_associate.py

from crc.api.common import ApiError
from crc.models.study import StudyAssociatedSchema
from crc.scripts.script import Script
from crc.services.study_service import StudyService


class GetStudyAssociate(Script):



    def get_description(self):
        return """
Returns how a single person is associated with a study and what access they need,
 or raises an error if the person is not associated with the study.
example : get_study_associate('sbp3ey') => {'uid':'sbp3ey','role':'Unicorn Herder', 'send_email': False, 
'access':True}

"""
    def do_task_validate_only(self, task, study_id, workflow_id, *args, **kwargs):
        if len(args) < 1:
            raise ApiError('no_user_id_specified', 'A uva uid is the sole argument to this function')
        return {'uid': 'sbp3ey', 'role': 'Unicorn Herder', 'send_email': False, 'access': True}

    def do_task(self, task, study_id, workflow_id, *args, **kwargs):
        if len(args) < 1:
            raise ApiError('no_user_id_specified', 'A uva uid is the sole argument to this function')
        if not isinstance(args[0], str):
            raise ApiError('argument_should_be_string', 'A uva uid is always a string, please check type')
        associate = StudyService.get_study_associate(study_id=study_id, uid=args[0])
        return StudyAssociatedSchema().dump(associate)