108 lines
3.8 KiB
Python
108 lines
3.8 KiB
Python
import requests
|
|
from decouple import config
|
|
|
|
import schemas
|
|
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_user_id(project_id, user_id):
|
|
data = {
|
|
"filter": {"userId": user_id}
|
|
}
|
|
return __get_live_sessions_ws(project_id=project_id, data=data)
|
|
|
|
|
|
def get_live_sessions_ws(project_id, body: schemas.LiveSessionsSearchPayloadSchema):
|
|
data = {
|
|
"filter": {},
|
|
"pagination": {"limit": body.limit, "page": body.page},
|
|
"sort": {"key": body.sort, "order": body.order}
|
|
}
|
|
for f in body.filters:
|
|
data["filter"][f.type] = f.value
|
|
return __get_live_sessions_ws(project_id=project_id, data=data)
|
|
|
|
|
|
def __get_live_sessions_ws(project_id, data):
|
|
project_key = projects.get_project_key(project_id)
|
|
try:
|
|
connected_peers = requests.post(config("assist") % config("S3_KEY") + f"/{project_key}", json=data,
|
|
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, data={"filter": {"sessionId": session_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.post(config("assistList") % config("S3_KEY") + f"/{project_key}",
|
|
json={"filter": {"sessionId": session_id}},
|
|
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
|