added check to column existense in migrations for rails 3

HighwayStar · Dec 18, 2012 · 9605172 · 9605172
1 parent 0fe9a30
commit 9605172
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/...grate/000_add_stealth_allowed_to_users.rb → ...218194801_add_stealth_allowed_to_users.rb b/...grate/000_add_stealth_allowed_to_users.rb → ...218194801_add_stealth_allowed_to_users.rb
@@ -2,30 +2,27 @@ class AddStealthAllowedToUsers < ActiveRecord::Migration
 
   # Rails 3
   def up
-    self.class.add_stealth_allowed
+    unless column_exists? :users, :stealth_allowed
+      add_column :users, :stealth_allowed, :boolean, :default => false
+    end
   end
 
   def down
-    self.class.remove_stealth_allowed
+    if column_exists? :users, :stealth_allowed
+      remove_column :users, :stealth_allowed
+    end
   end
 
   class << self
 
     # Rails 2
     def up
-      add_stealth_allowed
+      add_column :users, :stealth_allowed, :boolean, :default => false
     end
 
     def down
-      remove_stealth_allowed
-    end
-
-    def add_stealth_allowed
-      add_column :users, :stealth_allowed, :boolean, :default => false 
-    end
-
-    def remove_stealth_allowed
       remove_column :users, :stealth_allowed
     end
+
   end
 end