add group_of_line filter and summary on lines and add clean on empty groups of lines

1 files changed, 13 insertions, 2 deletions

diff --git a/app/models/clean_up.rb b/app/models/clean_up.rb
index 5b3d346ff..a6d26d882 100644
--- a/app/models/clean_up.rb
+++ b/app/models/clean_up.rb
@@ -4,7 +4,8 @@ class CleanUp
   extend ActiveModel::Naming
   
     
-  attr_accessor :expected_date, :keep_lines, :keep_stops , :keep_companies, :keep_networks
+  attr_accessor :expected_date, :keep_lines, :keep_stops , :keep_companies
+  attr_accessor :keep_networks, :keep_group_of_lines
   
   validates_presence_of :expected_date
 
@@ -21,7 +22,7 @@ class CleanUp
   def clean
     result = CleanUpResult.new
     # find and remove time_tables 
-    tms = Chouette::TimeTable.expired_on(Date.parse(expected_date))
+    tms = Chouette::TimeTable.validity_out_from_on?(Date.parse(expected_date))
     result.time_table_count = tms.size
     tms.each do |tm|
       tm.destroy
@@ -108,6 +109,16 @@ class CleanUp
         end
       end
     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.destroy
+        end
+      end
+    end
     result
   end