1 files changed, 11 insertions, 3 deletions
diff --git a/app/models/referential.rb b/app/models/referential.rb
index 91a88d02d..1794126a2 100644
--- a/app/models/referential.rb
+++ b/app/models/referential.rb
@@ -1,5 +1,5 @@
 # coding: utf-8
-class Referential < ActiveRecord::Base
+class Referential < ApplicationModel
   include DataFormatEnumerations
   include ObjectidFormatterSupport
 
@@ -168,6 +168,10 @@ class Referential < ActiveRecord::Base
     Chouette::TimeTable.all
   end
 
+  def time_table_dates
+    Chouette::TimeTableDate.all
+  end
+
   def timebands
     Chouette::Timeband.all
   end
@@ -184,6 +188,10 @@ class Referential < ActiveRecord::Base
     Chouette::VehicleJourneyFrequency.all
   end
 
+  def vehicle_journey_at_stops
+    Chouette::VehicleJourneyAtStop.all
+  end
+
   def routing_constraint_zones
     Chouette::RoutingConstraintZone.all
   end
@@ -233,7 +241,7 @@ class Referential < ActiveRecord::Base
     end
   end
 
-  def self.new_from(from, functional_scope)
+  def self.new_from(from, organisation)
     Referential.new(
       name: I18n.t("activerecord.copy", name: from.name),
       slug: "#{from.slug}_clone",
@@ -244,7 +252,7 @@ class Referential < ActiveRecord::Base
       stop_area_referential: from.stop_area_referential,
       created_from: from,
       objectid_format: from.objectid_format,
-      metadatas: from.metadatas.map { |m| ReferentialMetadata.new_from(m, functional_scope) }
+      metadatas: from.metadatas.map { |m| ReferentialMetadata.new_from(m, organisation) }
     )
   end