2 files changed, 38 insertions, 38 deletions
diff --git a/app/controllers/api/v1/netex_imports_controller.rb b/app/controllers/api/v1/netex_imports_controller.rb
index 2654fa088..94600e93b 100644
--- a/app/controllers/api/v1/netex_imports_controller.rb
+++ b/app/controllers/api/v1/netex_imports_controller.rb
@@ -25,56 +25,19 @@ module Api
 
       def create_models
         find_workbench
-        create_referential
         create_netex_import
       end
 
       def create_netex_import
         attributes = netex_import_params.merge creator: "Webservice"
-
-        attributes = attributes.merge referential_id: @new_referential.id
-
         @netex_import = Import::Netex.new attributes
         @netex_import.save!
-
-        unless @netex_import.referential
-          Rails.logger.info "Can't create referential for import #{@netex_import.id}: #{@new_referential.inspect} #{@new_referential.metadatas.inspect} #{@new_referential.errors.full_messages}"
-          @netex_import.messages.create criticity: :error, message_key: "referential_creation"
-        end
+        @netex_import.create_referential
       rescue ActiveRecord::RecordInvalid
         render json: {errors: @netex_import.errors}, status: 406
         finish_action!
       end
 
-      def create_referential
-        @new_referential =
-          Referential.new(
-            name: netex_import_params['name'],
-            organisation_id: @workbench.organisation_id,
-            workbench_id: @workbench.id,
-            metadatas: [metadata]
-          )
-        @new_referential.save
-      end
-
-      def metadata
-        metadata = ReferentialMetadata.new
-
-        if netex_import_params['file']
-          netex_file = STIF::NetexFile.new(netex_import_params['file'].to_io)
-          frame = netex_file.frames.first
-
-          if frame
-            metadata.periodes = frame.periods
-
-            line_objectids = frame.line_refs.map { |ref| "STIF:CODIFLIGNE:Line:#{ref}" }
-            metadata.line_ids = @workbench.lines.where(objectid: line_objectids).pluck(:id)
-          end
-        end
-
-        metadata
-      end
-
       def netex_import_params
         params
           .require('netex_import')
diff --git a/app/models/import/netex.rb b/app/models/import/netex.rb
index 2b0982229..83b654f81 100644
--- a/app/models/import/netex.rb
+++ b/app/models/import/netex.rb
@@ -10,6 +10,25 @@ class Import::Netex < Import::Base
 
   validates_presence_of :parent
 
+  def create_referential
+    self.referential =
+      Referential.new(
+        name: self.name,
+        organisation_id: workbench.organisation_id,
+        workbench_id: workbench.id,
+        metadatas: [referential_metadata]
+      )
+    self.referential.save
+    unless self.referential.valid?
+      Rails.logger.info "Can't create referential for import #{self.id}: #{referential.inspect} #{referential.metadatas.inspect} #{referential.errors.messages}"
+      if referential.metadatas.all?{|m| m.line_ids.empty?}
+        parent.messages.create criticity: :error, message_key: "referential_creation_missing_lines", message_attributes: {referential_name: referential.name}
+      else
+        parent.messages.create criticity: :error, message_key: "referential_creation", message_attributes: {referential_name: referential.name}
+      end
+    end
+  end
+
   private
 
   def iev_callback_url
@@ -21,4 +40,22 @@ class Import::Netex < Import::Base
       referential.destroy
     end
   end
+
+  def referential_metadata
+    metadata = ReferentialMetadata.new
+
+    if self.file
+      netex_file = STIF::NetexFile.new(self.file.path)
+      frame = netex_file.frames.first
+
+      if frame
+        metadata.periodes = frame.periods
+
+        line_objectids = frame.line_refs.map { |ref| "STIF:CODIFLIGNE:Line:#{ref}" }
+        metadata.line_ids = workbench.lines.where(objectid: line_objectids).pluck(:id)
+      end
+    end
+
+    metadata
+  end
 end