From 132538195818fb4e4b3924a34db2f1b40e7e96d6 Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Sun, 9 Jan 2022 17:24:09 +0200 Subject: [PATCH] under attack --- files/routes/admin.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/files/routes/admin.py b/files/routes/admin.py index 4fec5cb97..3ab6867de 100644 --- a/files/routes/admin.py +++ b/files/routes/admin.py @@ -397,6 +397,10 @@ def under_attack(v): g.db.add(ma) g.db.commit() data='{"value":"high"}' + + response = str(requests.patch(f'https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/settings/security_level', headers=CF_HEADERS, data=data)) + if response == "": return {"message": "Under attack mode disabled!"} + return {"error": "Failed to disable under attack mode."} else: f.write("yes") ma = ModAction( @@ -407,8 +411,9 @@ def under_attack(v): g.db.commit() data='{"value":"under_attack"}' - response = requests.patch(f'https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/settings/security_level', headers=CF_HEADERS, data=data) - return {"message": response.text} + response = str(requests.patch(f'https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/settings/security_level', headers=CF_HEADERS, data=data)) + if response == "": return {"message": "Under attack mode enabled!"} + return {"error": "Failed to enable under attack mode."} @app.get("/admin/badge_grant") @admin_level_required(2)