From c05d69bcfffcc731b387ab0136e6d1216bf65a99 Mon Sep 17 00:00:00 2001
From: Jacques Distler <distler@golem.ph.utexas.edu>
Date: Thu, 27 Aug 2009 16:57:37 -0500
Subject: [PATCH] Make upgrade_instiki Rake Task Database-Agnostic

Reported by James Herdman.
---
 lib/tasks/upgrade_instiki.rake | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/lib/tasks/upgrade_instiki.rake b/lib/tasks/upgrade_instiki.rake
index 49162073..be075a5e 100644
--- a/lib/tasks/upgrade_instiki.rake
+++ b/lib/tasks/upgrade_instiki.rake
@@ -1,9 +1,10 @@
-require 'sqlite3'
+require 'active_record'
 
-task :upgrade_instiki do
-  db = SQLite3::Database.new( "db/production.db.sqlite3" )
-  db.execute( "select * from webs" ) do |row|
-    if File.exists?('public/' + row[4])
+task :upgrade_instiki => :environment do
+  ActiveRecord::Base.establish_connection(:production)
+  webs = ActiveRecord::Base.connection.execute( "select * from webs" )
+  webs.each do |row|
+   if File.exists?('public/' + row[4])
       if File.exists?('webs/' + row[4])
         print "Warning! The directory webs/#{row[4]} already exists. Skipping.\n" 
       else