Merge pull request #131 from a8cteam51/fix/check-aw-tables

Check for existence of AutomateWoo and Action Scheduler tables
a8cteam51 · Sep 12, 2024 · 8f5ecc1 · 8f5ecc1
2 parents 6fe540b + 2561aad
commit 8f5ecc1
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/includes/scrub-options.php b/includes/scrub-options.php
@@ -93,9 +93,17 @@ function( $v ) use ( $modules_to_disable ) {
 
 	// Disable AutomateWoo workflows, clear the queue, and set scheduled actions to "done"
 	$wpdb->query( "UPDATE $wpdb->posts SET post_status = 'aw-disabled' WHERE post_type = 'aw_workflow' AND post_status = 'publish'" );
-	$wpdb->query( "DELETE FROM {$wpdb->prefix}automatewoo_queue" );
-	$wpdb->query( "DELETE FROM {$wpdb->prefix}automatewoo_queue_meta" );
-	$wpdb->query( "UPDATE {$wpdb->prefix}actionscheduler_actions SET status = 'done' WHERE status = 'pending' AND hook LIKE '%automatewoo%'" );
+
+	$table_name = $wpdb->prefix . 'automatewoo_queue';
+	if ( $wpdb->get_var( $wpdb->prepare( 'SHOW TABLES LIKE %s', $table_name ) ) === $table_name ) {
+		$wpdb->query( "DELETE FROM {$wpdb->prefix}automatewoo_queue" );
+		$wpdb->query( "DELETE FROM {$wpdb->prefix}automatewoo_queue_meta" );
+	}
+
+	$table_name = $wpdb->prefix . 'actionscheduler_actions';
+	if ( $wpdb->get_var( $wpdb->prepare( 'SHOW TABLES LIKE %s', $table_name ) ) === $table_name ) {
+		$wpdb->query( "UPDATE {$wpdb->prefix}actionscheduler_actions SET status = 'done' WHERE status = 'pending' AND hook LIKE '%automatewoo%'" );
+	}
 
 	update_option( 'safety_net_options_scrubbed', true );