Merge pull request #63 from jsirish/feature/import

Import/Export Sync
dynamic · Nov 10, 2015 · fa7f029 · fa7f029
2 parents dd431b2 + d00c341
commit fa7f029
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 36 deletions.
diff --git a/code/LocationAdmin.php b/code/LocationAdmin.php
@@ -2,36 +2,41 @@
 
 class LocationAdmin extends ModelAdmin
 {
-
     private static $managed_models = array(
         'Location',
-        'LocationCategory'
+        'LocationCategory',
     );
 
     private static $model_importers = array(
         'Location' => 'LocationCsvBulkLoader',
-        'LocationCategory' => 'CsvBulkLoader'
+        'LocationCategory' => 'CsvBulkLoader',
     );
 
     private static $menu_title = 'Locator';
     private static $url_segment = 'locator';
 
     public function getExportFields()
     {
-        return array(
-            'Title' => 'Name',
-            'Address' => 'Address',
-            'Suburb' => 'City',
-            'State' => 'State',
-            'Postcode' => 'Postal Code',
-            'Country' => 'Country',
-            'Website' => 'Website',
-            'Phone' => 'Phone',
-            'Fax' => 'Fax',
-            'EmailAddress' => 'Email Address',
-            'ShowInLocator' => 'Show',
-            'Lat' => 'Lat',
-            'Lng' => 'Lng'
-        );
+        if ($this->modelClass == 'Location') {
+            return array(
+                'Title' => 'Name',
+                'Address' => 'Address',
+                'Suburb' => 'City',
+                'State' => 'State',
+                'Postcode' => 'Postcode',
+                'Country' => 'Country',
+                'Website' => 'Website',
+                'Phone' => 'Phone',
+                'Fax' => 'Fax',
+                'EmailAddress' => 'EmailAddress',
+                'Category.Name' => 'Category',
+                'ShowInLocator' => 'ShowInLocator',
+                'Featured' => 'Featured',
+                'Lat' => 'Lat',
+                'Lng' => 'Lng',
+            );
+        }
+
+        return parent::getExportFields();
     }
-}
+}
diff --git a/code/LocationCsvBulkLoader.php b/code/LocationCsvBulkLoader.php
@@ -4,30 +4,27 @@ class LocationCsvBulkLoader extends CsvBulkLoader
 {
     public $columnMap = array(
         'Name' => 'Title',
-        //'Address' => 'Address',
         'City' => 'Suburb',
-        //'State' => 'State',
-        'Zip' => 'Postcode',
         'Category' => 'Category.Name',
-        'Email' => 'EmailAddress'
-    );
-
-    public $duplicateChecks = array(
-        'Address' => 'Address',
-        //'Website' => 'Website'
     );
 
     public $relationCallbacks = array(
-        'Category.Name' => array(
-            'relationname' => 'Category',
-            'callback' => 'getCatByName'
-        )
+       'Category.Name' => array(
+           'relationname' => 'Category',
+           'callback' => 'getCategoryByName',
+        ),
     );
 
-    public static function getCatByName(&$obj, $val, $record)
+    public static function getCategoryByName(&$obj, $val, $record)
     {
         $val = Convert::raw2sql($val);
-        return LocationCategory::get()->filter('Name', $val)->First();
-        //);
+        $category = LocationCategory::get()->filter(array('Name' => $val))->First();
+        if (!$category) {
+            $category = LocationCategory::create();
+            $category->Name = $val;
+            $category->write();
+        }
+
+        return $category;
     }
-}
+}