diff --git a/src/features/phone_esm/straw/main.py b/src/features/phone_esm/straw/main.py index e3ff1aca..306bb681 100644 --- a/src/features/phone_esm/straw/main.py +++ b/src/features/phone_esm/straw/main.py @@ -1,6 +1,40 @@ import pandas as pd -from esm_preprocess import QUESTIONNAIRE_IDS +QUESTIONNAIRE_IDS = { + "sleep_quality": 1, + "PANAS_positive_affect": 8, + "PANAS_negative_affect": 9, + "JCQ_job_demand": 10, + "JCQ_job_control": 11, + "JCQ_supervisor_support": 12, + "JCQ_coworker_support": 13, + "PFITS_supervisor": 14, + "PFITS_coworkers": 15, + "UWES_vigor": 16, + "UWES_dedication": 17, + "UWES_absorption": 18, + "COPE_active": 19, + "COPE_support": 20, + "COPE_emotions": 21, + "balance_life_work": 22, + "balance_work_life": 23, + "recovery_experience_detachment": 24, + "recovery_experience_relaxation": 25, + "symptoms": 26, + "appraisal_stressfulness_event": 87, + "appraisal_threat": 88, + "appraisal_challenge": 89, + "appraisal_event_time": 90, + "appraisal_event_duration": 91, + "appraisal_event_work_related": 92, + "appraisal_stressfulness_period": 93, + "late_work": 94, + "work_hours": 95, + "left_work": 96, + "activities": 97, + "coffee_breaks": 98, + "at_work_yet": 99, +} def straw_features(sensor_data_files, time_segment, provider, filter_data_by_segment, *args, **kwargs):