From 18df35b027e10c13ea4ce16e9b9d9b08f832ee7e Mon Sep 17 00:00:00 2001 From: Fluxticks Date: Sun, 13 Feb 2022 19:11:53 +0000 Subject: [PATCH] Added rollback to when there are db errors --- src/esportsbot/db_gateway.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/esportsbot/db_gateway.py b/src/esportsbot/db_gateway.py index 423fff9d..c82f2500 100644 --- a/src/esportsbot/db_gateway.py +++ b/src/esportsbot/db_gateway.py @@ -44,6 +44,7 @@ def list(db_model, **args): query = session.query(db_model).filter_by(**args).all() return query except Exception as err: + session.rollback() raise Exception(f"Error occurred when using list - {err}") @staticmethod @@ -62,6 +63,7 @@ def get(db_model, **args): query = session.query(db_model).filter_by(**args).all() return query[0] if query != [] else query except Exception as err: + session.rollback() raise Exception(f"Error occurred when using get - {err}") @staticmethod @@ -76,6 +78,7 @@ def update(model): session.add(model) session.commit() except Exception as err: + session.rollback() raise Exception(f"Error occurred when using update - {err}") @staticmethod @@ -90,6 +93,7 @@ def delete(model): session.delete(model) session.commit() except Exception as err: + session.rollback() raise Exception(f"Error occurred when using delete - {err}") @staticmethod @@ -104,4 +108,5 @@ def create(model): session.add(model) session.commit() except Exception as err: + session.rollback() raise Exception(f"Error occurred when using create - {err}")