2017-03-30 20:42:33 +03:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
require 'csv'
|
|
|
|
|
|
|
|
class ImportWorker
|
|
|
|
include Sidekiq::Worker
|
2017-04-24 01:38:37 +03:00
|
|
|
|
2017-04-04 01:53:20 +03:00
|
|
|
sidekiq_options queue: 'pull', retry: false
|
2017-03-30 20:42:33 +03:00
|
|
|
|
2017-04-11 22:40:14 +03:00
|
|
|
attr_reader :import
|
|
|
|
|
2017-03-30 20:42:33 +03:00
|
|
|
def perform(import_id)
|
2017-04-11 22:40:14 +03:00
|
|
|
@import = Import.find(import_id)
|
2017-03-30 20:42:33 +03:00
|
|
|
|
2017-10-04 01:39:32 +03:00
|
|
|
Import::RelationshipWorker.push_bulk(import_rows) do |row|
|
|
|
|
[@import.account_id, row.first, relationship_type]
|
2017-03-30 20:42:33 +03:00
|
|
|
end
|
|
|
|
|
2017-04-11 22:40:14 +03:00
|
|
|
@import.destroy
|
2017-03-30 20:42:33 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
2017-04-11 22:40:14 +03:00
|
|
|
def import_contents
|
|
|
|
Paperclip.io_adapters.for(@import.data).read
|
|
|
|
end
|
2017-03-30 20:42:33 +03:00
|
|
|
|
2017-10-04 01:39:32 +03:00
|
|
|
def relationship_type
|
|
|
|
case @import.type
|
|
|
|
when 'following'
|
|
|
|
'follow'
|
|
|
|
when 'blocking'
|
|
|
|
'block'
|
|
|
|
when 'muting'
|
|
|
|
'mute'
|
2017-03-30 20:42:33 +03:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-10-04 01:39:32 +03:00
|
|
|
def import_rows
|
2018-10-04 18:36:11 +03:00
|
|
|
rows = CSV.new(import_contents).reject(&:blank?)
|
|
|
|
rows = rows.take(FollowLimitValidator.limit_for_account(@import.account)) if @import.type == 'following'
|
|
|
|
rows
|
2017-03-30 20:42:33 +03:00
|
|
|
end
|
|
|
|
end
|