data model and rake task for course cohort providers

Author: alkesh

app/models/cohort.rb

@@ -8,6 +8,7 @@ class Cohort < ApplicationRecord
   has_many :statements, dependent: :restrict_with_exception
   has_many :delivery_partnerships, dependent: :destroy
   has_many :delivery_partners, through: :delivery_partnerships
+  has_many :course_cohorts, dependent: :destroy
 
   enum :funding, {
     zero: "zero",

app/models/course.rb

@@ -8,6 +8,9 @@ class Course < ApplicationRecord
   NPQ_SENCO = "npq-senco".freeze
 
   belongs_to :course_group, optional: true
+  has_many :course_cohorts, dependent: :destroy
+  has_many :course_cohort_providers, through: :course_cohorts
+  has_many :lead_providers, through: :course_cohort_providers
 
   validates :name, presence: true
   validates :identifier, presence: true, uniqueness: true

app/models/course_cohort.rb

@@ -0,0 +1,9 @@
+class CourseCohort < ApplicationRecord
+  belongs_to :course
+  belongs_to :cohort
+
+  has_many :course_cohort_providers, dependent: :destroy
+  has_many :lead_providers, through: :course_cohort_providers
+
+  validates :course_id, uniqueness: { scope: :cohort_id }
+end

app/models/course_cohort_provider.rb

@@ -0,0 +1,6 @@
+class CourseCohortProvider < ApplicationRecord
+  belongs_to :course_cohort
+  belongs_to :lead_provider
+
+  validates :course_cohort_id, uniqueness: { scope: :lead_provider_id }
+end

app/models/lead_provider.rb

@@ -90,6 +90,10 @@ class LeadProvider < ApplicationRecord
   has_many :delivery_partnerships
   has_many :delivery_partners, through: :delivery_partnerships
 
+  has_many :course_cohort_providers
+  has_many :course_cohorts, through: :course_cohort_providers
+  has_many :courses, through: :course_cohorts
+
   validates :name, presence: true
   validates :ecf_id, uniqueness: { case_sensitive: false }, allow_nil: true
 

config/analytics.yml

@@ -291,3 +291,15 @@ shared:
     - statement_id
     - created_at
     - updated_at
+  :course_cohort_providers:
+    - id
+    - course_cohort_id
+    - lead_provider_id
+    - created_at
+    - updated_at
+  :course_cohorts:
+    - id
+    - course_id
+    - cohort_id
+    - created_at
+    - updated_at

db/migrate/20260420093307_add_course_cohort_provider_tables.rb

@@ -0,0 +1,17 @@
+class AddCourseCohortProviderTables < ActiveRecord::Migration[8.0]
+  def change
+    create_table :course_cohorts do |t|
+      t.references :course, null: false, foreign_key: true
+      t.references :cohort, null: false, foreign_key: true
+      t.timestamps
+    end
+    add_index :course_cohorts, %i[course_id cohort_id], unique: true
+
+    create_table :course_cohort_providers do |t|
+      t.references :course_cohort, null: false, foreign_key: true
+      t.references :lead_provider, null: false, foreign_key: true
+      t.timestamps
+    end
+    add_index :course_cohort_providers, %i[course_cohort_id lead_provider_id], unique: true
+  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[8.0].define(version: 2026_04_01_112940) do
+ActiveRecord::Schema[8.0].define(version: 2026_04_20_093307) do
   # These are extensions that must be enabled in order to support this database
   enable_extension "btree_gin"
   enable_extension "citext"
@@ -241,6 +241,26 @@
     t.index ["statement_id"], name: "index_contracts_on_statement_id"
   end
 
+  create_table "course_cohort_providers", force: :cascade do |t|
+    t.bigint "course_cohort_id", null: false
+    t.bigint "lead_provider_id", null: false
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.index ["course_cohort_id", "lead_provider_id"], name: "idx_on_course_cohort_id_lead_provider_id_3527d5c43f", unique: true
+    t.index ["course_cohort_id"], name: "index_course_cohort_providers_on_course_cohort_id"
+    t.index ["lead_provider_id"], name: "index_course_cohort_providers_on_lead_provider_id"
+  end
+
+  create_table "course_cohorts", force: :cascade do |t|
+    t.bigint "course_id", null: false
+    t.bigint "cohort_id", null: false
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.index ["cohort_id"], name: "index_course_cohorts_on_cohort_id"
+    t.index ["course_id", "cohort_id"], name: "index_course_cohorts_on_course_id_and_cohort_id", unique: true
+    t.index ["course_id"], name: "index_course_cohorts_on_course_id"
+  end
+
   create_table "course_groups", force: :cascade do |t|
     t.string "name", null: false
     t.datetime "created_at", null: false
@@ -685,6 +705,10 @@
   add_foreign_key "contracts", "contract_templates"
   add_foreign_key "contracts", "courses"
   add_foreign_key "contracts", "statements"
+  add_foreign_key "course_cohort_providers", "course_cohorts"
+  add_foreign_key "course_cohort_providers", "lead_providers"
+  add_foreign_key "course_cohorts", "cohorts"
+  add_foreign_key "course_cohorts", "courses"
   add_foreign_key "courses", "course_groups"
   add_foreign_key "declarations", "applications"
   add_foreign_key "declarations", "cohorts"

docs/content_editors/images/erd.png

@@ -0,0 +1,30 @@
+namespace :course_cohort_providers do
+  desc "Populate course cohort providers"
+  task :load, %i[cohort_identifier course_to_provider_csv dry_run] => :versioned_environment do |_t, args|
+    dry_run = args[:dry_run] != "false"
+
+    logger = Rails.env.test? ? Rails.logger : Logger.new($stdout)
+
+    logger.info "Dry Run" if dry_run
+
+    raise "Missing required argument: cohort_identifier" unless args.cohort_identifier
+
+    cohort = Cohort.find_by(identifier: args.cohort_identifier)
+    raise "Cohort not found with identifier: #{args.cohort_identifier}" unless cohort
+    raise "Missing required argument: course_to_provider_csv" unless args.course_to_provider_csv
+
+    CourseCohort.transaction do
+      CSV.foreach(args[:course_to_provider_csv], headers: true, header_converters: :symbol, strip: true) do |row|
+        course = Course.find_by!(identifier: row[:course_identifier])
+        course_cohort = CourseCohort.find_or_create_by!(course:, cohort:)
+        lead_provider = LeadProvider.find_by!(name: row[:lead_provider_name])
+        course_cohort.course_cohort_providers.find_or_create_by!(lead_provider:)
+      end
+
+      if dry_run
+        logger.info "Dry run: rolling back"
+        raise ActiveRecord::Rollback
+      end
+    end
+  end
+end