Merge pull request #167 from seattleflu/childcare-add-event

Accommodate unscheduled event in childcare ETL
seattleflu · Nov 20, 2020 · a4d320d · a4d320d
2 parents 9112c52 + 8199577
commit a4d320d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/seattleflu/id3c/cli/command/etl/redcap_det_childcare.py b/lib/seattleflu/id3c/cli/command/etl/redcap_det_childcare.py
@@ -54,6 +54,7 @@ class StudyArm(Enum):
 
 ENROLLMENT_EVENT_NAME_PREFIX = 'enrollment_arm_'
 ENCOUNTER_EVENT_NAME_PREFIX = 'week'
+UNSCHEDULED_ENCOUNTER_EVENT_NAME = 'unscheduled_arm_1'
 COLLECTION_CODE = CollectionCode.HOME_HEALTH
 
 
@@ -154,7 +155,8 @@ def redcap_det_childcare(*, db: DatabaseSession, cache: TTLCache, det: dict,
 
         if redcap_record_instance.event_name.startswith(ENROLLMENT_EVENT_NAME_PREFIX):
             event_type = EventType.ENROLLMENT
-        elif redcap_record_instance.event_name.startswith(ENCOUNTER_EVENT_NAME_PREFIX):
+        elif redcap_record_instance.event_name.startswith(ENCOUNTER_EVENT_NAME_PREFIX) \
+            or redcap_record_instance.event_name == UNSCHEDULED_ENCOUNTER_EVENT_NAME:
             event_type = EventType.ENCOUNTER
         else:
             LOG.error(f'The record instance has an unexpected event name: {redcap_record_instance.event_name}')