diff options
Diffstat (limited to 'app')
| -rw-r--r-- | app/exporters/chouette/kml/exporter.rb | 86 | ||||
| -rw-r--r-- | app/views/api/kml/access_links/index.kml.erb | 2 | ||||
| -rw-r--r-- | app/views/api/kml/access_points/index.kml.erb | 2 | ||||
| -rw-r--r-- | app/views/api/kml/connection_links/index.kml.erb | 2 |
4 files changed, 66 insertions, 26 deletions
diff --git a/app/exporters/chouette/kml/exporter.rb b/app/exporters/chouette/kml/exporter.rb index aca4a1d09..a7adfab79 100644 --- a/app/exporters/chouette/kml/exporter.rb +++ b/app/exporters/chouette/kml/exporter.rb @@ -1,25 +1,48 @@ class Chouette::Kml::Exporter attr_reader :referential - attr_reader :kml_export + attr_reader :kml_export, :lines, :routes, :journey_patterns def initialize(referential, kml_export) @referential = referential @kml_export = kml_export + @lines = nil + @routes = nil + @journey_patterns = nil end - def lines(object, ids) + def select_lines(object, ids) if object == "network" - ids.present? ? referential.networks.find( ids.split(",") ).collect(&:lines).flatten : referential.networks.collect(&:lines).flatten + ids.present? ? Chouette::Line.includes(:routes).where( :network_id => ids.split(",")).order(:name) : + Chouette::Line.joins(:network).includes(:routes).order(:name) elsif object == "company" - ids.present? ? referential.companies.find( ids.split(",") ).collect(&:lines).flatten : referential.companies.collect(&:lines).flatten - elsif object == "line" - ids.present? ? referential.lines.find( ids.split(",") ): referential.lines + ids.present? ? Chouette::Line.includes(:routes).where( :company_id => ids.split(",")).order(:name) : + Chouette::Line.joins(:company).includes(:routes).order(:name) + elsif object == "line" && ids.present? + Chouette::Line.includes(:routes).where( :id => ids.split(",")).order(:name) else - referential.lines + Chouette::Line.includes(:routes).order(:name) end end + def routes_exportable? + routes && routes.size < 150 + end + + def lines_exportable? + lines && lines.size < 150 + end + + def journey_patterns_exportable? + journey_patterns && journey_patterns.size < 150 + end + + def log_overflow_warning( target_class) + kml_export.log_messages.create( :severity => "warning", :key => "EXPORT_ERROR|EXCEPTION", + :arguments => {"0" => I18n.t( 'export_log_messages.messages.overflow', + :count => 150, :data => target_class.model_name.human.pluralize)}) + end + def export(zip_file_path, options = {}) begin referential.switch @@ -30,35 +53,52 @@ class Chouette::Kml::Exporter Dir.mktmpdir(nil, "/tmp"){ |temp_dir| - lines_collected = lines( options[:o], options[:id] ).sort_by(&:name) - Chouette::Kml::LineExporter.save( lines_collected, temp_dir, kml_export) + @lines = select_lines( options[:o], options[:id] ) + @routes = lines.map(&:routes).flatten.sort {|a,b| (a.name && b.name) ? a.name <=> b.name : a.id <=> b.id} if lines_exportable? + @journey_patterns = Chouette::JourneyPattern.where( :route_id => routes.map(&:id) ).order(:name) if routes_exportable? + + if lines_exportable? + Chouette::Kml::LineExporter.save( lines, temp_dir, kml_export) + else + log_overflow_warning(Chouette::Line) + end - routes = lines_collected.collect(&:routes).flatten.uniq.sort_by(&:name) - Chouette::Kml::RouteExporter.save( routes, temp_dir, kml_export) + if routes_exportable? + Chouette::Kml::RouteExporter.save( routes, temp_dir, kml_export) + else + log_overflow_warning(Chouette::Route) if lines_exportable? + end - journey_patterns = routes.map { |r| r.journey_patterns}.flatten - Chouette::Kml::JourneyPatternExporter.save( journey_patterns, temp_dir, kml_export) + if journey_patterns_exportable? + Chouette::Kml::JourneyPatternExporter.save( journey_patterns, temp_dir, kml_export) + else + log_overflow_warning(Chouette::JourneyPattern) if routes_exportable? + end - stop_areas = lines_collected.collect(&:stop_areas).flatten.uniq.sort_by(&:name) - Chouette::Kml::StopAreaExporter.save( stop_areas, temp_dir, kml_export, "Quay") + # if too many lines + # there may be too many stop_areas + if lines_exportable? + stop_areas = Chouette::StopArea.joins(:children => [:stop_points => [:route => :line] ]).where(:lines => {:id => lines.map(&:id)}).uniq.order(:name) + Chouette::Kml::StopAreaExporter.save( stop_areas, temp_dir, kml_export, "Quay") - commercial_stop_areas = lines_collected.collect(&:commercial_stop_areas).flatten.uniq.sort_by(&:name) - Chouette::Kml::StopAreaExporter.save( commercial_stop_areas, temp_dir, kml_export, "CommercialStopPoint") + commercial_stop_areas = Chouette::StopArea.where( :id => stop_areas.map(&:parent_id).compact.uniq).order(:name) + Chouette::Kml::StopAreaExporter.save( commercial_stop_areas, temp_dir, kml_export, "CommercialStopPoint") + end - if(options[:o].present?) # Add all objects - stop_places = referential.stop_areas.stop_place.sort_by(&:name) + if( options[:o] == "line" and not options[:id].present?) # Add all objects + stop_places = referential.stop_areas.stop_place.order(:name) Chouette::Kml::StopAreaExporter.save( stop_places, temp_dir, kml_export, "StopPlace") - itls = referential.stop_areas.itl.sort_by(&:name) + itls = referential.stop_areas.itl.order(:name) Chouette::Kml::StopAreaExporter.save( itls, temp_dir, kml_export, "ITL") - connection_links = referential.connection_links.sort_by(&:name) + connection_links = referential.connection_links.order(:name) Chouette::Kml::ConnectionLinkExporter.save( connection_links, temp_dir, kml_export) - access_links = referential.access_links.sort_by(&:name) + access_links = referential.access_links.order(:name) Chouette::Kml::AccessLinkExporter.save(access_links, temp_dir, kml_export) - access_points = referential.access_points.sort_by(&:name) + access_points = referential.access_points.order(:name) Chouette::Kml::AccessPointExporter.save(access_points, temp_dir, kml_export) end diff --git a/app/views/api/kml/access_links/index.kml.erb b/app/views/api/kml/access_links/index.kml.erb index e625a1a21..871646c68 100644 --- a/app/views/api/kml/access_links/index.kml.erb +++ b/app/views/api/kml/access_links/index.kml.erb @@ -1,7 +1,7 @@ <?xml version="1.0" encoding="UTF-8"?> <kml xmlns="http://www.opengis.net/kml/2.2"> <Document> - <name><%= h(Chouette::AccessLink.model_name.human) %> : <%= h(@journey_pattern.name) %></name> + <name><%= h(Chouette::AccessLink.model_name.human) %></name> <% @access_links.each do |access_link| %> <Placemark id="<%= access_link.objectid %>" > <name><%= h(access_link.name) %></name> diff --git a/app/views/api/kml/access_points/index.kml.erb b/app/views/api/kml/access_points/index.kml.erb index 5e98260b6..f48846f91 100644 --- a/app/views/api/kml/access_points/index.kml.erb +++ b/app/views/api/kml/access_points/index.kml.erb @@ -1,7 +1,7 @@ <?xml version="1.0" encoding="UTF-8"?> <kml xmlns="http://www.opengis.net/kml/2.2"> <Document> - <name><%= h(Chouette::AccessPoint.model_name.human) %> : <%= h(@journey_pattern.name) %></name> + <name><%= h(Chouette::AccessPoint.model_name.human) %></name> <name>access</name> <% @access_points.each do |access_point| %> <Placemark id="<%= access_point.objectid %>" > diff --git a/app/views/api/kml/connection_links/index.kml.erb b/app/views/api/kml/connection_links/index.kml.erb index 2c900aca0..8eb6c6ea2 100644 --- a/app/views/api/kml/connection_links/index.kml.erb +++ b/app/views/api/kml/connection_links/index.kml.erb @@ -1,7 +1,7 @@ <?xml version="1.0" encoding="UTF-8"?> <kml xmlns="http://www.opengis.net/kml/2.2"> <Document> - <name><%= h(Chouette::ConnectionLink.model_name.human) %> : <%= h(@journey_pattern.name) %></name> + <name><%= h(Chouette::ConnectionLink.model_name.human) %></name> <% @connection_links.each do |connection_link| %> <Placemark id="<%= connection_link.objectid %>" > <name><%= h(connection_link.name) %></name> |
