"""企业实体 API"""
from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from typing import List

from app.database import get_db
from app.auth_middleware import require_auth
from app.models import Entity

router = APIRouter(prefix="/api/cma/entities", tags=["企业实体"],
    dependencies=[Depends(require_auth)],
)


@router.get("")
def list_entities(db: Session = Depends(get_db)):
    """获取企业列表"""
    entities = db.query(Entity).filter(Entity.status == "active").order_by(Entity.id).all()
    return {
        "data": [
            {
                "id": e.id,
                "name": e.name,
                "short_name": e.short_name,
                "industry": e.industry,
            }
            for e in entities
        ]
    }