push: code changes

backend/modules/nfproxy/models.py

@@ -15,13 +15,19 @@ class Service:
 
 
 class PyFilter:
-    def __init__(self, filter_id:int, name: str, blocked_packets: int, edited_packets: int, active: bool, **other):
-        self.id = filter_id
+    def __init__(self, name: str, blocked_packets: int, edited_packets: int, active: bool, db, **other):
         self.name = name
         self.blocked_packets = blocked_packets
         self.edited_packets = edited_packets
         self.active = active
+        self.__db = db
+
+    async def update(self):
+        self.__db.query("UPDATE pyfilter SET blocked_packets = ?, edited_packets = ? WHERE name = ?;", self.blocked_packets, self.edited_packets, self.name)
+
+    def __repr__(self):
+        return f"<PyFilter {self.name}>"
 
     @classmethod
-    def from_dict(cls, var: dict):
-        return cls(**var)
+    def from_dict(cls, var: dict, db):
+        return cls(**var, db=db)