fix: Fix parse error when data is missing (#77)

dwilkie · May 13, 2024 · e3aab0b · e3aab0b
1 parent 9602657
commit e3aab0b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/pumi/data_source/ncdd.rb b/lib/pumi/data_source/ncdd.rb
@@ -10,6 +10,9 @@ module Pumi
   module DataSource
     class NCDD
       CSV_HEADERS = %w[type code name_km name_latin reference note1 note2].freeze
+      MISSING_DATA = {
+        "1715" => { type: "ក្រុង" }
+      }.freeze
 
       AdministrativeUnit = Struct.new(:en, :km, :latin, :code_length, :group, :type, keyword_init: true)
       Row = Struct.new(:code, :name_km, :name_latin, :type, keyword_init: true) do
@@ -61,11 +64,13 @@ def parse_source_file(file)
       end
 
       def build_row(row)
+        code = parse_location_code(row)
+
         Row.new(
-          code: parse_location_code(row),
+          code:,
           name_km: row.fetch("name_km"),
           name_latin: row.fetch("name_latin"),
-          type: row.fetch("type")
+          type: row.fetch("type") || MISSING_DATA.dig(code, :type)
         )
       end