Ability to call Model.for with class constant

piotrj · Jun 14, 2011 · 4303506 · 4303506
1 parent 6b8bedd
commit 4303506
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/administer/model.rb b/lib/administer/model.rb
@@ -17,7 +17,8 @@ def all
       end
 
       def for(model_name)
-        Model.new(lookup(model_name))
+        klass = model_name.is_a?(Class) ? model_name : lookup(model_name)
+        Model.new(klass)
       end
 
       private