import requests from decouple import config from chalicelib.core import projects SESSION_PROJECTION_COLS = """s.project_id, s.session_id::text AS session_id, s.user_uuid, s.user_id, s.user_agent, s.user_os, s.user_browser, s.user_device, s.user_device_type, s.user_country, s.start_ts, s.user_anonymous_id, s.platform """ def get_live_sessions_ws(project_id, user_id=None): project_key = projects.get_project_key(project_id) params = {} if user_id and len(user_id) > 0: params["userId"] = user_id try: connected_peers = requests.get(config("assist") % config("S3_KEY") + f"/{project_key}", params, timeout=config("assistTimeout", cast=int, default=5)) if connected_peers.status_code != 200: print("!! issue with the peer-server") print(connected_peers.text) return [] live_peers = connected_peers.json().get("data", []) except requests.exceptions.Timeout: print("Timeout getting Assist response") live_peers = [] except Exception as e: print("issue getting Live-Assist response") print(str(e)) print("expected JSON, received:") try: print(connected_peers.text) except: print("couldn't get response") live_peers = [] for s in live_peers: s["live"] = True s["projectId"] = project_id live_peers = sorted(live_peers, key=lambda l: l.get("timestamp", 0), reverse=True) 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 str(l.get("sessionID")) == str(session_id): 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) try: connected_peers = requests.get(config("assistList") % config("S3_KEY") + f"/{project_key}", timeout=config("assistTimeout", cast=int, default=5)) if connected_peers.status_code != 200: print("!! issue with the peer-server") print(connected_peers.text) return False connected_peers = connected_peers.json().get("data", []) except requests.exceptions.Timeout: print("Timeout getting Assist response") return False except Exception as e: print("issue getting Assist response") print(str(e)) print("expected JSON, received:") try: print(connected_peers.text) except: print("couldn't get response") return False return str(session_id) in connected_peers def get_ice_servers(): return config("iceServers") if config("iceServers", default=None) is not None \ and len(config("iceServers")) > 0 else None