mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-11-18 23:21:30 +08:00
101 lines
3 KiB
Ruby
101 lines
3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module ModelExporters
|
|
class ExperimentExporter < ModelExporter
|
|
def initialize(experiment_id)
|
|
@include_archived = true
|
|
super()
|
|
@experiment = Experiment.find(experiment_id)
|
|
end
|
|
|
|
def export_template_to_dir
|
|
@asset_counter = 0
|
|
@include_archived = false
|
|
@experiment.transaction do
|
|
@experiment.uuid ||= SecureRandom.uuid
|
|
@dir_to_export = FileUtils.mkdir_p(
|
|
File.join("tmp/experiment_#{@experiment.id}" \
|
|
"_export_#{Time.now.to_i}")
|
|
).first
|
|
|
|
# Writing JSON file with experiment structure
|
|
File.write(
|
|
File.join(@dir_to_export, 'experiment.json'),
|
|
JSON.pretty_generate(experiment[0].as_json)
|
|
)
|
|
# Copying assets
|
|
assets_dir = File.join(@dir_to_export, 'assets')
|
|
copy_files(@assets_to_copy, :file, assets_dir) do
|
|
@asset_counter += 1
|
|
end
|
|
puts "Exported assets: #{@asset_counter}"
|
|
puts 'Done!'
|
|
return @dir_to_export
|
|
end
|
|
end
|
|
|
|
def experiment
|
|
if @include_archived
|
|
my_modules = @experiment.my_modules
|
|
my_module_groups = @experiment.my_module_groups
|
|
else
|
|
my_modules = @experiment.my_modules.active
|
|
my_module_groups = @experiment.my_module_groups.without_archived_modules
|
|
end
|
|
return {
|
|
experiment: @experiment,
|
|
user_assignments: @experiment.user_assignments.map do |ua|
|
|
user_assignment(ua)
|
|
end,
|
|
my_modules: my_modules.map { |m| my_module(m) },
|
|
my_module_groups: my_module_groups
|
|
}, @assets_to_copy
|
|
end
|
|
|
|
def user_assignment(user_assignment)
|
|
{
|
|
user_id: user_assignment.user_id,
|
|
assigned_by_id: user_assignment.assigned_by_id,
|
|
role_name: user_assignment.user_role.name,
|
|
assigned: user_assignment.assigned
|
|
}
|
|
end
|
|
|
|
def my_module(my_module)
|
|
{
|
|
my_module: my_module,
|
|
user_assignments: my_module.user_assignments.map do |ua|
|
|
user_assignment(ua)
|
|
end,
|
|
my_module_status_name: my_module.my_module_status&.name,
|
|
outputs: my_module.outputs,
|
|
my_module_tags: my_module.my_module_tags,
|
|
task_comments: my_module.task_comments,
|
|
my_module_repository_rows: my_module.my_module_repository_rows,
|
|
user_my_modules: my_module.user_my_modules,
|
|
protocols: my_module.protocols.map { |pr| protocol(pr) },
|
|
results: my_module.results.map { |res| result(res) }
|
|
}
|
|
end
|
|
|
|
def result(result)
|
|
@assets_to_copy.push(result.asset) if result.asset.present?
|
|
{
|
|
result: result,
|
|
result_comments: result.result_comments,
|
|
asset: result_assets_data(result.asset),
|
|
table: table(result.table),
|
|
result_text: result.result_text
|
|
}
|
|
end
|
|
|
|
def result_assets_data(asset)
|
|
return unless asset&.file&.attached?
|
|
|
|
{
|
|
asset: asset,
|
|
asset_blob: asset.file.blob
|
|
}
|
|
end
|
|
end
|
|
end
|