linted code

src/firewall/fwtype.py

@@ -6,7 +6,7 @@ class FWType(ABC):
     """Abstract base class for firewall types."""
 
     # Registry to store child classes
-    _registry: Dict[str, Type['FWType']] = {}
+    _registry: Dict[str, Type["FWType"]] = {}
 
     def __init_subclass__(cls, **kwargs):
         """Automatically register subclasses with their class name."""
@@ -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.
 
@@ -30,11 +30,13 @@ class FWType(ABC):
         """
         fw_type = fw_type.lower()
         if fw_type not in cls._registry:
-            available = ', '.join(cls._registry.keys())
-            raise ValueError(f"Unknown firewall type: '{fw_type}'. Available: {available}")
+            available = ", ".join(cls._registry.keys())
+            raise ValueError(
+                f"Unknown firewall type: '{fw_type}'. Available: {available}"
+            )
 
         return cls._registry[fw_type](**kwargs)
 
     @abstractmethod
-    def getBanlist(self,ips):
+    def getBanlist(self, ips):
         """Return the ruleset for the specific server"""

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])

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.