from typing import Union
import modules.database.schemas.nodes.users as user_schemas
import modules.database.schemas.nodes.workers.workers as worker_schemas
import modules.database.schemas.nodes.schools.schools as school_schemas

user_entities = [
    user_schemas.UserNode
]

worker_entities = Union[
    worker_schemas.TeacherNode,
    worker_schemas.StudentNode,
    worker_schemas.SchoolAdminNode,
    worker_schemas.DeveloperNode,
    worker_schemas.SuperAdminNode,
]

school_entities = Union[
    school_schemas.SchoolNode,
    school_schemas.DepartmentNode,
    school_schemas.SubjectClassNode,
    school_schemas.RoomNode,
]

entities = Union[
    user_entities,
    worker_entities,
    school_entities,
]