简单优化一下格式,结束了
This commit is contained in:
parent
322b3408ab
commit
1d3c024b35
@ -13,10 +13,10 @@ from fastapi import HTTPException, Response
|
|||||||
from models import Tenant, User # 假设你已导入 Tenant 和 User 模型
|
from models import Tenant, User # 假设你已导入 Tenant 和 User 模型
|
||||||
from dependencies import SessionDep # 假设 SessionDep 是数据库会话的依赖
|
from dependencies import SessionDep # 假设 SessionDep 是数据库会话的依赖
|
||||||
|
|
||||||
#列举所有租户
|
|
||||||
|
# 列举所有租户
|
||||||
@router.get("/api/s1/tenant")
|
@router.get("/api/s1/tenant")
|
||||||
async def get_tenant(response: Response, session: SessionDep, current_user: User = Depends(get_current_user)):
|
async def get_tenant(response: Response, session: SessionDep, current_user: User = Depends(get_current_user)):
|
||||||
|
|
||||||
if current_user.role != 0:
|
if current_user.role != 0:
|
||||||
raise HTTPException(status_code=403, detail="Only Superadmin can list all tenants.")
|
raise HTTPException(status_code=403, detail="Only Superadmin can list all tenants.")
|
||||||
|
|
||||||
@ -43,6 +43,7 @@ async def get_tenant(response: Response, session: SessionDep, current_user: User
|
|||||||
|
|
||||||
return {"tenants": tenant_data}
|
return {"tenants": tenant_data}
|
||||||
|
|
||||||
|
|
||||||
# 新增和修改租户
|
# 新增和修改租户
|
||||||
@router.post("/api/s1/tenant")
|
@router.post("/api/s1/tenant")
|
||||||
async def create_or_update_tenant(data: dict, session: SessionDep, current_user: User = Depends(get_current_user)):
|
async def create_or_update_tenant(data: dict, session: SessionDep, current_user: User = Depends(get_current_user)):
|
||||||
@ -101,19 +102,19 @@ async def create_or_update_tenant(data: dict, session: SessionDep, current_user:
|
|||||||
|
|
||||||
user = session.exec(user_query).first()
|
user = session.exec(user_query).first()
|
||||||
|
|
||||||
#如果找不到对应的用户,抛出错误
|
# 如果找不到对应的用户,抛出错误
|
||||||
if not user:
|
if not user:
|
||||||
raise HTTPException(status_code=404, detail="User not found")
|
raise HTTPException(status_code=404, detail="User not found")
|
||||||
user.password = password
|
user.password = password
|
||||||
session.add(user)
|
session.add(user)
|
||||||
session.commit()
|
session.commit()
|
||||||
print(user) #测试用
|
print(user) # 测试用
|
||||||
return {"message": "Tenant and User update successfully"}
|
return {"message": "Tenant and User update successfully"}
|
||||||
|
|
||||||
#删除租户
|
|
||||||
|
# 删除租户
|
||||||
@router.delete("/api/s1/tenant")
|
@router.delete("/api/s1/tenant")
|
||||||
async def delete_tenant(data: dict, session: SessionDep, current_user: User = Depends(get_current_user)):
|
async def delete_tenant(data: dict, session: SessionDep, current_user: User = Depends(get_current_user)):
|
||||||
|
|
||||||
if current_user.role != 0:
|
if current_user.role != 0:
|
||||||
raise HTTPException(status_code=403, detail="Only Superadmin can delete tenants.")
|
raise HTTPException(status_code=403, detail="Only Superadmin can delete tenants.")
|
||||||
|
|
||||||
|
|||||||
@ -5,3 +5,4 @@ pydantic~=2.9.2
|
|||||||
pydantic-settings~=2.6.1
|
pydantic-settings~=2.6.1
|
||||||
mysqlclient
|
mysqlclient
|
||||||
sqlmodel~=0.0.22
|
sqlmodel~=0.0.22
|
||||||
|
sqlalchemy~=2.0.36
|
||||||
Loading…
Reference in New Issue
Block a user