Add short_id method to Chouette::StifNetexObjectid class

3 files changed, 8 insertions, 4 deletions

diff --git a/app/models/chouette/netex_object_id.rb b/app/models/chouette/netex_object_id.rb
index 07d862992..441004c1e 100644
--- a/app/models/chouette/netex_object_id.rb
+++ b/app/models/chouette/netex_object_id.rb
@@ -5,7 +5,7 @@ class Chouette::NetexObjectId < String
   end
   alias_method :objectid?, :valid?
 
-  @@format = /^([A-Za-z_]+):([0-9A-Za-z_]+):([A-Za-z]+):([0-9A-Za-z_-]+)$/ 
+  @@format = /^([A-Za-z_]+):([0-9A-Za-z_]+):([A-Za-z]+):([0-9A-Za-z_-]+)$/
   cattr_reader :format
 
   def parts
@@ -27,7 +27,7 @@ class Chouette::NetexObjectId < String
   def local_id
     parts.try(:fourth)
   end
-  
+
   def self.create(provider_id, system_id, object_type, local_id)
     new [provider_id, system_id, object_type, local_id].join(":")
   end
diff --git a/app/models/chouette/object_id.rb b/app/models/chouette/object_id.rb
index 4f58048e1..0b122c91b 100644
--- a/app/models/chouette/object_id.rb
+++ b/app/models/chouette/object_id.rb
@@ -5,7 +5,7 @@ class Chouette::ObjectId < String
   end
   alias_method :objectid?, :valid?
 
-  @@format = /^([0-9A-Za-z_]+):([A-Za-z]+):([0-9A-Za-z_-]+)$/ 
+  @@format = /^([0-9A-Za-z_]+):([A-Za-z]+):([0-9A-Za-z_-]+)$/
   cattr_reader :format
 
   def parts
@@ -23,7 +23,7 @@ class Chouette::ObjectId < String
   def local_id
     parts.try(:third)
   end
-  
+
   def self.create(system_id, object_type, local_id)
     new [system_id, object_type, local_id].join(":")
   end
diff --git a/app/models/chouette/stif_netex_objectid.rb b/app/models/chouette/stif_netex_objectid.rb
index 3b11691d9..a0a91668a 100644
--- a/app/models/chouette/stif_netex_objectid.rb
+++ b/app/models/chouette/stif_netex_objectid.rb
@@ -26,6 +26,10 @@ class Chouette::StifNetexObjectid < String
     parts.try(:fourth)
   end
 
+  def short_id
+    local_id.try(:split, "-").try(:[], -1)
+  end
+
   def self.create(provider_id, object_type, local_id, boiv_id)
     new [provider_id, object_type, local_id, boiv_id].join(":")
   end