Get rating key for live sessions from websocket data

This commit is contained in:
JonnyWong16 2018-01-21 13:09:02 -08:00
parent aa3450bfcc
commit 5d8455d141

View file

@ -54,7 +54,7 @@ class ActivityHandler(object):
def get_rating_key(self): def get_rating_key(self):
if self.is_valid_session(): if self.is_valid_session():
return int(self.timeline['ratingKey']) return self.timeline['ratingKey']
return None return None
@ -65,6 +65,10 @@ class ActivityHandler(object):
if session_list: if session_list:
for session in session_list['sessions']: for session in session_list['sessions']:
if int(session['session_key']) == self.get_session_key(): if int(session['session_key']) == self.get_session_key():
# Live sessions don't have rating keys in sessions
# Get it from the websocket data
if not session['rating_key']:
session['rating_key'] = self.get_rating_key()
return session return session
return None return None