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


class LoginSuperUserRequestModel(Login):
    pass


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

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


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