Merge f43aa22 into 37365a1

msiemens · Jul 22, 2014 · dbe7131 · dbe7131
2 parents 37365a1 + f43aa22
commit dbe7131
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/tests/test_tables.py b/tests/test_tables.py
@@ -4,9 +4,8 @@
 def test_one_table(db):
     table1 = db.table('table1')
 
-    table1.insert({'int': 1, 'char': 'a'})
-    table1.insert({'int': 1, 'char': 'b'})
-    table1.insert({'int': 1, 'char': 'c'})
+    docs = [{'int':1, 'char':x} for x in 'abc']
+    table1.insert_multiple(docs)
 
     assert table1.get(where('int') == 1)['char'] == 'a'
     assert table1.get(where('char') == 'b')['char'] == 'b'

diff --git a/tinydb/database.py b/tinydb/database.py
@@ -229,6 +229,15 @@ def insert(self, element):
 
         self._write(data)
 
+    def insert_multiple(self, documents):
+        """
+        Insert multiple elements into the table.
+        Uses ``Table.insert``internally, so refer to the docs.
+        """
+
+        for item in documents:
+            self.insert(item)
+
     def remove(self, cond):
         """
         Remove the element matching the condition.