104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
"""Tenant management API routes."""
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
from ..database import get_db
|
|
from ..models.tenant import Tenant
|
|
from ..models.user import User
|
|
from ..schemas.tenant import TenantCreate, TenantUpdate, TenantResponse
|
|
from .deps import require_super_admin
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/", response_model=list[TenantResponse])
|
|
def list_tenants(
|
|
skip: int = 0,
|
|
limit: int = 100,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(require_super_admin)
|
|
):
|
|
"""List all tenants (super admin only)."""
|
|
tenants = db.query(Tenant).offset(skip).limit(limit).all()
|
|
return tenants
|
|
|
|
|
|
@router.post("/", response_model=TenantResponse, status_code=status.HTTP_201_CREATED)
|
|
def create_tenant(
|
|
tenant_data: TenantCreate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(require_super_admin)
|
|
):
|
|
"""Create a new tenant (super admin only)."""
|
|
# Check if name already exists
|
|
existing = db.query(Tenant).filter(Tenant.name == tenant_data.name).first()
|
|
if existing:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="Tenant with this name already exists"
|
|
)
|
|
|
|
tenant = Tenant(**tenant_data.model_dump())
|
|
db.add(tenant)
|
|
db.commit()
|
|
db.refresh(tenant)
|
|
return tenant
|
|
|
|
|
|
@router.get("/{tenant_id}", response_model=TenantResponse)
|
|
def get_tenant(
|
|
tenant_id: int,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(require_super_admin)
|
|
):
|
|
"""Get tenant by ID (super admin only)."""
|
|
tenant = db.query(Tenant).filter(Tenant.id == tenant_id).first()
|
|
if not tenant:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="Tenant not found"
|
|
)
|
|
return tenant
|
|
|
|
|
|
@router.put("/{tenant_id}", response_model=TenantResponse)
|
|
def update_tenant(
|
|
tenant_id: int,
|
|
tenant_data: TenantUpdate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(require_super_admin)
|
|
):
|
|
"""Update tenant (super admin only)."""
|
|
tenant = db.query(Tenant).filter(Tenant.id == tenant_id).first()
|
|
if not tenant:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="Tenant not found"
|
|
)
|
|
|
|
update_data = tenant_data.model_dump(exclude_unset=True)
|
|
for field, value in update_data.items():
|
|
setattr(tenant, field, value)
|
|
|
|
db.commit()
|
|
db.refresh(tenant)
|
|
return tenant
|
|
|
|
|
|
@router.delete("/{tenant_id}", status_code=status.HTTP_204_NO_CONTENT)
|
|
def delete_tenant(
|
|
tenant_id: int,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(require_super_admin)
|
|
):
|
|
"""Delete tenant (super admin only)."""
|
|
tenant = db.query(Tenant).filter(Tenant.id == tenant_id).first()
|
|
if not tenant:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="Tenant not found"
|
|
)
|
|
|
|
db.delete(tenant)
|
|
db.commit()
|