147 lines
4.9 KiB
Python
147 lines
4.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# @Time : 2024/11/19 下午8:04
|
|
# @FileName: manage_tanant.py
|
|
# @Software: PyCharm
|
|
from fastapi import APIRouter
|
|
from fastapi import HTTPException, Response, Depends, APIRouter
|
|
from typing import Optional, Annotated
|
|
from datetime import datetime, timedelta
|
|
from jose import JWTError, jwt
|
|
from sqlalchemy import delete
|
|
|
|
from sqlmodel import select
|
|
|
|
from models import Tenant, User, Project
|
|
from dependencies import *
|
|
|
|
router = APIRouter()
|
|
|
|
from fastapi import HTTPException, Response
|
|
from sqlalchemy.orm import Session
|
|
from models import Tenant, User # 假设你已导入 Tenant 和 User 模型
|
|
from dependencies import SessionDep # 假设 SessionDep 是数据库会话的依赖
|
|
|
|
#列举所有租户
|
|
@router.get("/api/s1/tenant")
|
|
async def get_tenant(response: Response, session: SessionDep, current_user: User = Depends(get_current_user)):
|
|
|
|
if current_user.role != 0:
|
|
raise HTTPException(status_code=403, detail="Only Superadmin can list all tenants.")
|
|
|
|
tenants = session.query(Tenant).all() # 获取所有租户
|
|
if not tenants:
|
|
raise HTTPException(status_code=404, detail="No tenants found")
|
|
|
|
tenant_data = []
|
|
for tenant in tenants:
|
|
# 获取该租户中 role=1 的第一个用户（如果存在）
|
|
tenant_user = next((user for user in tenant.users if user.role == 1), None)
|
|
|
|
# 获取该租户中除了 role=1 以外的用户数量
|
|
user_num = len([user for user in tenant.users if user.role != 1])
|
|
|
|
# 构建租户信息
|
|
tenant_info = {
|
|
"name": tenant.name,
|
|
"username": tenant_user.username if tenant_user else None, # 如果找到 role=1 的用户，返回其 username
|
|
"user_num": user_num # 除去 role=1 的用户数量
|
|
}
|
|
|
|
tenant_data.append(tenant_info)
|
|
|
|
return {"tenants": tenant_data}
|
|
|
|
# 新增和修改租户
|
|
@router.post("/api/s1/tenant")
|
|
async def create_or_update_tenant(data: dict, session: SessionDep, current_user: User = Depends(get_current_user)):
|
|
if current_user.role != 0:
|
|
raise HTTPException(status_code=403, detail="Only Superadmin can add or update tenants.")
|
|
|
|
name = data.get("name")
|
|
username = data.get("username")
|
|
password = data.get("password")
|
|
|
|
# 验证是否缺少必要参数
|
|
if not name:
|
|
raise HTTPException(status_code=400, detail="Need more name")
|
|
|
|
if username:
|
|
# 如果 username 不为空，判断为新建租户
|
|
# 检查租户名是否已存在
|
|
tenant_query = select(Tenant).where(Tenant.name == name)
|
|
existing_tenant = session.exec(tenant_query).first()
|
|
|
|
if existing_tenant:
|
|
raise HTTPException(status_code=409, detail="Tenant name already exists")
|
|
|
|
# 创建新租户
|
|
tenant = Tenant(name=name)
|
|
session.add(tenant)
|
|
session.commit()
|
|
session.refresh(tenant)
|
|
|
|
# 创建新用户
|
|
user = User(
|
|
username=username,
|
|
password=password, # 记得加密密码
|
|
role=1, # 默认role为1
|
|
tenant_id=tenant.id,
|
|
)
|
|
session.add(user)
|
|
|
|
# 提交事务
|
|
session.commit()
|
|
|
|
return {"message": "Tenant and User added successfully"}
|
|
|
|
else:
|
|
# 如果 username 为空，执行更新操作
|
|
# 根据租户名称查找 Tenant
|
|
tenant_query = select(Tenant).where(Tenant.name == name)
|
|
tenant = session.exec(tenant_query).first()
|
|
|
|
# 如果找不到对应的租户，抛出错误
|
|
if not tenant:
|
|
raise HTTPException(status_code=404, detail="Tenant not found")
|
|
|
|
# 找到租户后，根据 tenant_id 查找该租户下的所有用户
|
|
user_query = select(User).where(User.tenant_id == tenant.id)
|
|
|
|
user = session.exec(user_query).first()
|
|
|
|
#如果找不到对应的用户，抛出错误
|
|
if not user:
|
|
raise HTTPException(status_code=404, detail="User not found")
|
|
user.password = password
|
|
session.add(user)
|
|
session.commit()
|
|
print(user) #测试用
|
|
return {"message": "Tenant and User update successfully"}
|
|
|
|
#删除租户
|
|
@router.delete("/api/s1/tenant")
|
|
async def delete_tenant(data: dict, session: SessionDep, current_user: User = Depends(get_current_user)):
|
|
|
|
if current_user.role != 0:
|
|
raise HTTPException(status_code=403, detail="Only Superadmin can delete tenants.")
|
|
|
|
tenant_name = data.get("name")
|
|
|
|
if not tenant_name:
|
|
raise HTTPException(status_code=400, detail="Tenant name is required")
|
|
|
|
# 查找租户
|
|
tenant = session.exec(
|
|
select(Tenant).where(Tenant.name == tenant_name)).first()
|
|
if not tenant:
|
|
raise HTTPException(status_code=404, detail="Tenant not found")
|
|
|
|
# 删除与租户相关的用户
|
|
stmt = delete(User).where(User.tenant_id == tenant.id)
|
|
session.execute(stmt)
|
|
|
|
# 删除租户
|
|
session.delete(tenant)
|
|
session.commit()
|
|
|
|
return {"detail": "Tenant deleted successfully"} |