aboutsummaryrefslogtreecommitdiffstats
path: root/app/models/chouette
diff options
context:
space:
mode:
Diffstat (limited to 'app/models/chouette')
-rw-r--r--app/models/chouette/line.rb4
-rw-r--r--app/models/chouette/network.rb11
2 files changed, 13 insertions, 2 deletions
diff --git a/app/models/chouette/line.rb b/app/models/chouette/line.rb
index 16d7ec6f6..784e3f5b9 100644
--- a/app/models/chouette/line.rb
+++ b/app/models/chouette/line.rb
@@ -4,8 +4,8 @@ module Chouette
include LineRestrictions
include LineReferentialSupport
include ObjectidSupport
- extend StifTransportModeEnumerations
- extend StifTransportSubmodeEnumerations
+ include StifTransportModeEnumerations
+ include StifTransportSubmodeEnumerations
extend ActiveModel::Naming
# FIXME http://jira.codehaus.org/browse/JRUBY-6358
diff --git a/app/models/chouette/network.rb b/app/models/chouette/network.rb
index d64ef038a..9b3f2fe29 100644
--- a/app/models/chouette/network.rb
+++ b/app/models/chouette/network.rb
@@ -4,12 +4,23 @@ module Chouette
include NetworkRestrictions
include LineReferentialSupport
include ObjectidSupport
+ extend Enumerize
# FIXME http://jira.codehaus.org/browse/JRUBY-6358
self.primary_key = "id"
has_many :lines
attr_accessor :source_type_name
+ enumerize :source_type_name, in: %w(public_and_private_utilities
+ road_authorities
+ transit_operator
+ public_transport
+ passenger_transport_coordinating_authority
+ travel_information_service_provider
+ travel_agency
+ individual_subject_of_travel_itinerary
+ other_information)
+
validates_format_of :registration_number, :with => %r{\A[0-9A-Za-z_-]+\Z}, :allow_nil => true, :allow_blank => true
validates_presence_of :name