fix(physical-restore): Use overriden Pewwee class

In pewee implementation in execute_sql function
if we pass params as null
It will still set a blank tupple

Underthehood mysql implementation will then trigger to
assume all % as escape params, and as will eventually fail

So, this override class, will pass the user provided params
directly to mysql client object

Author: tanmoysrt

agent/database_physical_backup.py

@@ -7,9 +7,9 @@
 import time
 from random import randint
 
-import peewee
 import requests
 
+from agent.database import CustomPeeweeDB
 from agent.database_server import DatabaseServer
 from agent.job import job, step
 from agent.utils import compute_file_hash, decode_mariadb_filename
@@ -31,7 +31,7 @@ def __init__(
         if not databases:
             raise ValueError("At least one database is required")
         # Instance variable for internal use
-        self._db_instances: dict[str, peewee.MySQLDatabase] = {}
+        self._db_instances: dict[str, CustomPeeweeDB] = {}
         self._db_tables_locked: dict[str, bool] = {db: False for db in databases}
 
         # variables
@@ -296,7 +296,7 @@ def _unlock_tables(self, db_name):
         the tables will be unlocked automatically
         """
 
-    def get_db(self, db_name: str) -> peewee.MySQLDatabase:
+    def get_db(self, db_name: str) -> CustomPeeweeDB:
         instance = self._db_instances.get(db_name, None)
         if instance is not None:
             if not instance.is_connection_usable():
@@ -306,7 +306,7 @@ def get_db(self, db_name: str) -> peewee.MySQLDatabase:
             return instance
         if db_name not in self.databases:
             raise ValueError(f"Database {db_name} not found")
-        self._db_instances[db_name] = peewee.MySQLDatabase(
+        self._db_instances[db_name] = CustomPeeweeDB(
             db_name,
             user=self.db_user,
             password=self.db_password,

agent/database_physical_restore.py

@@ -6,9 +6,8 @@
 import shutil
 import subprocess
 
-import peewee
-
 from agent.base import AgentException
+from agent.database import CustomPeeweeDB
 from agent.database_physical_backup import (
     DatabaseConnectionClosedWithDatabase,
     get_path_of_physical_backup_metadata,
@@ -34,8 +33,8 @@ def __init__(
         if tables_to_restore is None:
             tables_to_restore = []
 
-        self._target_db_instance: peewee.MySQLDatabase = None
-        self._target_db_instance_for_myisam: peewee.MySQLDatabase = None
+        self._target_db_instance: CustomPeeweeDB = None
+        self._target_db_instance_for_myisam: CustomPeeweeDB = None
         self.target_db = target_db
         self.target_db_user = "root"
         self.target_db_password = target_db_root_password
@@ -317,13 +316,13 @@ def _perform_file_operations(self, engine: str):
                 os.path.join(self.target_db_directory, file),
             )
 
-    def _get_target_db(self) -> peewee.MySQLDatabase:
+    def _get_target_db(self) -> CustomPeeweeDB:
         if self._target_db_instance is not None:
             if not self._target_db_instance.is_connection_usable():
                 raise DatabaseConnectionClosedWithDatabase()
             return self._target_db_instance
 
-        self._target_db_instance = peewee.MySQLDatabase(
+        self._target_db_instance = CustomPeeweeDB(
             self.target_db,
             user=self.target_db_user,
             password=self.target_db_password,
@@ -335,13 +334,13 @@ def _get_target_db(self) -> peewee.MySQLDatabase:
         self._target_db_instance.execute_sql("SET SESSION wait_timeout = 14400;")
         return self._target_db_instance
 
-    def _get_target_db_for_myisam(self) -> peewee.MySQLDatabase:
+    def _get_target_db_for_myisam(self) -> CustomPeeweeDB:
         if self._target_db_instance_for_myisam is not None:
             if not self._target_db_instance_for_myisam.is_connection_usable():
                 raise DatabaseConnectionClosedWithDatabase()
             return self._target_db_instance_for_myisam
 
-        self._target_db_instance_for_myisam = peewee.MySQLDatabase(
+        self._target_db_instance_for_myisam = CustomPeeweeDB(
             self.target_db,
             user=self.target_db_user,
             password=self.target_db_password,