code/Task/UserFormsColumnCleanTask.php

@@ -38,32 +38,32 @@ class UserFormsColumnCleanTask extends BuildTask
         $schema = DataObject::getSchema();
 
         foreach ($this->tables as $db) {
-            $table = $schema->tableName($db);
             $columns = $schema->databaseFields($db);
-            $query = "SHOW COLUMNS FROM $table";
+            $query = "SHOW COLUMNS FROM $db";
             $liveColumns = DB::query($query)->column();
-            $query = "SHOW TABLES LIKE 'Backup_$table'";
+            $backedUp = 0;
+            $query = "SHOW TABLES LIKE 'Backup_$db'";
             $tableExists = DB::query($query)->value();
             if ($tableExists != null) {
-                $output->writeln("Tasks run already on $table exiting");
+                $output->writeln("Tasks run already on $db exiting");
                 return Command::SUCCESS;
             }
-            $backedUp = false;
-            foreach ($liveColumns as $column) {
-                if (!$backedUp) {
-                    $output->writeln("Backing up $table <br />");
-                    $output->writeln("Creating Backup_$table <br />");
+            $backedUp = 0;
+            foreach ($liveColumns as $index => $column) {
+                if ($backedUp == 0) {
+                    $output->writeln("Backing up $db <br />");
+                    $output->writeln("Creating Backup_$db <br />");
                     // backup table
-                    $query = "CREATE TABLE Backup_$table LIKE $table";
+                    $query = "CREATE TABLE Backup_$db LIKE $db";
                     DB::query($query);
-                    $output->writeln("Populating Backup_$table <br />");
-                    $query = "INSERT Backup_$table SELECT * FROM $table";
+                    $output->writeln("Populating Backup_$db <br />");
+                    $query = "INSERT Backup_$db SELECT * FROM $db";
                     DB::query($query);
-                    $backedUp = true;
+                    $backedUp = 1;
                 }
                 if (!isset($columns[$column]) && !in_array($column, $this->keepColumns ?? [])) {
-                    $output->writeln("Dropping $column from $table <br />");
-                    $query = "ALTER TABLE $table DROP COLUMN $column";
+                    $output->writeln("Dropping $column from $db <br />");
+                    $query = "ALTER TABLE $db DROP COLUMN $column";
                     DB::query($query);
                 }
             }