feat: add db transaction

api/services/account_service.py

@@ -258,23 +258,20 @@ class TenantService:
     def switch_tenant(account: Account, tenant_id: int = None) -> None:
         """Switch the current workspace for the account"""
 
-        with db.session.begin():
-            try:
-                tenant_account_join = TenantAccountJoin.query.filter_by(account_id=account.id, tenant_id=tenant_id).first()
-                TenantAccountJoin.query.filter_by(account_id=account.id).update({'current': False})
-
-                # Check if the tenant exists and the account is a member of the tenant
-                if not tenant_account_join:
-                    raise AccountNotLinkTenantError("Tenant not found or account is not a member of the tenant.")
-                else:
+        tenant_account_join = TenantAccountJoin.query.filter_by(account_id=account.id, tenant_id=tenant_id).first()
+        if not tenant_account_join:
+            raise AccountNotLinkTenantError("Tenant not found or account is not a member of the tenant.")
+        else: 
+            with db.session.begin():
+                try:
+                    TenantAccountJoin.query.filter_by(account_id=account.id).update({'current': False})
                     tenant_account_join.current = True
                     db.session.commit()
-
-                # Set the current tenant for the account
-                account.current_tenant_id = tenant_account_join.tenant_id
-            except exc.SQLAlchemyError:
-                db.session.rollback()
-                raise
+                    # Set the current tenant for the account
+                    account.current_tenant_id = tenant_account_join.tenant_id
+                except exc.SQLAlchemyError:
+                    db.session.rollback()
+                    raise
 
     @staticmethod
     def get_tenant_members(tenant: Tenant) -> List[Account]: