from pydantic import BaseModel
from ApiLayers.ApiValidations.Request import (
    Login,
    EmployeeSelection,
    OccupantSelection,
)


class LoginSuperUserResponseModel(BaseModel):
    pass


class SelectCompanyOrOccupantTypeSuperUserRequestModel(BaseModel):
    pass


class SelectCompanyOrOccupantTypeSuperUserResponseModel(BaseModel):
    pass


class EmployeeSelectionSuperUserRequestModel(BaseModel):
    pass


class EmployeeSelectionSuperUserResponseModel(BaseModel):
    pass


class OccupantSelectionSuperUserRequestModel(BaseModel):
    pass


class OccupantSelectionSuperUserResponseModel(BaseModel):
    pass


class AuthenticationRequestModels:
    LoginSuperUserRequestModel = Login
    SelectCompanyOrOccupantTypeSuperUserRequestModel = {
        "EmployeeSelection": EmployeeSelection,
        "OccupantSelection": OccupantSelection,
    }
    EmployeeSelectionSuperUserRequestModel = EmployeeSelectionSuperUserRequestModel
    OccupantSelectionSuperUserRequestModel = OccupantSelectionSuperUserRequestModel


class AuthenticationResponseModels:
    LoginSuperUserResponseModel = LoginSuperUserResponseModel
    SelectCompanyOrOccupantTypeSuperUserResponseModel = (
        SelectCompanyOrOccupantTypeSuperUserResponseModel
    )
    EmployeeSelectionSuperUserResponseModel = EmployeeSelectionSuperUserResponseModel
    OccupantSelectionSuperUserResponseModel = OccupantSelectionSuperUserResponseModel