Recreate DB connection if it is closed

1 files changed, 3 insertions, 0 deletions

diff --git a/db.py b/db.py
index f09a76a..c731f4e 100644
--- a/db.py
+++ b/db.py
@@ -54,6 +54,9 @@ class Db:
   def get_cursor(self):
     for _ in range(self.MAX_RETRIES):
       try:
+        if self.conn.closed:
+          self.connect()
+
         return self.conn.cursor()
       except psycopg2.Error:
         self.connect()