mirror of
https://github.com/mastodon/mastodon
synced 2024-12-22 10:38:14 +09:00
67 lines
1.7 KiB
Ruby
67 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'csv'
|
|
|
|
# A non-activerecord helper class for csv upload
|
|
class Admin::Import
|
|
include ActiveModel::Model
|
|
|
|
ROWS_PROCESSING_LIMIT = 20_000
|
|
|
|
attr_accessor :data
|
|
|
|
validates :data, presence: true
|
|
validate :validate_data
|
|
|
|
def data_file_name
|
|
data.original_filename
|
|
end
|
|
|
|
def csv_rows
|
|
csv_data.rewind
|
|
|
|
csv_data.take(ROWS_PROCESSING_LIMIT + 1)
|
|
end
|
|
|
|
private
|
|
|
|
def csv_data
|
|
return @csv_data if defined?(@csv_data)
|
|
|
|
csv_converter = lambda do |field, field_info|
|
|
case field_info.header
|
|
when '#domain'
|
|
field&.downcase&.strip
|
|
when '#public_comment'
|
|
field&.strip
|
|
when '#severity'
|
|
field&.downcase&.strip&.to_sym
|
|
when '#reject_media', '#reject_reports', '#obfuscate'
|
|
ActiveModel::Type::Boolean.new.cast(field&.downcase)
|
|
else
|
|
field
|
|
end
|
|
end
|
|
|
|
@csv_data = CSV.open(data.path, encoding: 'UTF-8', skip_blanks: true, headers: true, converters: csv_converter)
|
|
@csv_data.take(1) # Ensure the headers are read
|
|
@csv_data = CSV.open(data.path, encoding: 'UTF-8', skip_blanks: true, headers: ['#domain'], converters: csv_converter) unless @csv_data.headers&.first == '#domain'
|
|
@csv_data
|
|
end
|
|
|
|
def csv_row_count
|
|
return @csv_row_count if defined?(@csv_row_count)
|
|
|
|
csv_data.rewind
|
|
@csv_row_count = csv_data.take(ROWS_PROCESSING_LIMIT + 2).count
|
|
end
|
|
|
|
def validate_data
|
|
return if data.nil?
|
|
|
|
errors.add(:data, I18n.t('imports.errors.over_rows_processing_limit', count: ROWS_PROCESSING_LIMIT)) if csv_row_count > ROWS_PROCESSING_LIMIT
|
|
rescue CSV::MalformedCSVError => e
|
|
errors.add(:data, I18n.t('imports.errors.invalid_csv_file', error: e.message))
|
|
end
|
|
end
|