ENHANCEMENT allow for setting location query class at runtime

dynamic · Oct 12, 2016 · 99c5024 · 99c5024
1 parent a0e23a9
commit 99c5024
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/code/pages/Locator.php b/code/pages/Locator.php
@@ -47,6 +47,11 @@ class Locator extends Page
      */
     private static $description = 'Find locations on a map';
 
+    /**
+     * @var string
+     */
+    private static $location_class = 'Location';
+
     /**
      * @return FieldList
      */
@@ -98,7 +103,8 @@ public static function get_locations(
         $callback = null
     )
     {
-        $locations = Location::get()->filter($filter)->exclude($exclude);
+        $locationClass = Config::inst()->get('Locator', 'location_class');
+        $locations = $locationClass::get()->filter($filter)->exclude($exclude);
 
         if (!empty($filterAny)) {
             $locations = $locations->filterAny($filterAny);
@@ -196,7 +202,7 @@ class Locator_Controller extends Page_Controller
     /**
      * @var DataList|ArrayList
      */
-    private $locations;
+    protected $locations;
 
     /**
      * Set Requirements based on input from CMS