diff --git a/app/scripts/script1.py b/app/scripts/script1.py index a496e12..6c88cbb 100644 --- a/app/scripts/script1.py +++ b/app/scripts/script1.py @@ -7,7 +7,7 @@ from prompt_toolkit import prompt import logging import json import urllib3 - +import re urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) logging.basicConfig( level=logging.INFO, @@ -76,7 +76,7 @@ def ban_ip(args): ips_to_add.append(ip) else: - ips_to_add = args.ips.split(",") + ips_to_add = re.split(r'[, :]+', args.ips) ips = list(dict.fromkeys(ips + ips_to_add)) update_url = f"https://unifi.sectorq.eu/proxy/network/api/s/default/rest/firewallgroup/6782dbc2ffa79454f61fac43"