diff --git a/api/chalicelib/core/assist.py b/api/chalicelib/core/assist.py index 1be5c2f46..c8774d3b7 100644 --- a/api/chalicelib/core/assist.py +++ b/api/chalicelib/core/assist.py @@ -77,6 +77,15 @@ def get_live_sessions_ws(project_id): return live_peers +def get_live_session_by_id(project_id, session_id): + all_live = get_live_sessions_ws(project_id) + for l in all_live: + if l.get("sessionID") == session_id: + l["live"] = True + return l + return None + + def is_live(project_id, session_id, project_key=None): if project_key is None: project_key = projects.get_project_key(project_id) diff --git a/api/chalicelib/core/sessions.py b/api/chalicelib/core/sessions.py index 9349ac82b..1b2de05b3 100644 --- a/api/chalicelib/core/sessions.py +++ b/api/chalicelib/core/sessions.py @@ -101,7 +101,8 @@ def get_by_id2_pg(project_id, session_id, user_id, full_data=False, include_fav_ project_key=data["projectKey"]) return data - return None + else: + return assist.get_live_session_by_id(project_id=project_id, session_id=session_id) def __get_sql_operator(op: schemas.SearchEventOperator):