add init block support to #create and #create!

pelle · May 31, 2011 · 2b9a519 · 2b9a519
1 parent 4e19fdf
commit 2b9a519
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/couchrest/model/persistence.rb b/lib/couchrest/model/persistence.rb
@@ -118,8 +118,8 @@ def build_from_database(doc = {})
         #
         # ==== Returns
         #  returns the reloaded document
-        def create(attributes = {})
-          instance = new(attributes)
+        def create(attributes = {}, &block)
+          instance = new(attributes, &block)
           instance.create
           instance
         end
@@ -128,8 +128,8 @@ def create(attributes = {})
         #
         # ==== Returns
         #  returns the reloaded document or raises an exception
-        def create!(attributes = {})
-          instance = new(attributes)
+        def create!(attributes = {}, &block)
+          instance = new(attributes, &block)
           instance.create!
           instance
         end