From b277cf28e8a6f5636d0485ab8c04ea0660444440 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jos=C3=A9=20Valim?= Date: Sat, 4 Jul 2009 18:13:27 +0200 Subject: Allow scaffold controller to load action_orm files on demand. --- railties/lib/generators/named_base.rb | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'railties/lib/generators/named_base.rb') diff --git a/railties/lib/generators/named_base.rb b/railties/lib/generators/named_base.rb index 8f44000729..0d399c876c 100644 --- a/railties/lib/generators/named_base.rb +++ b/railties/lib/generators/named_base.rb @@ -129,13 +129,24 @@ module Rails # def orm_class @orm_class ||= begin + # Raise an error if the class_option :orm was not defined. unless self.class.class_options[:orm] raise "You need to have :orm as class option to invoke orm_class and orm_instance" end action_orm = "#{options[:orm].to_s.classify}::Generators::ActionORM" - action_orm.constantize - rescue NameError => e + + # If the orm was not loaded, try to load it at "generators/orm", + # for example "generators/active_record" or "generators/sequel". + begin + klass = action_orm.constantize + rescue NameError + require "generators/#{options[:orm]}" + end + + # Try once again after loading the file with success. + klass ||= action_orm.constantize + rescue Exception => e raise Error, "Could not load #{action_orm}, skipping controller. Error: #{e.message}." end end -- cgit v1.2.3