Merge pull request #28 from AnswerDotAI/add-returning-to-writes

Add returning to basic writes (INSERT, UPDATE, DELETE)

Author: jph00

sqlite_minutils/db.py

@@ -2804,7 +2804,7 @@ def build_insert_queries_and_params(
                 # them since it ignores the resulting integrity errors
                 if not_null:
                     placeholders.extend(not_null)
-                sql = "INSERT OR IGNORE INTO [{table}]({cols}) VALUES({placeholders});".format(
+                sql = "INSERT OR IGNORE INTO [{table}]({cols}) VALUES({placeholders}) RETURNING *;".format(
                     table=self.name,
                     cols=", ".join(["[{}]".format(p) for p in placeholders]),
                     placeholders=", ".join(["?" for p in placeholders]),
@@ -2815,7 +2815,7 @@ def build_insert_queries_and_params(
                 # UPDATE [book] SET [name] = 'Programming' WHERE [id] = 1001;
                 set_cols = [col for col in all_columns if col not in pks]
                 if set_cols:
-                    sql2 = "UPDATE [{table}] SET {pairs} WHERE {wheres}".format(
+                    sql2 = "UPDATE [{table}] SET {pairs} WHERE {wheres} RETURNING *".format(
                         table=self.name,
                         pairs=", ".join(
                             "[{}] = {}".format(col, conversions.get(col, "?"))
@@ -2843,7 +2843,7 @@ def build_insert_queries_and_params(
             elif ignore:
                 or_what = "OR IGNORE "
             sql = """
-                INSERT {or_what}INTO [{table}] ({columns}) VALUES {rows};
+                INSERT {or_what}INTO [{table}] ({columns}) VALUES {rows} RETURNING *;
             """.strip().format(
                 or_what=or_what,
                 table=self.name,