From 4d81ab9bfc2b3c3add8dd7034d801b4f7bc616fc Mon Sep 17 00:00:00 2001
From: Keiran Price <keiran.price@stfc.ac.uk>
Date: Mon, 30 Sep 2019 10:55:16 +0100
Subject: [PATCH] #49: Add refresh to inserted entities

---
 common/database_helpers.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/common/database_helpers.py b/common/database_helpers.py
index b2bcd754..b9e348c7 100644
--- a/common/database_helpers.py
+++ b/common/database_helpers.py
@@ -86,6 +86,7 @@ class CreateQuery(Query):
     def __init__(self, table, row):
         super().__init__(table)
         self.row = row
+        self.inserted_row = None
 
     def execute_query(self):
         """Determines if the row is a row object or dictionary then commits it to the table"""
@@ -100,7 +101,8 @@ def execute_query(self):
             record.MOD_ID = "user"  # These will need changing
         self.session.add(record)
         self.commit_changes()
-
+        self.session.refresh(record)
+        self.inserted_row = record
 
 class UpdateQuery(Query):