From fa3d533354fb872904a84b42a31173193a3688c9 Mon Sep 17 00:00:00 2001 From: Jeremy Zhang Date: Sun, 25 Mar 2018 20:24:39 +0000 Subject: [PATCH] Wait until got session --- discordbot/titanembeds/database/__init__.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/discordbot/titanembeds/database/__init__.py b/discordbot/titanembeds/database/__init__.py index e2f3f6c..9f5f2e1 100644 --- a/discordbot/titanembeds/database/__init__.py +++ b/discordbot/titanembeds/database/__init__.py @@ -33,7 +33,13 @@ class DatabaseInterface(object): @contextmanager def get_session(self) -> Session: - session = self._sessionmaker() # type: Session + got_session = False + while not got_session: + try: + session = self._sessionmaker() # type: Session + got_session = True + except: + pass try: yield session session.commit()