Export timezone info, too.

features/timezone.py

@@ -0,0 +1,30 @@
+from collections.abc import Collection
+
+import pandas as pd
+
+from config.models import Participant, Timezone
+from setup import db_engine, session
+
+
+def get_timezone_data(usernames: Collection) -> pd.DataFrame:
+    """
+    Read the data from the proximity sensor table and return it in a dataframe.
+
+    Parameters
+    ----------
+    usernames: Collection
+        A list of usernames to put into the WHERE condition.
+
+    Returns
+    -------
+    df_proximity: pd.DataFrame
+        A dataframe of proximity data.
+    """
+    query_timezone = (
+        session.query(Timezone, Participant.username)
+        .filter(Participant.id == Timezone.participant_id)
+        .filter(Participant.username.in_(usernames))
+    )
+    with db_engine.connect() as connection:
+        df_timezone = pd.read_sql(query_timezone.statement, connection)
+    return df_timezone

participants/prepare_usernames_file.py

@@ -1,5 +1,7 @@
 import datetime
+
 import pandas as pd
+from features.timezone import get_timezone_data
 from pyprojroot import here
 
 import participants.query_db
@@ -50,3 +52,12 @@ participants_usernames_empatica.to_csv(
     index=False,
     line_terminator="\n",
 )
+
+timezone_df = get_timezone_data(participants_inactive_usernames)
+
+timezone_df.to_csv(
+    here("rapids/data/external/timezone.csv"),
+    header=True,
+    index=False,
+    line_terminator="\n",
+)