lib/enrichment_db/helper.rb in enrichment_db-0.1.5 vs lib/enrichment_db/helper.rb in enrichment_db-0.1.6

@@ -12,6 +12,25 @@
 	def self.hash_float_str_to_float(hash)
 		hash.each_with_object({}) do |(k, v), h|
 			h[k] = float_str_to_float(v)
 		end
 	end
+
+  def self.format_geo_result(result)
+    if result.size == 1 
+      puts "Found object"
+      result = result.first
+  	  id = result['region_id']
+      result = EnrichmentDb::Helper.hash_float_str_to_float(result)
+      region_type = result['region_type']
+      region_type = region_type[0..-2]
+      region = EnrichmentDb::Geo.const_get(region_type).by_id(id.to_i)
+      region = EnrichmentDb::Helper.hash_float_str_to_float(region)
+      result['region'] = region
+      result
+    elsif result.size > 1
+      puts "More than 1 object found. Only wanted 1 object."
+    else 
+      puts "Nothing found"
+    end
+  end
 end
\ No newline at end of file