1 files changed, 7 insertions, 0 deletions
diff --git a/day9/task5_vue/backend/database/wrappers.py b/day9/task5_vue/backend/database/wrappers.py
index 30d34d9..5bb9761 100644
--- a/day9/task5_vue/backend/database/wrappers.py
+++ b/day9/task5_vue/backend/database/wrappers.py
@@ -8,6 +8,10 @@ class Wrapper(ABC):
         self.schemes = {}
 
     @abstractmethod
+    def close_connection(self):
+        pass
+
+    @abstractmethod
     def clear_table(self, table_name):
         pass
 
@@ -46,6 +50,9 @@ class MySQLWrapper(Wrapper):
             db=db_name
         )
 
+    def close_connection(self):
+        self.connection.close()
+
     def clear_table(self, table_name):
         cursor = self.connection.cursor()
         cursor.execute(f"START TRANSACTION; DELETE FROM `{table_name}`; COMMIT;")