from typing import Optional

from schemas.base import BaseSchema
from schemas.payroll import PayrollSchemeSchema


class PayRateSchemaBase(BaseSchema):
    name: str
    payroll_scheme: PayrollSchemeSchema
    base_rate: float
    overtime_rate: Optional[float] = None
    overtime_threshold: Optional[float] = None


class PayRateSchema(PayRateSchemaBase):
    id: int