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,
                'ldap_info': {
                    'uid': 'sbp3ey',
                    'display_name': "Simply Delightful",
                    'email_address': 'sbp3ey@virginia.edu',
                    'telephone_number': '',
                    'title': '',
                    'department': '',
                    'affiliation': '',
                    'sponsor_type': '',
                }}

    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)