Merge pull request #35 from juntatalor/juntatalor-patch-return_id_list

Fix return_id_list()
05bit · Sep 5, 2016 · 743b1e9 · 743b1e9
2 parents 515f3e5 + acd306e
commit 743b1e9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/peewee_async.py b/peewee_async.py
@@ -585,7 +585,10 @@ def insert(query):
     cursor = yield from _execute_query_async(query)
 
     if query.is_insert_returning:
-        result = (yield from cursor.fetchone())[0]
+        if query._return_id_list:
+            result = map(lambda x: x[0], (yield from cursor.fetchall()))
+        else:
+            result = (yield from cursor.fetchone())[0]
     else:
         result = yield from query.database.last_insert_id_async(
             cursor, query.model_class)