Commit eebb38cd authored by crazywoola's avatar crazywoola

feat: add db transaction

parent 653259c8
...@@ -258,18 +258,15 @@ class TenantService: ...@@ -258,18 +258,15 @@ class TenantService:
def switch_tenant(account: Account, tenant_id: int = None) -> None: def switch_tenant(account: Account, tenant_id: int = None) -> None:
"""Switch the current workspace for the account""" """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() 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: if not tenant_account_join:
raise AccountNotLinkTenantError("Tenant not found or account is not a member of the tenant.") raise AccountNotLinkTenantError("Tenant not found or account is not a member of the tenant.")
else: else:
with db.session.begin():
try:
TenantAccountJoin.query.filter_by(account_id=account.id).update({'current': False})
tenant_account_join.current = True tenant_account_join.current = True
db.session.commit() db.session.commit()
# Set the current tenant for the account # Set the current tenant for the account
account.current_tenant_id = tenant_account_join.tenant_id account.current_tenant_id = tenant_account_join.tenant_id
except exc.SQLAlchemyError: except exc.SQLAlchemyError:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment