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

sneed

master
Aevann1 2022-08-24 16:58:58 +02:00
parent d84b26fc77
commit 9773930d48
1 changed files with 10 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
files/helpers/wrappers.py
View File

@ -76,12 +76,16 @@ def get_logged_in_user():
if v: v.poor = session.get('poor')
if AEVANN_ID and request.headers.get("Cf-Ipcountry") == 'EG' and not (v and v.id == AEVANN_ID):
with open(f"/eg", "r+", encoding="utf-8") as f:
ip = request.headers.get('CF-Connecting-IP')
if f'{v}, {ip}' not in f.read():
t = str(time.strftime("%d/%B/%Y %H:%M:%S UTC", time.gmtime(time.time())))
f.write(f'{f.read()}{v}, {ip}, {t}\n')
if AEVANN_ID and request.headers.get("Cf-Ipcountry") == 'EG':
if v and v.id != AEVANN_ID:
with open(f"/eg", "r+", encoding="utf-8") as f:
ip = request.headers.get('CF-Connecting-IP')
if f'@{v.username}, ' not in f.read():
t = str(time.strftime("%d/%B/%Y %H:%M:%S UTC", time.gmtime(time.time())))
f.write(f'{f.read()}@{v.username}, {ip}, {t}\n')
elif not v and request.path not in ('/login','/signup'):
abort(401)
return v
def check_ban_evade(v):
@ -124,24 +128,6 @@ def auth_required(f):
wrapper.__name__ = f.__name__
return wrapper
def auth_trusted_server(func):
@functools.wraps(func)
def inner(*args, **kwargs):
if not TRUSTED_SERVER_PSK: abort(401)
auth = request.headers.get("Authorization", None)
if not auth: abort(401)
auth_words = auth.split(' ')
if len(auth_words) != 2 or auth_words[0] != 'TrustedServer':
abort(401)
if not auth_words[1] == TRUSTED_SERVER_PSK:
abort(403)
return make_response(func(*args, **kwargs))
return inner
def is_not_permabanned(f):
def wrapper(*args, **kwargs):