from datetime import datetime
from typing import List

from schemas.base import OkMessageSchema, BaseSchema
from schemas.user import UserSchema


# region Entities

class ActiveWorkShiftSchema(BaseSchema):
    id: int
    started_at: datetime
    user: UserSchema


# endregion

# region Responses

class StartShiftResponse(OkMessageSchema):
    pass


class FinishShiftResponse(OkMessageSchema):
    pass


class FinishShiftByIdResponse(OkMessageSchema):
    pass


class ActiveWorkShiftsResponse(BaseSchema):
    shifts: List[ActiveWorkShiftSchema]


class DeleteShiftResponse(OkMessageSchema):
    pass

# endregion