from rota_generator.shifts import SingleShift as SingleShiftBase, ShiftConstraint as ShiftConstraintBase
from rota_generator.workers import Worker as WorkerBase


class SingleShift(SingleShiftBase):

    id: int

    class Config:
        orm_mode = True

class SingleShiftCreate(SingleShiftBase):
    pass


#class ShiftConstraint(ShiftConstraintBase):
#    id: int
#
#    class Config:
#        orm_mode = True
#
#class ShiftConstraintCreate(ShiftConstraintBase):
#    pass

class Worker(WorkerBase):
    pass

    class Config:
        orm_mode = True

class WorkerCreate(WorkerBase):
    pass