zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-04 23:00:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refact: Tenant api (#14288)

Browse Source 
### What problem does this PR solve?

Refact: Tenant api

### Type of change

- [x] Refactoring
This commit is contained in:
Magicbook1108
2026-04-22 20:00:32 +08:00
committed by GitHub
parent 1434f8ade8
commit d1c62fc19d
5 changed files with 96 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

107
api/apps/tenant_app.py → api/apps/restful_apis/tenant_api.py
View File

@@ -1,5 +1,5 @@
#
# Copyright 2024 The InfiniFlow Authors. All Rights Reserved.
# Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
@@ -13,48 +13,56 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#
import logging
import asyncio
import logging
from api.apps import current_user, login_required
from api.db import UserTenantRole
from api.db.db_models import UserTenant
from api.db.services.user_service import UserTenantService, UserService
from api.db.services.user_service import UserService, UserTenantService
from api.utils.api_utils import (
get_data_error_result,
get_json_result,
get_request_json,
server_error_response,
validate_request,
)
from api.utils.web_utils import send_invite_email
from common import settings
from common.constants import RetCode, StatusEnum
from common.misc_utils import get_uuid
from common.time_utils import delta_seconds
from api.utils.api_utils import get_data_error_result, get_json_result, get_request_json, server_error_response, validate_request
from api.utils.web_utils import send_invite_email
from common import settings
from api.apps import login_required, current_user
@manager.route("/<tenant_id>/user/list", methods=["GET"]) # noqa: F821
@manager.route("/tenants/<tenant_id>/users", methods=["GET"]) # noqa: F821
@login_required
def user_list(tenant_id):
if current_user.id != tenant_id:
return get_json_result(
data=False,
message='No authorization.',
code=RetCode.AUTHENTICATION_ERROR)
message="No authorization.",
code=RetCode.AUTHENTICATION_ERROR,
)
try:
users = UserTenantService.get_by_tenant_id(tenant_id)
for u in users:
u["delta_seconds"] = delta_seconds(str(u["update_date"]))
for user in users:
user["delta_seconds"] = delta_seconds(str(user["update_date"]))
return get_json_result(data=users)
except Exception as e:
return server_error_response(e)
except Exception as exc:
return server_error_response(exc)
@manager.route('/<tenant_id>/user', methods=['POST']) # noqa: F821
@manager.route("/tenants/<tenant_id>/users", methods=["POST"]) # noqa: F821
@login_required
@validate_request("email")
async def create(tenant_id):
if current_user.id != tenant_id:
return get_json_result(
data=False,
message='No authorization.',
code=RetCode.AUTHENTICATION_ERROR)
message="No authorization.",
code=RetCode.AUTHENTICATION_ERROR,
)
req = await get_request_json()
invite_user_email = req["email"]
@@ -71,7 +79,8 @@ async def create(tenant_id):
if user_tenant_role == UserTenantRole.OWNER:
return get_data_error_result(message=f"{invite_user_email} is the owner of the team.")
return get_data_error_result(
message=f"{invite_user_email} is in the team, but the role: {user_tenant_role} is invalid.")
message=f"{invite_user_email} is in the team, but the role: {user_tenant_role} is invalid."
)
UserTenantService.save(
id=get_uuid(),
@@ -79,10 +88,10 @@ async def create(tenant_id):
tenant_id=tenant_id,
invited_by=current_user.id,
role=UserTenantRole.INVITE,
status=StatusEnum.VALID.value)
status=StatusEnum.VALID.value,
)
try:
user_name = ""
_, user = UserService.get_by_id(current_user.id)
if user:
@@ -93,52 +102,62 @@ async def create(tenant_id):
to_email=invite_user_email,
invite_url=settings.MAIL_FRONTEND_URL,
tenant_id=tenant_id,
inviter=user_name or current_user.email
inviter=user_name or current_user.email,
)
)
except Exception as e:
logging.exception(f"Failed to send invite email to {invite_user_email}: {e}")
return get_json_result(data=False, message="Failed to send invite email.", code=RetCode.SERVER_ERROR)
usr = invite_users[0].to_dict()
usr = {k: v for k, v in usr.items() if k in ["id", "avatar", "email", "nickname"]}
except Exception as exc:
logging.exception(f"Failed to send invite email to {invite_user_email}: {exc}")
return get_json_result(
data=False,
message="Failed to send invite email.",
code=RetCode.SERVER_ERROR,
)
return get_json_result(data=usr)
user = invite_users[0].to_dict()
user = {k: v for k, v in user.items() if k in ["id", "avatar", "email", "nickname"]}
return get_json_result(data=user)
@manager.route('/<tenant_id>/user/<user_id>', methods=['DELETE']) # noqa: F821
@manager.route("/tenants/<tenant_id>/users", methods=["DELETE"]) # noqa: F821
@login_required
def rm(tenant_id, user_id):
@validate_request("user_id")
async def rm(tenant_id):
req = await get_request_json()
user_id = req["user_id"]
if current_user.id != tenant_id and current_user.id != user_id:
return get_json_result(
data=False,
message='No authorization.',
code=RetCode.AUTHENTICATION_ERROR)
message="No authorization.",
code=RetCode.AUTHENTICATION_ERROR,
)
try:
UserTenantService.filter_delete([UserTenant.tenant_id == tenant_id, UserTenant.user_id == user_id])
return get_json_result(data=True)
except Exception as e:
return server_error_response(e)
except Exception as exc:
return server_error_response(exc)
@manager.route("/list", methods=["GET"]) # noqa: F821
@manager.route("/tenants", methods=["GET"]) # noqa: F821
@login_required
def tenant_list():
try:
users = UserTenantService.get_tenants_by_user_id(current_user.id)
for u in users:
u["delta_seconds"] = delta_seconds(str(u["update_date"]))
for user in users:
user["delta_seconds"] = delta_seconds(str(user["update_date"]))
return get_json_result(data=users)
except Exception as e:
return server_error_response(e)
except Exception as exc:
return server_error_response(exc)
@manager.route("/agree/<tenant_id>", methods=["PUT"]) # noqa: F821
@manager.route("/tenants/<tenant_id>", methods=["PATCH"]) # noqa: F821
@login_required
def agree(tenant_id):
try:
UserTenantService.filter_update([UserTenant.tenant_id == tenant_id, UserTenant.user_id == current_user.id],
{"role": UserTenantRole.NORMAL})
UserTenantService.filter_update(
[UserTenant.tenant_id == tenant_id, UserTenant.user_id == current_user.id],
{"role": UserTenantRole.NORMAL},
)
return get_json_result(data=True)
except Exception as e:
return server_error_response(e)
except Exception as exc:
return server_error_response(exc)