From 831497f80ac45330f955418053d617f580578c0e Mon Sep 17 00:00:00 2001
From: carnivuth <matti200042@gmail.com>
Date: Tue, 27 Jan 2026 17:53:11 +0100
Subject: [PATCH] linted code

---
 src/firewall/fwtype.py   |  2 +-
 src/firewall/iptables.py |  9 +++------
 src/firewall/raw.py      |  3 ++-
 src/handler.py           | 20 +++++---------------
 4 files changed, 11 insertions(+), 23 deletions(-)

diff --git a/src/firewall/fwtype.py b/src/firewall/fwtype.py
index 8617b4d..0e0e421 100644
--- a/src/firewall/fwtype.py
+++ b/src/firewall/fwtype.py
@@ -14,7 +14,7 @@ class FWType(ABC):
         cls._registry[cls.__name__.lower()] = cls
 
     @classmethod
-    def create(cls, fw_type: str, **kwargs) -> 'FWType':
+    def create(cls, fw_type: str, **kwargs) -> "FWType":
         """
         Factory method to create instances of child classes.
 
diff --git a/src/firewall/iptables.py b/src/firewall/iptables.py
index 73ac623..159171e 100644
--- a/src/firewall/iptables.py
+++ b/src/firewall/iptables.py
@@ -1,10 +1,11 @@
 from typing_extensions import override
 from firewall.fwtype import FWType
 
+
 class Iptables(FWType):
 
     @override
-    def getBanlist(self,ips) -> str:
+    def getBanlist(self, ips) -> str:
         """
         Generate iptables ban rules from an array of IP addresses.
 
@@ -29,11 +30,7 @@ class Iptables(FWType):
             ip = ip.strip()
 
             # Build the iptables command
-            rule_parts = [
-                "iptables",
-                "-A", chain,
-                "-s", ip
-            ]
+            rule_parts = ["iptables", "-A", chain, "-s", ip]
 
             # Add target
             rule_parts.extend(["-j", target])
diff --git a/src/firewall/raw.py b/src/firewall/raw.py
index 1e29e75..e0c82fe 100644
--- a/src/firewall/raw.py
+++ b/src/firewall/raw.py
@@ -1,10 +1,11 @@
 from typing_extensions import override
 from firewall.fwtype import FWType
 
+
 class Raw(FWType):
 
     @override
-    def getBanlist(self,ips) -> str:
+    def getBanlist(self, ips) -> str:
         """
         Generate raw list of bad IP addresses.
 
diff --git a/src/handler.py b/src/handler.py
index f780835..07c31b5 100644
--- a/src/handler.py
+++ b/src/handler.py
@@ -12,12 +12,6 @@ import os
 
 from database import get_database
 from config import Config, get_config
-from firewall.fwtype import FWType
-
-# imports for the __init_subclass__ method, do not remove pls
-from firewall.iptables import Iptables
-from firewall.raw import Raw
-
 
 from database import get_database
 from config import Config,get_config
@@ -947,19 +941,12 @@ class Handler(BaseHTTPRequestHandler):
 
         # API endpoint for downloading malicious IPs blocklist file
         if (
-            self.config.dashboard_secret_path
-            and request_path == f"{self.config.dashboard_secret_path}/api/get_banlist"
-        ):
-
-            # get fwtype from request params
-            fwtype = query_params.get("fwtype", ["iptables"])[0]
-
             self.config.dashboard_secret_path and
             request_path == f"{self.config.dashboard_secret_path}/api/get_banlist"
         ):
 
             # get fwtype from request params
-            fwtype = query_params.get("fwtype",["iptables"])[0]
+            fwtype = query_params.get("fwtype", ["iptables"])[0]
 
             # Query distinct suspicious IPs
             results = (
@@ -979,7 +966,10 @@ class Handler(BaseHTTPRequestHandler):
 
             self.send_response(200)
             self.send_header("Content-type", "text/plain")
-            self.send_header("Content-Disposition", f'attachment; filename="{fwtype}.txt"',)
+            self.send_header(
+                "Content-Disposition",
+                f'attachment; filename="{fwtype}.txt"',
+            )
             self.send_header("Content-Length", str(len(banlist)))
             self.end_headers()
             self.wfile.write(banlist.encode())