from course_service.maps.models import CourseRoute


class OutApiRoute:
    @staticmethod
    def change_id(id: int) -> str:
        return CourseRoute.objects.get(id=id).out_key

    @staticmethod
    def get_route_matrix(out_key):
        map_list = [i.map_course for i in CourseRoute.objects.get(out_key=out_key).pivotcoursemap_set.all()]
        return [[j.vertex.token for j in i.pivotvertex_set.all()] for i in map_list]

    @staticmethod
    def get_route(out_key):
        return CourseRoute.objects.get(out_key=out_key)