CostEvalPlatform/api/manage_tanant.py

# -*- coding: utf-8 -*-
# @Time    : 2024/11/19 下午8:04
# @FileName: manage_tanant.py
# @Software: PyCharm
from fastapi import HTTPException, Response, Depends, APIRouter
from typing import Optional, Annotated
from datetime import datetime, timedelta
from jose import JWTError, jwt

from sqlmodel import select

from models import Tenant, User, Project
from dependencies import *

router = APIRouter()

# 推送修改用注释
# @router.get(...)
# def example():
#     return "hello"

# 列举所有租户
from fastapi import HTTPException, Response
from sqlalchemy.orm import Session
from models import Tenant, User  # 假设你已导入 Tenant 和 User 模型
from dependencies import SessionDep  # 假设 SessionDep 是数据库会话的依赖


#列举所有租户
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):
    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_tenant(data: dict, session: SessionDep):
    tenant_id = data.get("tenant_id")
    name = data["name"]
    username = data["username"]
    password = data["password"]

    # 验证是否缺少必要参数
    if not name:
        raise HTTPException(status_code=400, detail="Need to provide name")


    # 更新租户还是新增租户
    if tenant_id:
        # 查找现有租户
        tenant = session.get(Tenant, tenant_id)
        if not tenant:
            raise HTTPException(status_code=404, detail="Tenant not found")

        # 更新项目内容
        tenant.name = name
    else:
        # 新增项目
        project = Project(
            name=name,
            requirement=requirement,
            start_time=start_time,
            deadline=deadline,
            owner_id=1  # 假设owner_id是1，之后应该是通过token获取owner_id吧
        )
        session.add(project)

    # 处理项目和用户的关联
    # 先清除现有的关联
    # 生成删除语句并执行
    print(project_id) #测试用
    stmt = delete(ProjectUserLink).where(ProjectUserLink.project_id == project.id)
    session.execute(stmt)
    session.commit()  # 提交事务

    # 重新建立与评估员和审核员的关系
    for username in estimators:
        user = next((user for user in users_estimators if user.username == username), None)
        if user:
            project_user_link = ProjectUserLink(project_id=project.id, user_id=user.id)
            session.add(project_user_link)

    for username in auditors:
        user = next((user for user in users_auditors if user.username == username), None)
        if user:
            project_user_link = ProjectUserLink(project_id=project.id, user_id=user.id)
            session.add(project_user_link)

    # 提交事务
    session.commit()
    session.refresh(project)

    return {"message": "Added or updated successfully",
            "information": project,
            }