diff --git a/app/scripts/script1.py b/app/scripts/script1.py index d2e9a9f..559b56e 100644 --- a/app/scripts/script1.py +++ b/app/scripts/script1.py @@ -10,6 +10,7 @@ class LoginError(Exception): def get_token(args): """Get CSRF token and cookies""" + print("Logging in to Unifi Controller...") session = requests.Session() login_url = "https://unifi.sectorq.eu/api/auth/login" payload_login = {"username": "jaydee", "password": "l4c1j4yd33Du5lo"} @@ -84,8 +85,8 @@ def ban_ip(args): def run(data): - ban_ip(data) - - - return f"{data.action} requested {data.ip}" + try: + ban_ip(data) + except: + return f"{data.action} requested {data.ip}"