synthesize hierarchy when there aren't any

pelias · Sep 21, 2017 · 29cd09d · 29cd09d
1 parent 6bedda2
commit 29cd09d
Show file tree

Hide file tree

Showing 3 changed files with 164 additions and 35 deletions.
diff --git a/src/components/extractFields.js b/src/components/extractFields.js
@@ -76,6 +76,19 @@ function getAbbreviation(properties) {
   return properties['wof:abbreviation'];
 }
 
+function getHierarchies(id, properties) {
+  if (_.isEmpty(_.get(properties, 'wof:hierarchy')) && _.has(properties, 'wof:placetype')) {
+    const hierarchy = {};
+    hierarchy[properties['wof:placetype'] + '_id'] = id;
+
+    return [hierarchy];
+
+  }
+
+  return _.defaultTo(properties['wof:hierarchy'], []);
+
+}
+
 /*
   This function extracts the fields from the json_object that we're interested
   in for creating Pelias Document objects.  If there is no hierarchy then a
@@ -94,9 +107,12 @@ module.exports.create = function map_fields_stream() {
       bounding_box: getBoundingBox(json_object.properties),
       population: getPopulation(json_object.properties),
       popularity: json_object.properties['misc:photo_sum'],
-      hierarchies: _.get(json_object, 'properties.wof:hierarchy', [])
+      hierarchies: getHierarchies(json_object.id, json_object.properties)
+      // hierarchies: _.get(json_object, 'properties.wof:hierarchy', [])
     };
 
+    // console.error(record.hierarchies);
+
     // use the QS altname if US county and available
     if (isUsCounty(record, json_object.properties['wof:country'], json_object.properties['qs:a2_alt'])) {
       record.name = json_object.properties['qs:a2_alt'];