Added extra query condition to account for values equal to 0. Removed…

… unused variable. Added extra query parameters
silverstripe · Feb 18, 2019 · d488361 · d488361
1 parent 89569f8
commit d488361
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Extensions/ElementalAreasExtension.php b/src/Extensions/ElementalAreasExtension.php
@@ -273,14 +273,14 @@ public function requireDefaultRecords()
         }
 
         $ownerClass = get_class($this->owner);
-        $tableName = $this->owner->getSchema()->tableName($ownerClass);
         $elementalAreas = $this->owner->getElementalRelations();
         $schema = $this->owner->getSchema();
 
         // There is no inbuilt filter for null values
         $where = [];
         foreach ($elementalAreas as $areaName) {
-            $where[] = $schema->sqlColumnForField($ownerClass, $areaName . 'ID') . ' IS NULL';
+            $queryDetails = $schema->sqlColumnForField($ownerClass, $areaName . 'ID');
+            $where[] = $queryDetails . ' IS NULL OR ' . $queryDetails . ' = 0' ;
         }
 
         foreach ($ownerClass::get()->where(implode(' OR ', $where)) as $elementalObject) {