From 4286f394b7db905caf506c1071b019b74f9517e7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rajmund=20Hru=C5=A1ka?= <rajmund.hruska@cesnet.cz>
Date: Wed, 12 Jan 2022 14:52:54 +0100
Subject: [PATCH] Fix: Use OperationalError from flask_sqlalchemy. (Redmine
 issue: #6205)

---
 lib/mentat/services/sqlstorage.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/lib/mentat/services/sqlstorage.py b/lib/mentat/services/sqlstorage.py
index 59a8d1981..c029f4ebb 100644
--- a/lib/mentat/services/sqlstorage.py
+++ b/lib/mentat/services/sqlstorage.py
@@ -29,7 +29,6 @@ __credits__ = "Pavel Kácha <pavel.kacha@cesnet.cz>, Andrea Kropáčová <andrea
 import copy
 from flask import Flask
 import flask_sqlalchemy
-from sqlalchemy.exc import OperationalError
 
 #
 # Custom libraries
@@ -50,7 +49,7 @@ class RetryingQuery(flask_sqlalchemy.orm.Query):
         for _ in range(2):
             try:
                 return super()._execute_and_instances(querycontext)
-            except OperationalError:
+            except flask_sqlalchemy.orm.exc.sa_exc.OperationalError:
                 self.session.close()
                 continue
 
-- 
GitLab