Check for temp as well

Author: jaredmdobson

mysql_ch_replicator/clickhouse_api.py

@@ -117,17 +117,6 @@ def get_databases(self):
         database_list = [row[0] for row in databases]
         return database_list
 
-    def table_exists(self, table_name):
-        """Check if a specific table exists in the current database"""
-        if self.database not in self.get_databases():
-            return False
-            
-        query = f"EXISTS TABLE `{self.database}`.`{table_name}`"
-        result = self.client.query(query)
-        if result.result_rows and result.result_rows[0][0] == 1:
-            return True
-        return False
-
     def validate_database_schema(self, expected_tables):
         """Validates that all expected tables exist in the database and returns missing tables"""
         if self.database not in self.get_databases():

mysql_ch_replicator/db_replicator.py

@@ -180,7 +180,7 @@ def run(self):
 
             if self.state.status != Status.NONE:
                 # ensure target database still exists
-                if self.target_database not in self.clickhouse_api.get_databases():
+                if self.target_database not in self.clickhouse_api.get_databases() and f"{self.target_database}_tmp" not in self.clickhouse_api.get_databases():
                     logger.warning(f'database {self.target_database} missing in CH')
                     logger.warning('will run replication from scratch')
                     self.state.remove()