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

import json

from SpiffWorkflow.bpmn.PythonScriptEngine import Box
from SpiffWorkflow.util.metrics import timeit

from crc import session
from crc.api.common import ApiError
from crc.api.workflow import get_workflow
from crc.models.protocol_builder import ProtocolBuilderInvestigatorType
from crc.models.study import StudyModel, StudySchema
from crc.api import workflow as workflow_api
from crc.scripts.script import Script
from crc.services.document_service import DocumentService
from crc.services.file_service import FileService
from crc.services.protocol_builder import ProtocolBuilderService
from crc.services.study_service import StudyService


class StudyInfo(Script):
    """Please see the detailed description that is provided below. """

    pb = ProtocolBuilderService()
    type_options = ['info', 'investigators', 'roles', 'details', 'documents', 'sponsors']

    # This is used for test/workflow validation, as well as documentation.
    example_data = {
        "StudyInfo": {
            "info": {
                "id": 12,
                "title": "test",
                "primary_investigator_id": 21,
                "user_uid": "dif84",
                "sponsor": "sponsor",
                "ind_number": "1234",
                "inactive": False
            },
            "sponsors": [
                {
                    "COMMONRULEAGENCY": None,
                    "SPONSOR_ID": 2453,
                    "SP_NAME": "Abbott Ltd",
                    "SP_TYPE": "Private",
                    "SP_TYPE_GROUP_NAME": None,
                    "SS_STUDY": 2
                },
                {
                    "COMMONRULEAGENCY": None,
                    "SPONSOR_ID": 2387,
                    "SP_NAME": "Abbott-Price",
                    "SP_TYPE": "Incoming Sub Award",
                    "SP_TYPE_GROUP_NAME": "Government",
                    "SS_STUDY": 2
                },
                {
                    "COMMONRULEAGENCY": None,
                    "SPONSOR_ID": 1996,
                    "SP_NAME": "Abernathy-Heidenreich",
                    "SP_TYPE": "Foundation/Not for Profit",
                    "SP_TYPE_GROUP_NAME": "Other External Funding",
                    "SS_STUDY": 2
                }
            ],

            "investigators": {
                'PI': {
                    'label': ProtocolBuilderInvestigatorType.PI.value,
                    'display': 'Always',
                    'unique': 'Yes',
                    'user_id': 'dhf8r',
                    'display_name': 'Dan Funk',
                    'given_name': 'Dan',
                    'email': 'dhf8r@virginia.edu',
                    'telephone_number': '+1 (434) 924-1723',
                    'title': "E42:He's a hoopy frood",
                    'department': 'E0:EN-Eng Study of Parallel Universes',
                    'affiliation': 'faculty',
                    'sponsor_type': 'Staff'},
                'SC_I': {
                    'label': 'Study Coordinator I',
                    'display': 'Always',
                    'unique': 'Yes',
                    'user_id': None},
                'DC': {
                    'label': 'Department Contact',
                    'display': 'Optional',
                    'unique': 'Yes',
                    'user_id': 'asd3v',
                    'error': 'Unable to locate a user with id asd3v in LDAP'},
                'DEPT_CH': {
                    'label': 'Department Chair',
                    'display': 'Always',
                    'unique': 'Yes',
                    'user_id': 'lb3dp'}
            },
            "documents": {
                'AD_CoCApp': {'category1': 'Ancillary Document', 'category2': 'CoC Application', 'category3': '',
                              'Who Uploads?': 'CRC', 'id': '12',
                              'description': 'Certificate of Confidentiality Application', 'required': False,
                              'study_id': 1, 'code': 'AD_CoCApp',
                              'display_name': 'Ancillary Document / CoC Application',
                              'count': 0, 'files': []},
                'UVACompl_PRCAppr': {'category1': 'UVA Compliance', 'category2': 'PRC Approval', 'category3': '',
                                     'Who Uploads?': 'CRC', 'id': '6',
                                     'description': "Cancer Center's PRC Approval Form",
                                     'required': True, 'study_id': 1, 'code': 'UVACompl_PRCAppr',
                                     'display_name': 'UVA Compliance / PRC Approval', 'count': 1, 'files': [
                        {'file_id': 10,
                         'task_id': 'fakingthisout',
                         'workflow_id': 2,
                         'workflow_spec_id': 'docx'}],
                                     'status': 'complete'}
            },
            "details":
                {},
        }
    }

    def example_to_string(self, key):
        return json.dumps(self.example_data['StudyInfo'][key], indent=2, separators=(',', ': '))

    def get_description(self):
        return """study_info(TYPE), where TYPE is one of 'info', 'investigators', 'details', or 'documents'.

Adds details about the current study to the Task Data.  The type of information required should be 
provided as an argument.  The following arguments are available:

### Info ###
Returns the basic information such as the id and title
```
{info_example}
```

### Investigators ###
Returns detailed information about related personnel.
The order returned is guaranteed to match the order provided in the investigators.xslx reference file.
Detailed information is added in from LDAP about each personnel based on their user_id. 
```
{investigators_example}
```

### Investigator Roles ###
Returns a list of all investigator roles, populating any roles with additional information available from
the Protocol Builder and LDAP.  Its basically just like Investigators, but it includes all the roles, rather
that just those that were set in Protocol Builder.
```
{investigators_example}
```


### Details ###
Returns detailed information about variable keys read in from the Protocol Builder.

### Documents ###
Returns a list of all documents that might be related to a study, reading all columns from the irb_documents.xsl 
file. Including information about any files that were uploaded or generated that relate to a given document. 
Please note this is just a few examples, ALL known document types are returned in an actual call.
```
{documents_example}
```


        """.format(info_example=self.example_to_string("info"),
                   investigators_example=self.example_to_string("investigators"),
                   documents_example=self.example_to_string("documents"),
                   )

    def do_task_validate_only(self, task, study_id, workflow_id, *args, **kwargs):
        """For validation only, pretend no results come back from pb"""
        self.check_args(args, 2)
        # Assure the reference file exists (a bit hacky, but we want to raise this error early, and cleanly.)
        FileService.get_reference_file_data(DocumentService.DOCUMENT_LIST)
        FileService.get_reference_file_data(StudyService.INVESTIGATOR_LIST)
        # we call the real do_task so we can
        # seed workflow validations with settings from studies in PB Mock
        # in order to test multiple paths thru the workflow
        return self.do_task(task, study_id, workflow_id, args[0])

    @timeit
    def do_task(self, task, study_id, workflow_id, *args, **kwargs):
        self.check_args(args, 2)
        prefix = None
        if len(args) > 1:
            prefix = args[1]
        cmd = args[0]
        # study_info = {}
        # if self.__class__.__name__ in task.data:
        #     study_info = task.data[self.__class__.__name__]
        retval = None
        if cmd == 'info':
            study = session.query(StudyModel).filter_by(id=study_id).first()
            schema = StudySchema()
            retval = schema.dump(study)
        if cmd == 'investigators':
            retval = StudyService().get_investigators(study_id)
        if cmd == 'roles':
            retval = StudyService().get_investigators(study_id, all=True)
        if cmd == 'details':
            retval = self.pb.get_study_details(study_id)
        if cmd == 'sponsors':
            retval = self.pb.get_sponsors(study_id)
        if cmd == 'documents':
            retval = StudyService().get_documents_status(study_id)

        return self.box_it(retval, prefix)

    def box_it(self, retval, prefix = None):
        if isinstance(retval, list):
            return [Box(item) for item in retval]
        if isinstance(retval, dict) and prefix is not None:
            return Box({x: retval[x] for x in retval.keys() if x[:len(prefix)] == prefix})
        elif isinstance(retval, dict):
            return Box(retval)


    def check_args(self, args, maxlen=1):
        if len(args) < 1 or len(args) > maxlen or (args[0] not in StudyInfo.type_options):
            raise ApiError(code="missing_argument",
                           message="The StudyInfo script requires a single argument which must be "
                                   "one of %s" % ",".join(StudyInfo.type_options))