diff options
Diffstat (limited to 'app/models')
| -rw-r--r-- | app/models/clean_up.rb | 189 |
1 files changed, 86 insertions, 103 deletions
diff --git a/app/models/clean_up.rb b/app/models/clean_up.rb index cc8a6e537..986fbdd24 100644 --- a/app/models/clean_up.rb +++ b/app/models/clean_up.rb @@ -1,138 +1,121 @@ -class CleanUp - include ActiveModel::Validations - include ActiveModel::Conversion - extend ActiveModel::Naming +class CleanUp < ActiveRecord::Base + include AASM + belongs_to :referential + validates :expected_date, presence: true + after_commit :perform_cleanup, :on => :create - attr_accessor :expected_date, :keep_lines, :keep_stops , :keep_companies - attr_accessor :keep_networks, :keep_group_of_lines + def perform_cleanup + CleanUpWorker.perform_async(self.id) + end + + aasm column: :status do + state :new, :initial => true + state :pending + state :successful + state :failed + + event :run, after: :update_started_at do + transitions :from => [:new, :failed], :to => :pending + end - validates_presence_of :expected_date + event :successful, after: :update_ended_at do + transitions :from => [:pending, :failed], :to => :successful + end - def initialize(attributes = {}) - attributes.each do |name, value| - send("#{name}=", value) + event :failed, after: :update_ended_at do + transitions :from => :pending, :to => :failed end end - def persisted? - false + def update_started_at + update_attribute(:started_at, Time.now) end - def physical_stop_areas - Chouette::StopArea.physical.includes(:stop_points).where(:stop_points => {id: nil}) + def update_ended_at + update_attribute(:ended_at, Time.now) end - def clean_physical_stop_areas - Chouette::StopArea.where(id: self.physical_stop_areas.pluck(:id)).delete_all + def clean + # as foreign keys are presents , delete method can be used for faster performance + # find and remove time_tables + result = CleanUpResult.new + tms = Chouette::TimeTable.validity_out_from_on?(expected_date) + result.time_table_count = tms.size + tms.each.map(&:delete) + + result.vehicle_journey_count = self.clean_vehicle_journeys + result.journey_pattern_count = self.clean_journey_patterns + result.route_count = self.clean_routes + result.line_count = self.clean_lines unless keep_lines + + unless keep_stops + result.stop_count += self.clean_physical_stop_areas + result.stop_count += self.clean_commercial_stop_areas + result.stop_count += self.clean_stop_place_stop_areas + result.stop_count += self.clean_itl_stop_areas + end + + # If asked remove companies without lines or vehicle journeys + result.company_count = self.clean_companies unless keep_companies + # If asked remove networks without lines + result.network_count = self.clean_networks unless keep_networks + # If asked remove group_of_lines without lines + result.group_of_line_count = self.clean_group_of_lines unless keep_group_of_lines + result end - def commercial_stop_areas - Chouette::StopArea.commercial + def clean_physical_stop_areas + ids = Chouette::StopArea.physical.includes(:stop_points).where(:stop_points => {id: nil}).pluck(:id) + Chouette::StopArea.physical.where(id: ids).delete_all end - def stop_place_stop_areas - Chouette::StopArea.stop_place + def clean_commercial_stop_areas + ids = Chouette::StopArea.commercial.where.not(parent_id: nil).pluck(:parent_id) + Chouette::StopArea.commercial.where.not(id: ids).delete_all end - def itl_stop_areas - Chouette::StopArea.itl + def clean_stop_place_stop_areas + ids = Chouette::StopArea.stop_place.includes(:stop_points).where(:stop_points => {id: nil}).pluck(:id) + Chouette::StopArea.stop_place.where(id: ids).delete_all end - def vehicle_journeys - Chouette::VehicleJourney.includes(:time_tables).where(:time_tables => {id: nil}) + def clean_itl_stop_areas + ids = Chouette::StopArea.itl.includes(:stop_points).where(:stop_points => {id: nil}).pluck(:id) + Chouette::StopArea.itl.where(id: ids).delete_all end + def clean_vehicle_journeys - Chouette::VehicleJourney.where(id: self.vehicle_journeys.pluck(:id)).delete_all + ids = Chouette::VehicleJourney.includes(:time_tables).where(:time_tables => {id: nil}).pluck(:id) + Chouette::VehicleJourney.where(id: ids).delete_all end - def lines - Chouette::Line.includes(:routes).where(:routes => {id: nil}) - end def clean_lines - Chouette::Line.where(id: self.lines.pluck(:id)).delete_all + ids = Chouette::Line.includes(:routes).where(:routes => {id: nil}).pluck(:id) + Chouette::Line.where(id: ids).delete_all end - def routes - Chouette::Route.includes(:journey_patterns).where(:journey_patterns => {id: nil}) - end def clean_routes - Chouette::Route.where(id: self.routes.pluck(:id)).delete_all + ids = Chouette::Route.includes(:journey_patterns).where(:journey_patterns => {id: nil}).pluck(:id) + Chouette::Route.where(id: ids).delete_all end - def journey_patterns - Chouette::JourneyPattern.includes(:vehicle_journeys).where(:vehicle_journeys => {id: nil}) - end def clean_journey_patterns - Chouette::JourneyPattern.where(id: self.journey_patterns.pluck(:id)).delete_all + ids = Chouette::JourneyPattern.includes(:vehicle_journeys).where(:vehicle_journeys => {id: nil}).pluck(:id) + Chouette::JourneyPattern.where(id: ids).delete_all end - def clean - # as foreign keys are presents , delete method can be used for faster performance - result = CleanUpResult.new - # find and remove time_tables - tms = Chouette::TimeTable.validity_out_from_on?(Date.parse(expected_date)) - result.time_table_count = tms.size - tms.each do |tm| - tm.delete - end - - result.vehicle_journey_count = self.clean_vehicle_journeys - result.journey_pattern_count = self.clean_journey_patterns - result.route_count = self.clean_routes - - result.line_count = self.clean_lines if keep_lines == "0" - result.stop_count = self.clean_physical_stop_areas if keep_stops == "0" - - if keep_stops == "0" - commercial_stop_areas.find_each do |csp| - if csp.children.size == 0 - result.stop_count += 1 - csp.delete - end - end - stop_place_stop_areas.find_each do |sp| - if sp.children.size == 0 - result.stop_count += 1 - sp.delete - end - end - itl_stop_areas.find_each do |itl| - if itl.routing_stops.size == 0 - result.stop_count += 1 - itl.delete - end - end - end - # if asked remove companies without lines or vehicle journeys - if keep_companies == "0" - Chouette::Company.find_each do |c| - if c.lines.size == 0 - result.company_count += 1 - c.delete - end - end - end - - # if asked remove networks without lines - if keep_networks == "0" - Chouette::Network.find_each do |n| - if n.lines.size == 0 - result.network_count += 1 - n.delete - end - end - end + def clean_companies + ids = Chouette::Company.includes(:lines).where(:lines => {id: nil}).pluck(:id) + Chouette::Company.where(id: ids).delete_all + end - # if asked remove group_of_lines without lines - if keep_group_of_lines == "0" - Chouette::GroupOfLine.find_each do |n| - if n.lines.size == 0 - result.group_of_line_count += 1 - n.delete - end - end - end - result + def clean_networks + ids = Chouette::Network.includes(:lines).where(:lines => {id: nil}).pluck(:id) + Chouette::Network.where(id: ids).delete_all end + def clean_group_of_lines + ids = Chouette::GroupOfLine.includes(:lines).where(:lines => {id: nil}).pluck(:id) + Chouette::GroupOfLine.where(id: ids).delete_all + end end |