refactoring: firewall model changed using ufw model (due to docker iussues)

backend/modules/firewall/models.py

@@ -1,7 +1,9 @@
 from enum import Enum
+from utils import PortType
+from pydantic import BaseModel
 
 class Rule:
-    def __init__(self, proto: str, src:str, dst:str, port_src_from:str, port_dst_from:str, port_src_to:str, port_dst_to:str, action:str, mode:str):
+    def __init__(self, proto: str, src:str, dst:str, port_src_from:str, port_dst_from:str, port_src_to:str, port_dst_to:str, action:str, mode:str, **other):
         self.proto = proto
         self.src = src
         self.dst = dst
@@ -13,21 +15,10 @@ class Rule:
         self.input_mode = mode == "in"
         self.output_mode = mode == "out"
         self.forward_mode = mode == "forward"
-        
     
     @classmethod
     def from_dict(cls, var: dict):
-        return cls(
-            proto=var["proto"],
-            src=var["src"],
-            dst=var["dst"],
-            port_dst_from=var["port_dst_from"],
-            port_dst_to=var["port_dst_to"],
-            port_src_from=var["port_src_from"],
-            port_src_to=var["port_src_to"],
-            action=var["action"],
-            mode=var["mode"]
-        )
+        return cls(**var)
 
 class Protocol(str, Enum):
     TCP = "tcp",
@@ -44,4 +35,37 @@ class Mode(str, Enum):
 class Action(str, Enum):
     ACCEPT = "accept",
     DROP = "drop",
-    REJECT = "reject"
+    REJECT = "reject"
+    
+class RuleModel(BaseModel):
+    active: bool
+    name: str
+    proto: Protocol
+    src: str
+    dst: str
+    port_src_from: PortType
+    port_dst_from: PortType
+    port_src_to: PortType
+    port_dst_to: PortType
+    action: Action
+    mode:Mode
+
+class RuleFormAdd(BaseModel):
+    rules: list[RuleModel]
+    policy: Action
+    
+class RuleInfo(BaseModel):
+    rules: list[RuleModel]
+    policy: Action
+    enabled: bool
+
+class RenameForm(BaseModel):
+    name:str
+
+class FirewallSettings(BaseModel):
+    keep_rules: bool
+    allow_loopback: bool
+    allow_established: bool
+    allow_icmp: bool
+    multicast_dns: bool
+    allow_upnp: bool