Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2025-06-16 11:25:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

Check for is peak before connecting to gateway

Browse Source
This commit is contained in:
Jeremy Zhang
2018-11-17 20:33:39 +00:00
parent a9d41cdb28
commit 140083b9c8
3 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
webapp/titanembeds/blueprints/embed/embed.py
View File

@ -1,6 +1,6 @@
from flask import Blueprint, render_template, abort, redirect, url_for, session, request
from flask_babel import gettext
from titanembeds.utils import check_guild_existance, guild_query_unauth_users_bool, guild_accepts_visitors, guild_unauthcaptcha_enabled, is_int, redisqueue
from titanembeds.utils import check_guild_existance, guild_query_unauth_users_bool, guild_accepts_visitors, guild_unauthcaptcha_enabled, is_int, redisqueue, get_online_embed_user_keys
from titanembeds.oauth import generate_guild_icon_url, generate_avatar_url
from titanembeds.database import db, Guilds, UserCSS, list_disabled_guilds
from config import config
@ -60,6 +60,10 @@ def parse_url_domain(url):
if parsed.netloc != "":
return parsed.netloc
return url
def is_peak(guild_id):
usrs = get_online_embed_user_keys(guild_id)
return (len(usrs["AuthenticatedUsers"]) + len(usrs["UnauthenticatedUsers"])) > 10
@embed.route("/<int:guild_id>")
def guild_embed(guild_id):
@ -93,7 +97,8 @@ def guild_embed(guild_id):
cssvariables=parse_css_variable(customcss),
same_target=request.args.get("sametarget", False) == "true",
userscalable=request.args.get("userscalable", "True").lower().startswith("t"),
fixed_sidenav=request.args.get("fixedsidenav", "False").lower().startswith("t")
fixed_sidenav=request.args.get("fixedsidenav", "False").lower().startswith("t"),
is_peak=is_peak(guild_id)
)
abort(404)