Built in checks to see if table and column exist, closes #11

elgentos · Jul 11, 2018 · 4ba7438 · 4ba7438
1 parent 1d00ec9
commit 4ba7438
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/Elgentos/Masquerade/Console/RunCommand.php b/src/Elgentos/Masquerade/Console/RunCommand.php
@@ -112,6 +112,18 @@ protected function execute(InputInterface $input, OutputInterface $output)
      */
     private function fakeData($table)
     {
+        if (!$this->db->getSchemaBuilder()->hasTable($table['name'])) {
+            $this->output->writeln('Table ' . $table['name'] . ' does not exist.');
+            return;
+        }
+
+        foreach ($table['columns'] as $columnName => $columnData) {
+            if (!$this->db->getSchemaBuilder()->hasColumn($table['name'], $columnName)) {
+                unset($table['columns'][$columnName]);
+                $this->output->writeln('Column ' . $columnName . ' in table ' . $table['name'] . ' does not exist; skip it.');
+            }
+        }
+
         $this->output->writeln('');
         $this->output->writeln('Updating ' . $table['name']);