[DB] Attempt reconnection if current connection was lost

1 files changed, 8 insertions, 1 deletions

diff --git a/pysite/database.py b/pysite/database.py
index a8c6c559..bb0c3415 100644
--- a/pysite/database.py
+++ b/pysite/database.py
@@ -226,7 +226,14 @@ class RethinkDB:
         """
 
         if not new_connection:
-            result = query.run(self.conn)
+            try:
+                result = query.run(self.conn)
+            except rethinkdb.ReqlDriverError as e:
+                if e.message == "Connection is closed.":
+                    self.log.warning("Connection was closed, attempting with a new connection...")
+                    result = query.run(self.get_connection(connect_database))
+                else:
+                    raise
         else:
             result = query.run(self.get_connection(connect_database))