check if record hasMethod getLocalisedTables before running

arillo · Jan 23, 2020 · 610bd23 · 610bd23
1 parent 68c9308
commit 610bd23
Showing 1 changed file with 18 additions and 16 deletions.
diff --git a/src/FluentHelper.php b/src/FluentHelper.php
@@ -28,25 +28,27 @@ public static function force_delete(
         DataObject $record,
         $deleteVersioned = true
     ) {
-        foreach ($record->getLocalisedTables() as $table => $value)
-        {
-            $tables = [
-                $table . '_' . FluentExtension::SUFFIX,
-            ];
-
-            if ($deleteVersioned)
+        if($record->hasMethod('getLocalisedTables')){
+            foreach ($record->getLocalisedTables() as $table => $value)
             {
-                $tables[] = $table . '_' . FluentExtension::SUFFIX . '_Live';
-            }
+                $tables = [
+                    $table . '_' . FluentExtension::SUFFIX,
+                ];
 
-            foreach ($tables as $deleteTable)
-            {
-                SQLDelete::create("\"{$deleteTable}\"")
-                    ->addWhere(["\"{$deleteTable}\".\"RecordID\"" => $record->ID])
-                    ->execute()
-                ;
+                if ($deleteVersioned)
+                {
+                    $tables[] = $table . '_' . FluentExtension::SUFFIX . '_Live';
+                }
+
+                foreach ($tables as $deleteTable)
+                {
+                    SQLDelete::create("\"{$deleteTable}\"")
+                        ->addWhere(["\"{$deleteTable}\".\"RecordID\"" => $record->ID])
+                        ->execute()
+                    ;
+                }
             }
+            return $record;
         }
-        return $record;
     }
 }