diff options
Diffstat (limited to 'lib/bulk_imports/pipeline')
-rw-r--r-- | lib/bulk_imports/pipeline/context.rb | 8 | ||||
-rw-r--r-- | lib/bulk_imports/pipeline/extracted_data.rb | 2 |
2 files changed, 9 insertions, 1 deletions
diff --git a/lib/bulk_imports/pipeline/context.rb b/lib/bulk_imports/pipeline/context.rb index 3c69c729f36..d753f888671 100644 --- a/lib/bulk_imports/pipeline/context.rb +++ b/lib/bulk_imports/pipeline/context.rb @@ -16,6 +16,14 @@ module BulkImports @entity ||= tracker.entity end + def portable + @portable ||= entity.group || entity.project + end + + def import_export_config + @import_export_config ||= ::BulkImports::FileTransfer.config_for(portable) + end + def group @group ||= entity.group end diff --git a/lib/bulk_imports/pipeline/extracted_data.rb b/lib/bulk_imports/pipeline/extracted_data.rb index c9e54b61dd3..0b36c068298 100644 --- a/lib/bulk_imports/pipeline/extracted_data.rb +++ b/lib/bulk_imports/pipeline/extracted_data.rb @@ -6,7 +6,7 @@ module BulkImports attr_reader :data def initialize(data: nil, page_info: {}) - @data = Array.wrap(data) + @data = data.is_a?(Enumerator) ? data : Array.wrap(data) @page_info = page_info end |