Export timezone info, too.

rapids
junos 2021-12-01 17:08:24 +01:00
parent 0e6a18a660
commit 6eca98962f
2 changed files with 41 additions and 0 deletions

View File

@ -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

View File

@ -1,5 +1,7 @@
import datetime import datetime
import pandas as pd import pandas as pd
from features.timezone import get_timezone_data
from pyprojroot import here from pyprojroot import here
import participants.query_db import participants.query_db
@ -50,3 +52,12 @@ participants_usernames_empatica.to_csv(
index=False, index=False,
line_terminator="\n", 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",
)