Rc/test epic integration

corehq/motech/fhir/tasks.py

@@ -1,3 +1,6 @@
+import dateutil, requests, jwt, time, uuid
+from datetime import datetime
+
 from collections import namedtuple
 from typing import Generator, List
 from uuid import uuid4
@@ -12,8 +15,9 @@
 from corehq import toggles
 from corehq.apps.celery import periodic_task, task
 from corehq.apps.hqcase.utils import submit_case_blocks
+from corehq.apps.hqcase.case_helper import CaseHelper
 from corehq.form_processor.exceptions import CaseNotFound
-from corehq.form_processor.models import CommCareCase
+from corehq.form_processor.models import CommCareCase, CommCareCaseIndex
 from corehq.motech.const import (
     IMPORT_FREQUENCY_DAILY,
     IMPORT_FREQUENCY_MONTHLY,
@@ -394,5 +398,252 @@
     )
 
 
+def generate_epic_jwt():
+    key = settings.EPIC_PRIVATE_KEY
+    # token will expire in 4 mins
+    exp = int(time.time()) + 240
+    jti = str(uuid.uuid4())
+    header = {
+        "alg": "RS256",
+        "typ": "JWT",
+    }
+    payload = {
+        "iss": settings.EPIC_CLIENT_ID,
+        "sub": settings.EPIC_CLIENT_ID,
+        "aud": "https://fhir.epic.com/interconnect-fhir-oauth/oauth2/token",
+        "jti": jti,
+        "exp": exp
+    }
+    token = jwt.encode(payload, key, algorithm="RS256", headers=header)
+    return token
+
+
+def request_epic_access_token():
+    headers = {
+        "Content_Type": "application/x-www-form-urlencoded",
+    }
+    data = {
+        "grant_type": "client_credentials",
+        "client_assertion_type": "urn:ietf:params:oauth:client-assertion-type:jwt-bearer",
+        "client_assertion": generate_epic_jwt()
+    }
+    url = "https://fhir.epic.com/interconnect-fhir-oauth/oauth2/token"
+    response = requests.post(url, data=data, headers=headers)
+    if response.status_code == 200:
+        return response.json().get('access_token')
+    elif response.status_code >= 400:
+        return response.raise_for_status()
+
+
+def get_patient_fhir_id(given_name, family_name, birthdate, access_token):
+    url = f"https://fhir.epic.com/interconnect-fhir-oauth/api/FHIR/R4/Patient?birthdate={birthdate}&family={family_name}&given={given_name}&_format=json"
+    headers = {
+        'authorization': 'Bearer %s' % access_token,
+    }
+    response = requests.get(url, headers=headers)
+    if response.status_code == 200:
+        response_json = response.json()
+        fhir_id = None
+        entry = response_json.get('entry')[0]
+        if entry: 
+            resource = entry.get('resource')
+            if resource:
+                fhir_id = resource.get('id')
+        return fhir_id
+    elif response.status_code >= 400:
+        response.raise_for_status()
+
+
+# TODO add time param 12 weeks from study start date
+def get_epic_appointments_for_patient(fhir_id, access_token):
+    appointments = []
+    headers = {
+        'authorization': 'Bearer %s' % access_token,
+    }
+    url = f"https://fhir.epic.com/interconnect-fhir-oauth/api/FHIR/R4/Appointment?&patient={fhir_id}&service-category=appointment&_format=json"
+    response = requests.get(url, headers=headers)
+    if response.status_code == 200:
+        json_response = response.json()
+        entries = json_response['entry']
+        for entry in entries:
+            appointments.append(entry)
+    elif response.status_code >= 400:
+        response.raise_for_status()
+    return appointments
+
+
+def convert_date_and_time_to_utc_timestamp(date, time):
+    date_time = date + "T" + time
+    utc_zone = dateutil.tz.gettz('UTC')
+    # Hardcoded for MGH study
+    local_zone = dateutil.tz.gettz('America/New_York')
+    local_datetime = datetime.fromisoformat(date_time)
+    local_datetime = local_datetime.replace(tzinfo=local_zone)
+    utc_datetime = local_datetime.astimezone(utc_zone)
+    utc_iso_format = utc_datetime.strftime('%Y-%m-%dT%H:%M:%SZ')
+
+    return utc_iso_format
+
+
+def convert_utc_timestamp_to_date_and_time(utc_timestamp):
+    utc_zone = dateutil.tz.gettz('UTC')
+    local_zone = dateutil.tz.gettz('America/New_York')
+    utc_datetime = datetime.fromisoformat(utc_timestamp.replace('Z', ''))
+    utc_datetime = utc_datetime.replace(tzinfo=utc_zone)
+    local_datetime = utc_datetime.astimezone(local_zone)
+    date = local_datetime.strftime('%Y-%m-%d')
+    time = local_datetime.strftime('%H:%M')
+
+    return date, time
+
+
+def sync_all_appointments_domain(domain):
+    try:
+        access_token = request_epic_access_token()
+    except Exception:
+        return None
+    # get all patient case ids for domain
+    patient_case_ids = CommCareCase.objects.get_open_case_ids_in_domain_by_type(domain, 'patient')
+    # get all patient cases
+    patient_cases = CommCareCase.objects.get_cases(patient_case_ids)
+
+    # get extension (appointment) cases ids, for all patients
+    appointment_case_ids = CommCareCaseIndex.objects.get_extension_case_ids(
+        domain, patient_case_ids, False, 'appointment')
+    # get appointment cases in commcare
+    appointment_cases = CommCareCase.objects.get_cases(appointment_case_ids)
+
+    # get fhir ids for appointments currently in commcare
+    appointment_fhir_ids = [
+        appointment_case.get_case_property('fhir_id') for appointment_case in appointment_cases
+    ]
+
+    for patient in patient_cases:
+        patient_helper = CaseHelper(case=patient, domain=domain)
+        patient_fhir_id = patient.get_case_property('patient_fhir_id')
+        if not patient_fhir_id:
+            given = patient.get_case_property('given_name')
+            family = patient.get_case_property('family_name')
+            birthdate = patient.get_case_property('birthdate')
+            patient_fhir_id = get_patient_fhir_id(given, family, birthdate, access_token)
+
+            if patient_fhir_id is not None:
+                patient_helper.update({'properties': {
+                    'patient_fhir_id': patient_fhir_id,
+                }})
+
+        epic_appointments_to_add = []
+        epic_appointments_to_update = []
+        # Get all appointments for patient from epic
+        epic_appointment_records = get_epic_appointments_for_patient(patient_fhir_id, access_token)
+        for appointment in epic_appointment_records:
+            appointment_resource = appointment.get('resource')
+            appointment_id = appointment_resource.get('id')
+            if appointment_id and appointment_id not in appointment_fhir_ids:
+                epic_appointments_to_add.append(appointment)
+            elif appointment_id:
+                epic_appointments_to_update.append(appointment)
+
+        # Add new appointments to commcare
+        for appointment in epic_appointments_to_add:
+            appointment_create_helper = CaseHelper(domain=domain)
+            appointment_resource = appointment.get('resource')
+            if appointment_resource is not None:
+                appointment_description = appointment_resource.get('description') or 'NO DESCRIPTION LISTED'
+                appointment_fhir_timestamp = appointment_resource.get('start')
+                appointment_date, appointment_time = convert_utc_timestamp_to_date_and_time(
+                    appointment_fhir_timestamp)
+                appointment_fhir_id = appointment_resource.get('id')
+                reason = None
+                practitioner = None
+                for p in appointment_resource.get('participant'):
+                    actor = p.get('actor')
+                    if actor and 'Practitioner' in actor.get('reference'):
+                        practitioner = actor.get('display')
+                        break
+                reason_code = appointment_resource.get('reasonCode')
+                if reason_code and reason_code[0] is not None:
+                    reason = reason_code[0].get('text')
+            host_case_id = patient.get_case_property('case_id')
+            appointment_case_data = {
+                'case_name': appointment_fhir_timestamp + appointment_description,
+                'case_type': 'appointment',
+                'indices': {
+                    'patient': {
+                        'case_id': host_case_id,
+                        'case_type': 'patient',
+                        'relationship': 'extension',
+                    }
+                },
+                'properties': {
+                    'appointment_description': appointment_description,
+                    'appointment_fhir_timestamp': appointment_fhir_timestamp,
+                    'appointment_date': appointment_date,
+                    'appointment_time': appointment_time,
+                    'patient_fhir_id': patient_fhir_id,
+                    'fhir_id': appointment_fhir_id,
+                    'reason': reason,
+                    'practitioner': practitioner
+                }
+            }
+            appointment_create_helper.create_case(appointment_case_data)
+
+        # Update existing appointments in commcare if properties have changed in epic
+        for appointment in epic_appointments_to_update:
+            epic_properties_map = {}
+            appointment_resource = appointment.get('resource')
+            if appointment_resource is not None:
+                appointment_description = appointment_resource.get('description') or 'NO DESCRIPTION LISTED'
+                appointment_fhir_timestamp = appointment_resource.get('start')
+                appointment_date, appointment_time = convert_utc_timestamp_to_date_and_time(
+                    appointment_fhir_timestamp)
+                appointment_fhir_id = appointment_resource.get('id')
+                reason = None
+                practitioner = None
+                for p in appointment_resource.get('participant'):
+                    actor = p.get('actor')
+                    if actor and 'Practitioner' in actor.get('reference'):
+                        practitioner = actor.get('display')
+                        break
+                reason_code = appointment_resource.get('reasonCode')
+                if reason_code and reason_code[0] is not None:
+                    reason = reason_code[0].get('text')
+                epic_properties_map.update({
+                    'appointment_description': appointment_description,
+                    'appointment_fhir_timestamp': appointment_fhir_timestamp,
+                    'practitioner': practitioner,
+                    'reason': reason
+                })
+            appointment_case = None
+            for case in appointment_cases:
+                if case.get_case_property('fhir_id') == appointment_fhir_id:
+                    appointment_case = case
+                    break
+            appointment_update_helper = CaseHelper(case=appointment_case, domain=domain)
+            case_properties_to_update = {}
+            changes = False
+            # check for changes and add to case_properties_to_update
+            for k, v in epic_properties_map.items():
+                current_value = appointment_case.get_case_property(k)
+                if current_value != v:
+                    changes = True
+                    case_properties_to_update.update({k: v})
+                    if k == 'appointment_fhir_timestamp':
+                        appointment_date, appointment_time = convert_utc_timestamp_to_date_and_time(v)
+                        case_properties_to_update.update({
+                            'appointment_date': appointment_date,
+                            'appointment_time': appointment_time,
+                        })
+
+            if changes:
+                appointment_update_helper.update({'properties': case_properties_to_update})
+
+
+@periodic_task(run_every=crontab(hour="*", minute=1), queue=settings.CELERY_PERIODIC_QUEUE)
+def sync_all_epic_appointments():
+    for domain in toggles.MGH_EPIC_STUDY.get_enabled_domains():
+        sync_all_appointments_domain(domain)
+
+
 class ServiceRequestNotActive(Exception):
     pass

corehq/toggles/__init__.py

@@ -2240,6 +2240,13 @@ def _commtrackify(domain_name, toggle_is_enabled):
     help_link="https://confluence.dimagi.com/display/GS/FHIR+API+Documentation",
 )
 
+MGH_EPIC_STUDY = StaticToggle(
+    'mgh_epic_study',
+    'Enable updating/creating extension cases through open epic API',
+    TAG_CUSTOM,
+    namespaces=[NAMESPACE_DOMAIN],
+)
+
 AUTO_DEACTIVATE_MOBILE_WORKERS = StaticToggle(
     'auto_deactivate_mobile_workers',
     'Development flag for auto-deactivation of mobile workers. To be replaced '

settings.py

@@ -1014,6 +1014,9 @@ def _pkce_required(client_id):
 # encryption or signing workflows.
 HQ_PRIVATE_KEY = None
 
+EPIC_PRIVATE_KEY = None
+EPIC_CLIENT_ID = None
+
 KAFKA_BROKERS = ['localhost:9092']
 KAFKA_API_VERSION = None