feat: add invite menmber workspace all create

pull/8904/head
Joe 2 years ago
parent 7e4d105e8b
commit 5165894b1a

@ -684,17 +684,15 @@ class RegisterService:
if open_id is not None or provider is not None: if open_id is not None or provider is not None:
AccountService.link_account_integrate(provider, open_id, account) AccountService.link_account_integrate(provider, open_id, account)
if not dify_config.ALLOW_CREATE_WORKSPACE: if dify_config.ALLOW_CREATE_WORKSPACE:
raise WorkSpaceNotAllowedCreateError() tenant = TenantService.create_tenant(f"{account.name}'s Workspace")
tenant = TenantService.create_tenant(f"{account.name}'s Workspace") TenantService.create_tenant_member(tenant, account, role="owner")
TenantService.create_tenant_member(tenant, account, role="owner") account.current_tenant = tenant
account.current_tenant = tenant tenant_was_created.send(tenant)
tenant_was_created.send(tenant)
db.session.commit() db.session.commit()
except WorkSpaceNotAllowedCreateError: except WorkSpaceNotAllowedCreateError:
db.session.rollback() db.session.rollback()
raise
except Exception as e: except Exception as e:
db.session.rollback() db.session.rollback()
logging.error(f"Register failed: {e}") logging.error(f"Register failed: {e}")

Loading…
Cancel
Save