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

Administrator can now disable misbehaving servers

Browse Source
This commit is contained in:
Jeremy Zhang
2018-01-05 08:52:22 +00:00
parent 6c4b850669
commit bf47f9457c
14 changed files with 253 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
webapp/titanembeds/decorators.py
View File

@ -1,5 +1,6 @@
from functools import wraps
from flask import url_for, redirect, session, jsonify, abort
from flask import url_for, redirect, session, jsonify, abort, request
from titanembeds.database import list_disabled_guilds
def valid_session_required(api=False):
def decorator(f):
@ -26,3 +27,14 @@ def discord_users_only(api=False):
return f(*args, **kwargs)
return decorated_function
return decorator
def abort_if_guild_disabled():
def decorator(f):
@wraps(f)
def decorated_function(*args, **kwargs):
guild_id = request.args.get("guild_id", None)
if guild_id in list_disabled_guilds():
return ('', 423)
return f(*args, **kwargs)
return decorated_function
return decorator