Fix tasks when running standalone

* Only require environment task if running Rails
rovermicrover · Jan 17, 2012 · 6659064 · 6659064
1 parent e1f462e
commit 6659064
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/sitemap_generator/tasks.rb b/lib/sitemap_generator/tasks.rb
@@ -21,7 +21,10 @@
 
   # Require sitemap_generator after loading the Rails environment.  We still need the require
   # in case we are installed as a gem and are setup to not automatically be required.
-  task :require_environment => :environment do
+  task :require_environment do
+    if defined?(Rails)
+      Rake::Task['environment'].invoke
+    end
     require 'sitemap_generator'
   end