from typing import List

from schemas.base import BaseSchema, OkMessageSchema
from schemas.payrate import PayRateSchemaBase, PayRateSchema
from schemas.payroll import PayrollSchemeSchema


# region Requests

class CreatePayRateRequest(BaseSchema):
    data: PayRateSchemaBase


class UpdatePayRateRequest(BaseSchema):
    data: PayRateSchema


class DeletePayRateRequest(BaseSchema):
    pay_rate_id: int


# endregion

# region Responses

class GetAllPayrollSchemeResponse(BaseSchema):
    payroll_schemas: List[PayrollSchemeSchema]


class GetAllPayRatesResponse(BaseSchema):
    pay_rates: List[PayRateSchema]


class CreatePayRateResponse(OkMessageSchema):
    pass


class UpdatePayRateResponse(OkMessageSchema):
    pass


class DeletePayRateResponse(OkMessageSchema):
    pass
# endregion