TPI Production Ready Seeds

app/admin/cp_assessments.rb

@@ -48,6 +48,7 @@
     year_columns = collection.flat_map(&:emissions_all_years).uniq.sort
 
     column :id
+    column(:company_id) { |a| a.company.id }
     column(:company) { |a| a.company.name }
     column :assessment_date
     column :publication_date, &:publication_date_csv

app/admin/instrument.rb

@@ -4,6 +4,7 @@
   permit_params :name, :instrument_type_id
 
   config.batch_actions = false
+  config.sort_order = 'name_asc'
 
   decorate_with InstrumentDecorator
 

app/admin/laws_sectors.rb

@@ -1,5 +1,6 @@
 ActiveAdmin.register LawsSector do
   config.batch_actions = false
+  config.sort_order = 'name_asc'
 
   menu parent: 'Laws', priority: 5
 

app/admin/mq_assessments.rb

@@ -75,6 +75,7 @@
 
   csv do
     column :id
+    column(:company_id) { |a| a.company.id }
     column(:company) { |a| a.company.name }
     column :assessment_date
     column :publication_date, &:publication_date_csv

app/models/company.rb

@@ -46,7 +46,7 @@ class Company < ApplicationRecord
 
   validates :ca100, inclusion: {in: [true, false]}
   validates_presence_of :name, :slug, :isin, :market_cap_group
-  validates_uniqueness_of :slug, :isin, :name
+  validates_uniqueness_of :slug, :name
 
   def to_s
     name

app/services/csv_import/companies.rb

@@ -24,6 +24,8 @@ def resource_klass
     end
 
     def prepare_company(row)
+      return prepare_overridden_resource(row) if override_id
+
       find_record_by(:id, row) ||
         find_record_by(:isin, row) ||
         find_record_by(:name, row) ||
@@ -40,10 +42,13 @@ def company_attributes(row)
         name: row[:name],
         isin: row[:isin],
         sector: find_or_create_tpi_sector(row[:sector]),
-        market_cap_group: row[:market_cap_group],
+        market_cap_group: row[:market_cap_group].downcase,
         geography: geographies[row[:geography_iso]],
         headquarters_geography: geographies[row[:headquarters_geography_iso]],
         ca100: row[:ca100],
+        sedol: row[:sedol],
+        latest_information: row[:latest_information],
+        historical_comments: row[:historical_comments],
         visibility_status: row[:visibility_status]
       }
     end

app/services/csv_import/cp_assessments.rb

@@ -31,9 +31,9 @@ def prepare_assessment(row)
     end
 
     def find_company!(row)
-      return unless row[:company].present?
+      return unless row[:company_id].present?
 
-      Company.where('lower(name) = ?', row[:company].downcase).first!
+      Company.find(row[:company_id])
     end
 
     def assessment_attributes(row)
@@ -47,7 +47,7 @@ def assessment_attributes(row)
     end
 
     def assessment_date(row)
-      CSVImport::DateUtils.safe_parse(row[:assessment_date], ['%Y-%m-%d'])
+      CSVImport::DateUtils.safe_parse(row[:assessment_date], ['%Y-%m-%d']) if row[:assessment_date]
     end
 
     def publication_date(row)

app/services/csv_import/mq_assessments.rb

@@ -41,9 +41,9 @@ def prepare_assessment(row)
     end
 
     def find_company!(row)
-      return unless row[:company].present?
+      return unless row[:company_id].present?
 
-      Company.where('lower(name) = ?', row[:company].downcase).first!
+      Company.find(row[:company_id])
     end
 
     def assessment_attributes(row)

db/migrate/20191121170141_remove_isin_index_from_companies.rb

@@ -0,0 +1,5 @@
+class RemoveIsinIndexFromCompanies < ActiveRecord::Migration[6.0]
+  def change
+    remove_index :companies, :isin
+  end
+end

db/schema.rb

@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 2019_11_20_162650) do
+ActiveRecord::Schema.define(version: 2019_11_21_170141) do
 
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
@@ -83,7 +83,6 @@
     t.index ["discarded_at"], name: "index_companies_on_discarded_at"
     t.index ["geography_id"], name: "index_companies_on_geography_id"
     t.index ["headquarters_geography_id"], name: "index_companies_on_headquarters_geography_id"
-    t.index ["isin"], name: "index_companies_on_isin", unique: true
     t.index ["market_cap_group"], name: "index_companies_on_market_cap_group"
     t.index ["sector_id"], name: "index_companies_on_sector_id"
     t.index ["slug"], name: "index_companies_on_slug", unique: true

db/seeds.rb

@@ -28,6 +28,60 @@
   sector.update!(cp_unit: sector_cp_unit) if sector.new_record? && sector_cp_unit.present?
 end
 
+# instruments: Instrument Type & Instrument
+# envs: ALL
+# rubocop:disable Style/WordArray
+[
+  ['Capacity-building', ['Knowledge generation',
+                         'Sharing and dissemination',
+                         'Research and development',
+                         'Education and training']],
+  ['Regulation', ['Standards and obligations',
+                  'Building codes',
+                  'Zoning and spatial planning',
+                  'Disclosure obligations']],
+  ['Incentives', ['Taxes', 'Subsidies']],
+  ['Governance and planning', ['Creating bodies/institutions',
+                               'Designing processes',
+                               'Developing plans and strategies',
+                               'Assigning responsibilities to other levels of government',
+                               'Monitoring and evaluation']],
+  ['Direct investment', ['Public goods - early warning systems',
+                         'Public goods - other',
+                         'Social safety nets',
+                         'Provision of climate finance']]
+].each do |inst_type, instruments|
+  type = InstrumentType.find_or_create_by!(name: inst_type)
+  instruments.each do |inst|
+    Instrument.find_or_create_by!(name: inst, instrument_type: type)
+  end
+end
+# rubocop:enable Style/WordArray
+
+# Laws sectors: Names
+# envs: ALL
+[
+  'Agriculture',
+  'Residential and Commercial',
+  'Coastal zones',
+  'Economy-wide',
+  'Energy',
+  'Health',
+  'Industry',
+  'LULUCF',
+  'Social development',
+  'Tourism',
+  'Transportation',
+  'Urban',
+  'Waste',
+  'Water',
+  'Rural',
+  'Environment',
+  'Other'
+].each do |sector|
+  LawsSector.find_or_create_by!(name: sector)
+end
+
 if Rails.env.development? || ENV['SEED_DATA']
   # import geographies
   TimedLogger.log('Import geographies') do
@@ -37,8 +91,12 @@
 
   # import companies
   TimedLogger.log('Import companies') do
-    file = File.open(Rails.root.join('db', 'seeds', 'companies.csv'), 'r')
-    CSVImport::Companies.new(file).call
+    file = File.open(Rails.root.join('db', 'seeds', 'tpi-companies.csv'), 'r')
+    importer = CSVImport::Companies.new(file)
+    importer.override_id = true
+    importer.call
+
+    ActiveRecord::Base.connection.execute("select setval('companies_id_seq',max(id)) from companies;")
   end
 
   # import CP Benchmarks
@@ -60,6 +118,9 @@
 
     file = File.open(Rails.root.join('db', 'seeds', 'mq-assessments-M2.csv'), 'r')
     CSVImport::MQAssessments.new(file).call
+
+    file = File.open(Rails.root.join('db', 'seeds', 'mq-assessments-M3.csv'), 'r')
+    CSVImport::MQAssessments.new(file).call
   end
 
   # import Legislations