diff --git a/api/chalicelib/core/sessions/sessions_replay.py b/api/chalicelib/core/sessions/sessions_replay.py index 46b469737..a796dbb09 100644 --- a/api/chalicelib/core/sessions/sessions_replay.py +++ b/api/chalicelib/core/sessions/sessions_replay.py @@ -21,7 +21,23 @@ def __group_metadata(session, project_metadata): def get_pre_replay(project_id, session_id): + with pg_client.PostgresClient() as cur: + query = cur.mogrify( + f"""\ + SELECT encode(file_key,'hex') AS file_key + FROM public.sessions + WHERE project_id = %(project_id)s + AND session_id = %(session_id)s;""", + {"project_id": project_id, "session_id": session_id} + ) + cur.execute(query=query) + + data = cur.fetchone() + file_key = None + if data is not None: + file_key = data['file_key'] return { + 'fileKey': file_key, 'domURL': [sessions_mobs.get_first_url(project_id=project_id, session_id=session_id, check_existence=False)]}