Fix: #7: return inserted object ID for UUID primary key

05bit · Aug 26, 2015 · e811af7 · e811af7
1 parent a2d1985
commit e811af7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/peewee_async.py b/peewee_async.py
@@ -219,8 +219,12 @@ def insert(query):
          "with wrong query class %s" % str(query))
 
     cursor = yield from cursor_with_query(query)
-    result = yield from query.database.last_insert_id_async(
-        cursor, query.model_class)
+
+    if query.is_insert_returning:
+        result = yield from cursor.fetchone()
+    else:
+        result = yield from query.database.last_insert_id_async(
+            cursor, query.model_class)
 
     cursor.release()
     if result:

diff --git a/tests.py b/tests.py
@@ -215,7 +215,6 @@ def test():
         last_id = self.run_until_complete(test())
         self.assertTrue(last_id is not None)
 
-    @unittest.skip("not implemented")
     def test_insert_one_row_uuid_query(self):
         @asyncio.coroutine
         def test():