From 3f954e2b9fecc1297b5f89cb13ff53df2d5e42ef Mon Sep 17 00:00:00 2001
From: Joe <1264204425@qq.com>
Date: Tue, 10 Sep 2024 10:46:35 +0800
Subject: [PATCH] feat: register service WorkSpaceNotAllowedCreateError

---
 api/services/account_service.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/api/services/account_service.py b/api/services/account_service.py
index 7cbbc8c428..3ee24f0a51 100644
--- a/api/services/account_service.py
+++ b/api/services/account_service.py
@@ -627,6 +627,9 @@ class RegisterService:
             tenant_was_created.send(tenant)
 
             db.session.commit()
+        except WorkSpaceNotAllowedCreateError:
+            db.session.rollback()
+            raise
         except Exception as e:
             db.session.rollback()
             logging.error(f"Register failed: {e}")