gerudo
/
rDrama
forked from rDrama/rDrama
1
0
Fork 0
Code Pull Requests Activity

fix keyerror when referrer is https://watchpeopledie.tv/notifications/messages?page=2

master
Aevann 2023-09-28 01:09:09 +03:00
parent 5ea9bfcde0
commit ad16527a40
1 changed files with 35 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
files/routes/chat.py
View File

@ -49,6 +49,7 @@ def auth_required_socketio(f):
v = get_logged_in_user() v = get_logged_in_user()
if not v: return '', 401 if not v: return '', 401
if v.is_permabanned: return '', 403 if v.is_permabanned: return '', 403
g.referrer = request.referrer.split('?')[0]
return make_response(f(*args, v=v, **kwargs)) return make_response(f(*args, v=v, **kwargs))
wrapper.__name__ = f.__name__ wrapper.__name__ = f.__name__
return wrapper return wrapper
@ -58,6 +59,7 @@ def is_not_banned_socketio(f):
v = get_logged_in_user() v = get_logged_in_user()
if not v: return '', 401 if not v: return '', 401
if v.is_suspended: return '', 403 if v.is_suspended: return '', 403
g.referrer = request.referrer.split('?')[0]
return make_response(f(*args, v=v, **kwargs)) return make_response(f(*args, v=v, **kwargs))
wrapper.__name__ = f.__name__ wrapper.__name__ = f.__name__
return wrapper return wrapper
@ -94,7 +96,7 @@ def chat(v):
@socketio.on('speak') @socketio.on('speak')
@is_not_banned_socketio @is_not_banned_socketio
def speak(data, v): def speak(data, v):
if request.referrer not in ALLOWED_REFERRERS: if g.referrer not in ALLOWED_REFERRERS:
return '', 400 return '', 400
image = None image = None
@ -128,30 +130,30 @@ def speak(data, v):
self_only = True self_only = True
else: else:
del muted[vname] del muted[vname]
emit("online", [online[request.referrer], muted], room=request.referrer, broadcast=True) emit("online", [online[g.referrer], muted], room=g.referrer, broadcast=True)
if SITE == 'rdrama.net' and v.admin_level < PERMS['BYPASS_ANTISPAM_CHECKS']: if SITE == 'rdrama.net' and v.admin_level < PERMS['BYPASS_ANTISPAM_CHECKS']:
def shut_up(): def shut_up():
self_only = True self_only = True
muted_until = int(time.time() + 600) muted_until = int(time.time() + 600)
muted[vname] = muted_until muted[vname] = muted_until
emit("online", [online[request.referrer], muted], room=request.referrer, broadcast=True) emit("online", [online[g.referrer], muted], room=g.referrer, broadcast=True)
if not self_only: if not self_only:
identical = [x for x in list(messages[request.referrer].values())[-5:] if v.id == x['user_id'] and text == x['text']] identical = [x for x in list(messages[g.referrer].values())[-5:] if v.id == x['user_id'] and text == x['text']]
if len(identical) >= 3: shut_up() if len(identical) >= 3: shut_up()
if not self_only: if not self_only:
count = len([x for x in list(messages[request.referrer].values())[-12:] if v.id == x['user_id']]) count = len([x for x in list(messages[g.referrer].values())[-12:] if v.id == x['user_id']])
if count >= 10: shut_up() if count >= 10: shut_up()
if not self_only: if not self_only:
count = len([x for x in list(messages[request.referrer].values())[-25:] if v.id == x['user_id']]) count = len([x for x in list(messages[g.referrer].values())[-25:] if v.id == x['user_id']])
if count >= 20: shut_up() if count >= 20: shut_up()
data = { data = {
"id": id, "id": id,
"quotes": quotes if messages[request.referrer].get(quotes) else '', "quotes": quotes if messages[g.referrer].get(quotes) else '',
"hat": v.hat_active(v)[0], "hat": v.hat_active(v)[0],
"user_id": v.id, "user_id": v.id,
"username": v.username, "username": v.username,
@ -171,53 +173,53 @@ def speak(data, v):
username = i.group(1).lower() username = i.group(1).lower()
muted_until = int(int(i.group(2)) * 60 + time.time()) muted_until = int(int(i.group(2)) * 60 + time.time())
muted[username] = muted_until muted[username] = muted_until
emit("online", [online[request.referrer], muted], room=request.referrer, broadcast=True) emit("online", [online[g.referrer], muted], room=g.referrer, broadcast=True)
self_only = True self_only = True
if self_only or v.shadowbanned or execute_blackjack(v, None, text, "chat"): if self_only or v.shadowbanned or execute_blackjack(v, None, text, "chat"):
emit('speak', data) emit('speak', data)
else: else:
emit('speak', data, room=request.referrer, broadcast=True) emit('speak', data, room=g.referrer, broadcast=True)
messages[request.referrer][id] = data messages[g.referrer][id] = data
messages[request.referrer] = dict(list(messages[request.referrer].items())[-250:]) messages[g.referrer] = dict(list(messages[g.referrer].items())[-250:])
typing = [] typing = []
return '', 204 return '', 204
def refresh_online(): def refresh_online():
emit("online", [online[request.referrer], muted], room=request.referrer, broadcast=True) emit("online", [online[g.referrer], muted], room=g.referrer, broadcast=True)
cache.set('loggedin_chat', len(online[f'{SITE_FULL}/chat']), timeout=0) cache.set('loggedin_chat', len(online[f'{SITE_FULL}/chat']), timeout=0)
@socketio.on('connect') @socketio.on('connect')
@auth_required_socketio @auth_required_socketio
def connect(v): def connect(v):
if request.referrer not in ALLOWED_REFERRERS: if g.referrer not in ALLOWED_REFERRERS:
return '', 400 return '', 400
if request.referrer == f'{SITE_FULL}/notifications/messages': if g.referrer == f'{SITE_FULL}/notifications/messages':
join_room(v.id) join_room(v.id)
return '' return ''
join_room(request.referrer) join_room(g.referrer)
if [v.username, v.id, v.name_color, v.patron] not in online[request.referrer]: if [v.username, v.id, v.name_color, v.patron] not in online[g.referrer]:
for val in online.values(): for val in online.values():
if [v.username, v.id, v.name_color, v.patron] in val: if [v.username, v.id, v.name_color, v.patron] in val:
val.remove([v.username, v.id, v.name_color, v.patron]) val.remove([v.username, v.id, v.name_color, v.patron])
if [v.username, v.id, v.name_color, v.patron] not in online[request.referrer]: if [v.username, v.id, v.name_color, v.patron] not in online[g.referrer]:
online[request.referrer].append([v.username, v.id, v.name_color, v.patron]) online[g.referrer].append([v.username, v.id, v.name_color, v.patron])
refresh_online() refresh_online()
emit('typing', typing[request.referrer], room=request.referrer) emit('typing', typing[g.referrer], room=g.referrer)
return '', 204 return '', 204
@socketio.on('disconnect') @socketio.on('disconnect')
@auth_required_socketio @auth_required_socketio
def disconnect(v): def disconnect(v):
if request.referrer != f'{SITE_FULL}/notifications/messages': if g.referrer != f'{SITE_FULL}/notifications/messages':
for val in online.values(): for val in online.values():
if [v.username, v.id, v.name_color, v.patron] in val: if [v.username, v.id, v.name_color, v.patron] in val:
val.remove([v.username, v.id, v.name_color, v.patron]) val.remove([v.username, v.id, v.name_color, v.patron])
@ -226,12 +228,12 @@ def disconnect(v):
if v.username in val: if v.username in val:
val.remove(v.username) val.remove(v.username)
if request.referrer not in ALLOWED_REFERRERS: if g.referrer not in ALLOWED_REFERRERS:
return '', 400 return '', 400
elif request.referrer == f'{SITE_FULL}/notifications/messages': elif g.referrer == f'{SITE_FULL}/notifications/messages':
leave_room(v.id) leave_room(v.id)
else: else:
leave_room(request.referrer) leave_room(g.referrer)
refresh_online() refresh_online()
return '', 204 return '', 204
@ -239,30 +241,30 @@ def disconnect(v):
@socketio.on('typing') @socketio.on('typing')
@is_not_banned_socketio @is_not_banned_socketio
def typing_indicator(data, v): def typing_indicator(data, v):
if request.referrer not in ALLOWED_REFERRERS: if g.referrer not in ALLOWED_REFERRERS:
return '', 400 return '', 400
if data and v.username not in typing[request.referrer]: if data and v.username not in typing[g.referrer]:
typing[request.referrer].append(v.username) typing[g.referrer].append(v.username)
elif not data and v.username in typing[request.referrer]: elif not data and v.username in typing[g.referrer]:
typing[request.referrer].remove(v.username) typing[g.referrer].remove(v.username)
emit('typing', typing[request.referrer], room=request.referrer, broadcast=True) emit('typing', typing[g.referrer], room=g.referrer, broadcast=True)
return '', 204 return '', 204
@socketio.on('delete') @socketio.on('delete')
@admin_level_required(PERMS['POST_COMMENT_MODERATION']) @admin_level_required(PERMS['POST_COMMENT_MODERATION'])
def delete(id, v): def delete(id, v):
if request.referrer not in ALLOWED_REFERRERS: if g.referrer not in ALLOWED_REFERRERS:
return '', 400 return '', 400
for k, val in messages[request.referrer].items(): for k, val in messages[g.referrer].items():
if k == id: if k == id:
del messages[request.referrer][k] del messages[g.referrer][k]
break break
emit('delete', id, room=request.referrer, broadcast=True) emit('delete', id, room=g.referrer, broadcast=True)
return '', 204 return '', 204