87 lines
3.0 KiB
Ruby
87 lines
3.0 KiB
Ruby
#
|
|
# Copyright (C) 2011 Instructure, Inc.
|
|
#
|
|
# This file is part of Canvas.
|
|
#
|
|
# Canvas is free software: you can redistribute it and/or modify it under
|
|
# the terms of the GNU Affero General Public License as published by the Free
|
|
# Software Foundation, version 3 of the License.
|
|
#
|
|
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License along
|
|
# with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
module SIS
|
|
class GroupMembershipImporter < BaseImporter
|
|
|
|
def process
|
|
start = Time.now
|
|
importer = Work.new(@batch, @root_account, @logger)
|
|
yield importer
|
|
@logger.debug("Group Users took #{Time.now - start} seconds")
|
|
return importer.success_count
|
|
end
|
|
|
|
private
|
|
class Work
|
|
attr_accessor :success_count
|
|
|
|
def initialize(batch, root_account, logger)
|
|
@batch = batch
|
|
@root_account = root_account
|
|
@logger = logger
|
|
@success_count = 0
|
|
@groups_cache = {}
|
|
end
|
|
|
|
def add_group_membership(user_id, group_id, status)
|
|
user_id = user_id.to_s
|
|
group_id = group_id.to_s
|
|
@logger.debug("Processing Group User #{[user_id, group_id, status].inspect}")
|
|
raise ImportError, "No group_id given for a group user" if group_id.blank?
|
|
raise ImportError, "No user_id given for a group user" if user_id.blank?
|
|
raise ImportError, "Improper status \"#{status}\" for a group user" unless status =~ /\A(accepted|deleted)/i
|
|
|
|
pseudo = @root_account.pseudonyms.where(sis_user_id: user_id).first
|
|
user = pseudo.try(:user)
|
|
|
|
group = @groups_cache[group_id]
|
|
group ||= @root_account.all_groups.where(sis_source_id: group_id).first
|
|
@groups_cache[group.sis_source_id] = group if group
|
|
|
|
raise ImportError, "User #{user_id} didn't exist for group user" unless user
|
|
raise ImportError, "Group #{group_id} didn't exist for group user" unless group
|
|
|
|
# can't query group.group_memberships, since that excludes deleted memberships
|
|
group_membership = GroupMembership.where(group_id: group, user_id: user).first
|
|
group_membership ||= group.group_memberships.build(:user => user)
|
|
|
|
group_membership.sis_batch_id = @batch.id if @batch
|
|
|
|
case status
|
|
when /accepted/i
|
|
group_membership.workflow_state = 'accepted'
|
|
when /deleted/i
|
|
group_membership.workflow_state = 'deleted'
|
|
end
|
|
|
|
if group_membership.valid?
|
|
group_membership.save
|
|
else
|
|
msg = "A group user did not pass validation "
|
|
msg += "(" + "user: #{user_id}, group: #{group_id}, error: "
|
|
msg += group_membership.errors.full_messages.join(", ") + ")"
|
|
raise ImportError, msg
|
|
end
|
|
@success_count += 1
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|