lib/tasks/data_tasks.rake in unidom-geo-china-0.1 vs lib/tasks/data_tasks.rake in unidom-geo-china-0.2

@@ -2,61 +2,104 @@
 
 namespace :unidom do
   namespace :geo do
     namespace :china do
 
-      # bundle exec rake unidom:geo:china:import
-      #   file=/data.csv
-      #   from_date=2020-01-01
-      #   scheme_id=
-      #   scheme_type=
-      task import: :environment do
+      namespace :region do
 
-        include ::Unidom::Common::DataHelper
+        # bundle exec rake unidom:geo:china:region:import
+        #   file=/data.csv
+        #   from_date=2020-01-01
+        #   scheme_id=
+        #   scheme_type=
+        task import: :environment do
 
-        file_name   = ENV['file']
-        scheme_id   = ENV['scheme_id']||::Unidom::Common::NULL_UUID
-        scheme_type = ENV['scheme_type']||''
-        opened_at   = parse_time ENV['from_date']
+          include ::Unidom::Common::DataHelper
 
-        updated_count = 0
-        created_count = 0
+          file_name   = ENV['file']
+          scheme_id   = ENV['scheme_id']||::Unidom::Common::NULL_UUID
+          scheme_type = ENV['scheme_type']||''
+          opened_at   = parse_time ENV['from_date']
 
-        region_entities = ::Unidom::Geo::China::Region.scheme_id_is(scheme_id).scheme_type_is(scheme_type).select('id, name, virtual, numeric_code, alphabetic_code, scheme_id, scheme_type, opened_at, closed_at, defunct, updated_at').to_a
+          updated_count = 0
+          created_count = 0
 
-        each_csv_row file_name do |region|
+          region_entities = ::Unidom::Geo::China::Region.scheme_id_is(scheme_id).scheme_type_is(scheme_type).select('id, name, virtual, numeric_code, alphabetic_code, scheme_id, scheme_type, opened_at, closed_at, defunct, updated_at').to_a
 
-          numeric_code    = region['numeric_code']
-          alphabetic_code = region['alphabetic_code']
+          each_csv_row file_name do |region|
 
-          attributes = { name: region['name'], virtual: region['virtual'], scheme_id: scheme_id, scheme_type: scheme_type, opened_at: opened_at }
-          attributes[:alphabetic_code] = alphabetic_code if alphabetic_code.present?
+            numeric_code    = region['numeric_code']
+            alphabetic_code = region['alphabetic_code']
 
-          if region_entities.present?
-            found_region_entities = region_entities.select { |region_entity| region_entity.numeric_code==numeric_code }
-            if found_region_entities.present?
-              found_region_entities.each do |found_region_entity|
-                found_region_entity.assign_attributes attributes
-                if found_region_entity.changed?
-                  found_region_entity.save!
-                  updated_count += 1
+            attributes = { name: region['name'], virtual: region['virtual'], scheme_id: scheme_id, scheme_type: scheme_type, opened_at: opened_at }
+            attributes[:alphabetic_code] = alphabetic_code if alphabetic_code.present?
+
+            if region_entities.present?
+              found_region_entities = region_entities.select { |region_entity| region_entity.numeric_code==numeric_code }
+              if found_region_entities.present?
+                found_region_entities.each do |found_region_entity|
+                  found_region_entity.assign_attributes attributes
+                  if found_region_entity.changed?
+                    found_region_entity.save!
+                    updated_count += 1
+                  end
                 end
+              else
+                attributes[:numeric_code] = numeric_code
+                ::Unidom::Geo::China::Region.create! attributes
+                created_count += 1
               end
             else
               attributes[:numeric_code] = numeric_code
               ::Unidom::Geo::China::Region.create! attributes
               created_count += 1
             end
-          else
-            attributes[:numeric_code] = numeric_code
-            ::Unidom::Geo::China::Region.create! attributes
-            created_count += 1
+
           end
 
+          puts "#{created_count} China Regions were created. #{updated_count} China Regions were updated per CSV."
+          puts "#{created_count+updated_count} China Regions were handled totally."
+
         end
 
-        puts "#{created_count} China Regions were created. #{updated_count} China Regions were updated per CSV."
-        puts "#{created_count+updated_count} China Regions were handled totally."
+      end
+
+      namespace :town do
+
+        # bundle exec rake unidom:geo:china:town:import
+        #   file=/data.csv
+        #   from_date=2020-01-01
+        task import: :environment do
+
+          include ::Unidom::Common::DataHelper
+
+          file_name   = ENV['file']
+          opened_at   = parse_time ENV['from_date']
+
+          updated_count = 0
+          created_count = 0
+
+          each_csv_row file_name do |region|
+
+            numeric_code = region['numeric_code']
+
+            town   = Unidom::Geo::China::Town.numeric_coded_as(numeric_code).first
+            county = Unidom::Geo::China::Region.numeric_coded_as(numeric_code[0..5]).valid_at.alive.first
+            if town.present?
+              town.assign_attributes region: county, name: region['name'], virtual: region['virtual']
+              town.save!
+              updated_count += 1
+            else
+              town = Unidom::Geo::China::Town.create! region: county, numeric_code: numeric_code, name: region['name'], virtual: region['virtual'], opened_at: opened_at
+              created_count += 1
+            end
+
+          end
+
+          puts "#{created_count} China Regions were created. #{updated_count} China Regions were updated per CSV."
+          puts "#{created_count+updated_count} China Regions were handled totally."
+
+        end
 
       end
 
     end
   end