Rake task to import labels for citations

Author: matt-bernhardt

lib/tasks/citation.rake

@@ -60,4 +60,26 @@ namespace :citations do
       end
     end
   end
+
+  # The import task is used to ingest labels for use by the citation scoring
+  # process. This process will overwrite any existing labels for records in the
+  # provided CSV file.
+  #
+  # This is meant to be used to load local data only.
+  desc 'Import labels for citations'
+  task :import, %i[path] => :environment do |_task, args|
+    raise ArgumentError.new, 'Path is required' if args.path.blank?
+
+    # Again, this task is only meant for local data.
+    data = File.read(args.path)
+
+    CSV.parse(data, headers: true) do |row|
+      Rails.logger.info("Parsing new row for term #{row['TermID']}, label: #{row['Label']}")
+      term = Term.find_by(id: row['TermID'])
+
+      term.label = row['Label']
+
+      term.save
+    end
+  end
 end