fix exception handling of sqlalchemy

Miguel Barão
1 parent 66a8c53f
Showing 1 changed file with 6 additions and 5 deletions Show diff stats
perguntations/app.py
@@ -10,9 +10,9 @@ import json
 import logging
 from os import path
-# user installed packages
+# installed packages
 import bcrypt
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, exc
 from sqlalchemy.orm import sessionmaker
 # this project
@@ -73,9 +73,10 @@ class App():
         try:
             yield session
             session.commit()
-        except Exception:
+        except exc.SQLAlchemyError:
             logger.error('DB rollback!!!')
             session.rollback()
+            raise
         finally:
             session.close()
@@ -359,7 +360,7 @@ class App():
         try:
             with self.db_session() as sess:
                 sess.add(Student(id=uid, name=name, password=''))
-        except Exception:
-            logger.error('Insert failed: student %s already exists.', uid)
+        except exc.SQLAlchemyError:
+            logger.error('Insert failed: student %s already exists?', uid)
         else:
             logger.info('New student inserted: %s, %s', uid, name)