"""Company models for ROA2WEB applications."""

from typing import List, Optional
from pydantic import BaseModel


class Company(BaseModel):
    """Model for a company/firma."""
    id_firma: int
    name: str
    schema_name: str
    fiscal_code: Optional[str] = None
    is_active: bool = True


class CompanyListResponse(BaseModel):
    """Response model for list of companies."""
    companies: List[Company]
    total_count: int