Skip to content

Commit

Permalink
added get_exitquestionnaire methods to Case model
Browse files Browse the repository at this point in the history
  • Loading branch information
Antonior26 committed Aug 19, 2019
1 parent 6479d47 commit 8527eef
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 3 deletions.
20 changes: 18 additions & 2 deletions pycipapi/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,8 @@ class ClinicalReport(object):
def __init__(self, **kwargs):
self.clinical_report_data = kwargs.get('clinical_report_data')
self.created_at = kwargs.get('created_at')
self.exit_questionnaire = kwargs.get('exit_questionnaire')
self.exit_questionnaire = ExitQuestionnaire(**kwargs.get('exit_questionnaire')) if\
kwargs.get('exit_questionnaire') else None
self.clinical_report_version = kwargs.get('clinical_report_version')
self.valid = kwargs.get('valid')
self.cva_variants_status = kwargs.get('cva_variants_status')
Expand Down Expand Up @@ -188,7 +189,7 @@ def _load_data(self, **kwargs):
self.files = kwargs.get('files')
self.interpretation_request_data = kwargs.get('interpretation_request_data')
self.interpreted_genome = [InterpretedGenome(**ig) for ig in kwargs.get('interpreted_genome', [])]
self.clinical_report = kwargs.get('clinical_report')
self.clinical_report = [ClinicalReport(**cr) for cr in kwargs.get('clinical_report', [])]
self.workspaces = kwargs.get('workspaces')

@property
Expand Down Expand Up @@ -382,6 +383,21 @@ def submit_clinical_report(self, cip_api_client, payload, partner_id, analysis_t
payload=payload, partner_id=partner_id, analysis_type=analysis_type, report_id=report_id, **params
)))

def get_exit_questionnaire(self):
if self.has_clinical_reports:
list_of_cr = self.clinical_report
list_of_cr.reverse()
for cr in list_of_cr:
if cr.exit_questionnaire:
return cr.exit_questionnaire
return None

def get_exit_questionnaires(self):
if self.has_clinical_reports:
for cr in self.clinical_report:
if cr.exit_questionnaire:
yield cr.exit_questionnaire


class CipApiOverview(object):
def __init__(self, **kwargs):
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

setup(
name='pycipapi',
version='0.7.0',
version='0.8.0',
packages=find_packages(),
scripts=[],
url='https://github.com/genomicsengland/pycipapi',
Expand Down

0 comments on commit 8527eef

Please sign in to comment.