Added reset button

backend/modules/firegex.py

@@ -39,6 +39,7 @@ class FiregexTables:
         else: raise Exception(err)
     
     def init(self):
+        self.reset()
         code, out, err = self.raw_cmd({"create":{"table":{"name":self.table_name,"family":"inet"}}})
         if code == 0:
             self.cmd(
@@ -61,10 +62,13 @@ class FiregexTables:
                     "policy":"accept"
                 }}}
             )
-        self.reset()
+        
             
     def reset(self):
-        self.cmd({"flush":{"table":{"name":"firegex","family":"inet"}}})
+        self.raw_cmd(
+            {"flush":{"table":{"name":"firegex","family":"inet"}}},
+            {"delete":{"table":{"name":"firegex","family":"inet"}}},
+        )
 
     def list(self):
         return self.cmd({"list": {"ruleset": None}})["nftables"]

backend/modules/sqlite.py

@@ -50,18 +50,24 @@ class SQLite():
         self.conn.row_factory = dict_factory
 
     def backup(self):
-        if self.conn:
-            with open(self.db_name, "rb") as f:
-                self.__backup = f.read()
+        with open(self.db_name, "rb") as f:
+            self.__backup = f.read()
     
     def restore(self):
+        were_active = True if self.conn else False
+        self.disconnect()
         if self.__backup:
             with open(self.db_name, "wb") as f:
                 f.write(self.__backup)
             self.__backup = None
-
+        if were_active: self.connect()
+            
+    def delete_backup(self):
+        self.__backup = None
+    
     def disconnect(self) -> None:
         if self.conn: self.conn.close()
+        self.conn = None
 
     def create_schema(self, tables = {}) -> None:
         if self.conn: