diff --git a/app/db.py b/app/db.py index 4cf56fd..5a66b9e 100644 --- a/app/db.py +++ b/app/db.py @@ -31,6 +31,7 @@ class DB: except Exception as e: if in_transaction and self._connection: conn.rollback() + raise finally: if in_transaction: self._transaction_depth -= 1 @@ -126,7 +127,7 @@ class DB: def where(self, condition, operator = "="): if isinstance(condition, dict): for key, value in condition.items(): - self._where.append((key, "=", value)) + self._where.append((key, operator, value)) elif isinstance(condition, list): for c in condition: self._where.append(c)