"""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()