feat: register service WorkSpaceNotAllowedCreateError

2024-09-10 10:46:35 +08:00 · 2024-09-10 10:46:35 +08:00 · 3f954e2b9f
commit 3f954e2b9f
parent 4893631d65
1 changed files with 3 additions and 0 deletions
--- 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}")