discourse/app/jobs/regular/export_csv_file.rb

99 lines
2.6 KiB
Ruby
Raw Normal View History

2014-08-09 18:28:57 +08:00
require 'csv'
require_dependency 'system_message'
module Jobs
class ExportCsvFile < Jobs::Base
sidekiq_options retry: false
attr_accessor :current_user
def initialize
@file_name = ""
end
def execute(args)
entity = args[:entity]
@current_user = User.find_by(id: args[:user_id])
raise Discourse::InvalidParameters.new(:entity) if entity.blank?
case entity
when 'user'
query = ::AdminUserIndexQuery.new
user_data = query.find_users_query.to_a
data = Array.new
2014-08-09 18:28:57 +08:00
user_data.each do |user|
user_array = Array.new
2014-09-06 23:41:05 +08:00
group_names = get_group_names(user).join(';')
user_array = get_user_fields(user)
2014-09-06 23:41:05 +08:00
user_array.push(group_names) if group_names != ''
data.push(user_array)
2014-08-09 18:28:57 +08:00
end
end
if data && data.length > 0
set_file_path
write_csv_file(data)
end
notify_user
end
private
2014-09-06 23:41:05 +08:00
def get_group_names(user)
group_names = []
2014-10-24 03:06:45 +08:00
groups = user.groups
2014-09-06 23:41:05 +08:00
groups.each do |group|
group_names.push(group.name)
end
return group_names
end
def get_user_fields(user)
2014-11-08 22:07:21 +08:00
csv_user_attrs = ['id','name','username','email','title','created_at']
csv_user_stats = ['topics_entered','posts_read_count','time_read','topic_count','post_count','likes_given','likes_received']
user_array = []
2014-11-08 22:07:21 +08:00
csv_user_attrs.each do |attr|
user_array.push(user.attributes[attr])
end
2014-11-08 22:07:21 +08:00
csv_user_stats.each do |stat|
user_array.push(user.user_stat.attributes[stat])
end
return user_array
end
2014-08-09 18:28:57 +08:00
def set_file_path
@file_name = "export_#{SecureRandom.hex(4)}.csv"
# ensure directory exists
dir = File.dirname("#{ExportCsv.base_directory}/#{@file_name}")
FileUtils.mkdir_p(dir) unless Dir.exists?(dir)
end
def write_csv_file(data)
# write to CSV file
CSV.open(File.expand_path("#{ExportCsv.base_directory}/#{@file_name}", __FILE__), "w") do |csv|
data.each do |value|
csv << value
2014-08-09 18:28:57 +08:00
end
end
end
def notify_user
if @current_user
if @file_name != "" && File.exists?("#{ExportCsv.base_directory}/#{@file_name}")
SystemMessage.create_from_system_user(@current_user, :csv_export_succeeded, download_link: "#{Discourse.base_url}/admin/export_csv/#{@file_name}/download", file_name: @file_name)
else
SystemMessage.create_from_system_user(@current_user, :csv_export_failed)
end
end
end
end
end