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 GetStudyAssociates(Script):

    argument_error_message = "You must supply at least one argument to the " \
                             "update_study_associates task, an array of objects in the form " \
                             "{'uid':'someid', 'role': 'text', 'send_email: 'boolean', " \
                             "'access':'boolean'} "


    def get_description(self):
        return """Returns all people associated with the study - Will always return the study owner as assocated
example : get_study_associates() => [{'uid':'sbp3ey','role':'Unicorn Herder', 'send_email': False, 'access':True}] 

"""

    def do_task_validate_only(self, task, study_id, workflow_id, *args, **kwargs):
        study_associates = [
            {'uid': 'dhf8r', 'role': 'Chief Bee Keeper', 'send_email': True, 'access': True,
             'ldap_info': {
                    'uid': 'dhf8r',
                    'display_name': "Dan Funk",
                     'email_address': 'dhf8r@virginia.edu',
                     'telephone_number': '',
                     'title': '',
                     'department': '',
                     'affiliation': '',
                    'sponsor_type': '',
             }},
            {'uid': 'lb3dp', 'role': 'Chief Cat Herder', 'send_email': True, 'access': True,
             'ldap_info': {
                 'uid': 'dhf8r',
                 'display_name': "Dan Funk",
                 'email_address': 'dhf8r@virginia.edu',
                 'telephone_number': '',
                 'title': '',
                 'department': '',
                 'affiliation': '',
                 'sponsor_type': ''
             }},
        ]
        return study_associates

    def do_task(self, task, study_id, workflow_id, *args, **kwargs):
        return StudyAssociatedSchema(many=True).dump(StudyService.get_study_associates(study_id))